如何在Apache Beam项目中直接使用google-cloud-storage

时间:2018-04-25 09:33:48

标签: java google-cloud-storage google-cloud-dataflow apache-beam

我们正在开发一个Apache Beam项目(版本2.4.0),我们还希望通过google-cloud-storage API直接使用存储桶。但是,将一些波束依赖性与云存储相结合会导致难以解决的依赖性问题。

我们看到Beam 2.4.0依赖于云存储1.22.0,所以这就是我们在下面使用它的原因。我们遇到了与1.27.0相同的问题。以下pom.xml指定了我们在项目中使用的四个梁依赖关系,其中最后两个导致问题。

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>com.bol</groupId>
    <artifactId>beam-plus-storage</artifactId>
    <version>1.0-SNAPSHOT</version>
    <properties>
        <beam.version>2.4.0</beam.version>
    </properties>

    <dependencies>
        <!-- These first two dependencies do not clash -->
        <dependency>
            <groupId>org.apache.beam</groupId>
            <artifactId>beam-runners-direct-java</artifactId>
            <version>${beam.version}</version>
        </dependency>
        <dependency>
            <groupId>org.apache.beam</groupId>
            <artifactId>beam-sdks-java-extensions-join-library</artifactId>
            <version>${beam.version}</version>
        </dependency>
        <!-- This one leads to java.lang.ClassNotFoundException: com.google.api.gax.rpc.HeaderProvider -->
        <dependency>
            <groupId>org.apache.beam</groupId>
            <artifactId>beam-sdks-java-io-google-cloud-platform</artifactId>
            <version>${beam.version}</version>
        </dependency>
        <!-- This one leads to java.lang.NoSuchMethodError: com.google.api.services.storage.Storage$Objects$List.setUserProject(...) -->
        <dependency>
            <groupId>org.apache.beam</groupId>
            <artifactId>beam-runners-google-cloud-dataflow-java</artifactId>
            <version>${beam.version}</version>
        </dependency>

        <dependency>
            <groupId>com.google.cloud</groupId>
            <artifactId>google-cloud-storage</artifactId>
            <version>1.22.0</version>
        </dependency>

    </dependencies>
</project>

以下是存储API的最小工作/损坏用法,列出了公共存储桶中的文件。

import com.google.api.gax.paging.Page;
import com.google.cloud.storage.Blob;
import com.google.cloud.storage.Storage;
import com.google.cloud.storage.StorageOptions;

public class CloudStorageReader {

    public static void main(String[] args) {
        Storage storage = StorageOptions.getDefaultInstance().getService();
        Page<Blob> list = storage.list("gcp-public-data-landsat", Storage.BlobListOption.currentDirectory(), Storage.BlobListOption.prefix("LC08/PRE/044/034/LC80440342016259LGN00/"));
        for (Blob blob : list.getValues()) {
            System.out.println(blob);
        }
    }
}

删除最后两个依赖项时,列出存储桶的内容可以正常工作。使用java-io beam依赖项时,找不到HeaderProvider类。由于数据流依赖性,找不到setUserProject方法。请参阅pom中的注释以获取完整的错误消息。

我们花了很多时间尝试修复HeaderProvider错误,这是导入所有四个波束依赖关系时出现的错误。我们为冲突依赖项添加了显式导入,并在梁导入中添加了排除。每次我们添加显式依赖项时,都会弹出另一个相关问题。我们尝试了maven阴影,由于我们的项目包装不实用,所以从来没有让它起作用。

最后,我们为云存储交互创建了一个单独的子模块+ jar,为我们的打包/运行带来了更多的复杂性。

最后一点,我们在尝试使用BigQuery API时遇到了同样的问题,但通过重新使用包私有光束代码解决了这个问题。

如果某人 有(相对简单的)方法让这些库一起工作,或者可以确认这确实是一个具有挑战性的依赖性问题,可能需要在Apache Beam中解决,这将是非常棒的

1 个答案:

答案 0 :(得分:5)

您可以利用Beam包含的FileSystems API列出存储桶,读取/写入文件以及删除云存储上的对象,而不是为云存储添加单独的依赖项。下面是一个示例,它列出了存储桶下的所有文件,然后将其中一个文件读入字符串。

// Set the default pipeline options so the various filesystems are
// loaded into the registry. This shouldn't be necessary if used
// within a pipeline.
FileSystems.setDefaultPipelineOptions(PipelineOptionsFactory.create());

// List Bucket
MatchResult listResult = FileSystems.match("gs://filesystems-demo/**/*");
listResult
    .metadata()
    .forEach(
        metadata -> {
          ResourceId resourceId = metadata.resourceId();
          System.out.println(resourceId.toString());
        });


// Read file
ResourceId existingFileResourceId = FileSystems
    .matchSingleFileSpec("gs://filesystems-demo/test-file1.csv")
    .resourceId();

try (ByteArrayOutputStream out = new ByteArrayOutputStream();
    ReadableByteChannel readerChannel = FileSystems.open(existingFileResourceId);
    WritableByteChannel writerChannel = Channels.newChannel(out)) {
  ByteStreams.copy(readerChannel, writerChannel);

  System.out.println("File contents: \n" + out.toString());
}


// Write file
String contentToWrite = "Laces out Dan!";

ResourceId newFileResourceId = FileSystems
    .matchNewResource("gs://filesystems-demo/new-file.txt", false);

try (ByteArrayInputStream in = new ByteArrayInputStream(contentToWrite.getBytes());
    ReadableByteChannel readerChannel = Channels.newChannel(in);
    WritableByteChannel writerChannel = FileSystems.create(newFileResourceId, MimeTypes.TEXT)) {

  ByteStreams.copy(readerChannel, writerChannel);
}