src/main/java/com/google/devtools/build/lib/actions/StaticInputMetadataProvider.java

// Copyright 2019 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.actions;

import com.google.common.collect.ImmutableMap;
import java.util.Collection;
import java.util.Map;
import javax.annotation.Nullable;

/** A {@link InputMetadataProvider} backed by static data */
public final class StaticInputMetadataProvider implements InputMetadataProvider {

  private static final StaticInputMetadataProvider EMPTY =
      new StaticInputMetadataProvider(ImmutableMap.of());

  public static StaticInputMetadataProvider empty() {
    return EMPTY;
  }

  private final ImmutableMap<ActionInput, FileArtifactValue> inputToMetadata;
  private final ImmutableMap<String, ActionInput> execPathToInput;

  public StaticInputMetadataProvider(Map<ActionInput, FileArtifactValue> inputToMetadata) {
    this.inputToMetadata = ImmutableMap.copyOf(inputToMetadata);
    this.execPathToInput = constructExecPathToInputMap(inputToMetadata.keySet());
  }

  private static ImmutableMap<String, ActionInput> constructExecPathToInputMap(
      Collection<ActionInput> inputs) {
    ImmutableMap.Builder<String, ActionInput> builder =
        ImmutableMap.builderWithExpectedSize(inputs.size());
    for (ActionInput input : inputs) {
      builder.put(input.getExecPath().getPathString(), input);
    }
    return builder.buildOrThrow();
  }

  @Nullable
  @Override
  public FileArtifactValue getInputMetadata(ActionInput input) {
    return inputToMetadata.get(input);
  }

  @Nullable
  @Override
  public ActionInput getInput(String execPath) {
    return execPathToInput.get(execPath);
  }
}