src/main/java/com/google/devtools/build/lib/buildeventstream/LocalFilesArtifactUploader.java

// Copyright 2018 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.buildeventstream;

import static com.google.common.util.concurrent.Futures.immediateFuture;

import com.google.common.util.concurrent.ListenableFuture;
import com.google.devtools.build.lib.buildeventstream.BuildEvent.LocalFile;
import com.google.devtools.build.lib.buildeventstream.BuildEvent.LocalFile.LocalFileType;
import com.google.devtools.build.lib.buildeventstream.PathConverter.FileUriPathConverter;
import com.google.devtools.build.lib.vfs.Path;
import io.netty.util.AbstractReferenceCounted;
import io.netty.util.ReferenceCounted;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import javax.annotation.Nullable;

/** An uploader that simply turns paths into local file URIs. */
public class LocalFilesArtifactUploader extends AbstractReferenceCounted
    implements BuildEventArtifactUploader {
  private static final FileUriPathConverter FILE_URI_PATH_CONVERTER = new FileUriPathConverter();
  private final ConcurrentHashMap<Path, Boolean> fileIsDirectory = new ConcurrentHashMap<>();

  @Override
  public ListenableFuture<PathConverter> upload(Map<Path, LocalFile> files) {
    return immediateFuture(new PathConverterImpl(files));
  }

  @Override
  protected void deallocate() {
    // Intentionally left empty
  }

  @Override
  public ReferenceCounted touch(Object o) {
    return this;
  }

  @Override
  public boolean mayBeSlow() {
    return false;
  }

  private class PathConverterImpl implements PathConverter {
    private final Map<Path, LocalFile> paths;

    private PathConverterImpl(Map<Path, LocalFile> paths) {
      this.paths = paths;
    }

    @Nullable
    @Override
    public String apply(Path path) {
      LocalFile localFile = paths.get(path);
      if (localFile == null) {
        // We should throw here, the file wasn't declared in BuildEvent#referencedLocalFiles
        return null;
      }
      LocalFileType type = localFile.type;
      if (type.equals(LocalFileType.OUTPUT_DIRECTORY)
          || type.equals(LocalFileType.OUTPUT_SYMLINK)) {
        return null;
      }
      if (type.equals(LocalFileType.OUTPUT)
          && fileIsDirectory.computeIfAbsent(path, Path::isDirectory)) {
        return null;
      }
      return FILE_URI_PATH_CONVERTER.apply(path);
    }
  }
}