src/main/java/com/google/devtools/build/lib/cmdline/LabelCodec.java

// Copyright 2023 The Bazel Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package com.google.devtools.build.lib.cmdline;

import static com.google.devtools.build.lib.unsafe.UnsafeProvider.getFieldOffset;
import static com.google.devtools.build.lib.unsafe.UnsafeProvider.unsafe;

import com.google.devtools.build.lib.skyframe.serialization.AsyncDeserializationContext;
import com.google.devtools.build.lib.skyframe.serialization.InterningObjectCodec;
import com.google.devtools.build.lib.skyframe.serialization.SerializationContext;
import com.google.devtools.build.lib.skyframe.serialization.SerializationException;
import com.google.errorprone.annotations.Keep;
import com.google.protobuf.CodedInputStream;
import com.google.protobuf.CodedOutputStream;
import java.io.IOException;

/** Custom codec needed to correctly handle name interning. */
@Keep // Used reflectively.
final class LabelCodec extends InterningObjectCodec<Label> {
  private static final long PACKAGE_IDENTIFIER_OFFSET;
  private static final long NAME_OFFSET;

  static {
    try {
      PACKAGE_IDENTIFIER_OFFSET = getFieldOffset(Label.class, "packageIdentifier");
      NAME_OFFSET = getFieldOffset(Label.class, "name");
    } catch (NoSuchFieldException e) {
      throw new ExceptionInInitializerError(e);
    }
  }

  @Override
  public Class<Label> getEncodedClass() {
    return Label.class;
  }

  @Override
  public void serialize(SerializationContext context, Label obj, CodedOutputStream codedOut)
      throws SerializationException, IOException {
    context.serialize(obj.getPackageIdentifier(), codedOut);
    context.serialize(obj.getName(), codedOut);
  }

  @Override
  public Label deserializeInterned(AsyncDeserializationContext context, CodedInputStream codedIn)
      throws SerializationException, IOException {
    Label label;
    try {
      label = (Label) unsafe().allocateInstance(Label.class);
    } catch (ReflectiveOperationException e) {
      throw new SerializationException("Could not instantiate Label", e);
    }
    context.deserialize(codedIn, label, PACKAGE_IDENTIFIER_OFFSET);
    context.deserialize(codedIn, label, LabelCodec::setName);
    return label;
  }

  @Override
  public Label intern(Label label) {
    return Label.getLabelInterner().intern(label);
  }

  private static void setName(Label label, Object name) {
    unsafe().putObject(label, NAME_OFFSET, Label.internIfConstantName((String) name));
  }
}