如何通过Java API分块上传到谷歌云存储？

Question

我正在尝试，在一个Java的环境中，将日志文件以块的形式写入Google云存储。我有一个解析原始日志文件并生成JSON行的进程；我将JSON行存储在缓冲区中，每次缓冲区达到5MGB左右时，我都希望写入GCS中的同一文件，直到原始raw源代码被完全解析。我有一个类似的设置写入AWS3。由于内存问题，大块写作是完成的。

我设法向GCS写入了一个文件，如下所示（gcsService是一个配置了身份验证等的存储对象）：

private void uploadStream(String path, String name, String contentType, InputStream stream, String bucketName) throws IOException, GeneralSecurityException {
    InputStreamContent contentStream = new InputStreamContent(contentType, stream);
    StorageObject objectMetadata = new StorageObject()
                .setName(path+"/"+name)
                .setAcl(Arrays.asList(new ObjectAccessControl().setEntity("allUsers").setRole("READER")));
    Storage.Objects.Insert insertRequest = gcsService.objects()
                .insert(bucketName, objectMetadata, contentStream);   
    insertRequest.execute();
}

不幸的是，我一直无法弄清楚如何以块的形式写到GCS。谷歌的文档似乎建议了两种方法。一个涉及“可恢复的”插入请求:https://cloud.google.com/storage/docs/jsonapi/v1/how-tos/upload

另一种方法涉及“compose”请求:https://cloud.google.com/storage/docs/jsonapi/v1/objects/compose

我一直试图得到一个“可恢复的”上传设置，但我不能使它工作。

有什么想法吗？我的具体问题是：

• 什么是以块形式上载到GCS的优雅和/或合适的方式？
• 有没有人知道如何在Java通过插入请求设置可恢复的上载到GCS？那能行吗？

Answer 1

让它起作用了-这是个麻烦。我的问题的答案如下：

• “可恢复”上载在Java很管用，是一种优雅的方式，也许是将文件分块上载到GCS的首选方式（我不是专家，所以我不确定）。
• 可以在Java设置“可恢复”上载，如下所述。

最后我有两种方法--一种用于启动上传，另一种用于发送块。

private String initiateResumableUpload() throws IOException {
        String URI = "https://storage.googleapis.com/" + bucket + "/" + path;
        GenericUrl url = new GenericUrl(URI);
        HttpRequest req = requestFactory.buildPostRequest(url, new ByteArrayContent("text/plain", new byte[0]));
        HttpHeaders headers = new HttpHeaders();
        headers.set("x-goog-resumable", "start");
        headers.setContentLength((long) 0);
        headers.setContentType("text/plain");
        req.setHeaders(headers);
        req.setReadTimeout((int) DEFAULT_TIMEOUT);
        req.setResponseHeaders(headers);
        HttpResponse resp;
        try {
            resp = req.execute();
        } catch (IOException e) {
            throw e;
        }
        if (resp.getStatusCode() == 201) {
            String location = resp.getHeaders().getLocation();
            return location;

        } else {
            throw new IOException();
        }
    }

requestFactory应该知道您适当生成的凭据。

private void writeChunk(final boolean isFinalChunk) throws HttpResponseException, IOException {
    System.out.println("Writing chunk number " + Integer.toString(chunkCount) + ".");

    try (InputStream inputStream = new ByteBufInputStream(buffer)) {
        int length = Math.min(buffer.readableBytes(), DEFAULT_UPLOAD_CHUNK_SIZE);
        HttpContent contentsend = new InputStreamContent("text/plain", inputStream);

        String URI = location;
        GenericUrl url = new GenericUrl(URI);
        HttpRequest req = requestFactory.buildPutRequest(url, contentsend);

        int offset = chunkCount*DEFAULT_UPLOAD_CHUNK_SIZE;
        long limit = offset + length;
        HttpHeaders headers = new HttpHeaders();
        headers.setContentLength((long) length);
        headers.setContentRange("bytes " + (length == 0 ? "*" : offset + "-" + (limit - 1)) + (isFinalChunk ? "/" + limit : "/*"));

        req.setHeaders(headers);

        req.setReadTimeout((int) DEFAULT_TIMEOUT);

        try {
            req.execute();
            } 
        catch (HttpResponseException e) {
                if(e.getMessage().equals("308 Resume Incomplete"))
                {
                    ++chunkCount;
                }
                else
                {
                    throw e;
                }
            }
        catch (Exception e) {
            throw e;
        }
    }
}

我的缓冲区是io.netty.buffer.bytebuf。

我的与GCS相关的导入包括：

import com.google.api.client.googleapis.auth.oauth2.GoogleCredential;
import com.google.api.client.googleapis.javanet.GoogleNetHttpTransport;
import com.google.api.client.http.ByteArrayContent;
import com.google.api.client.http.GenericUrl;
import com.google.api.client.http.HttpContent;
import com.google.api.client.http.HttpHeaders;
import com.google.api.client.http.HttpRequest;
import com.google.api.client.http.HttpRequestFactory;
import com.google.api.client.http.HttpResponse;
import com.google.api.client.http.HttpResponseException;
import com.google.api.client.http.HttpTransport;

上面的代码可能有一些bug，但它确实成功地将一个文件块写入GCS。

我还设法通过一个不同的库和“compose”请求来完成任务。但“可恢复”的方式似乎更合适。

干杯祝你好运。