AWS Amazon S3 Java SDK-在上传大文件时过期时刷新凭证/令牌

时间:2018-08-13 17:04:26

标签: java amazon-web-services amazon-s3 aws-sdk aws-sdk-java-2.0

我正在尝试将一个大文件上传到使用令牌的服务器,并且令牌在10分钟后过期,因此,如果我上传一个小文件,它将起作用,因此,如果文件太大,则会出现一些问题,并且将在访问被拒绝时尝试永久上传

所以我需要刷新 BasicAWSCredentials 中的令牌,该令牌比 AWSStaticCredentialsProvider 中使用的令牌多,因此我不确定该怎么做,请帮助=)< / p>

值得一提的是,我们使用提供令牌的本地服务器(不是Amazon Cloud),为方便起见,我们使用Amazon的代码。

这是我的代码:

public void uploadMultipart(File file) throws Exception {
    //this method will give you a initial token for a given user, 
    //than calculates when a new token is needed and will refresh it just when necessary

    String token = getUsetToken();
    String existingBucketName = myTenant.toLowerCase() + ".package.upload";
    String endPoint = urlAPI + "s3/buckets/";
    String strSize = FileUtils.byteCountToDisplaySize(FileUtils.sizeOf(file));
    System.out.println("File size: " + strSize);

    AwsClientBuilder.EndpointConfiguration endpointConfiguration = new AwsClientBuilder.EndpointConfiguration(endPoint, null);//note: Region has to be null
    //AWSCredentialsProvider        
    BasicAWSCredentials sessionCredentials = new BasicAWSCredentials(token, "NOT_USED");//secretKey should be set to NOT_USED

    AmazonS3 s3 = AmazonS3ClientBuilder
            .standard()
            .withCredentials(new AWSStaticCredentialsProvider(sessionCredentials))
            .withEndpointConfiguration(endpointConfiguration)
            .enablePathStyleAccess()
            .build();

    int maxUploadThreads = 5;
    TransferManager tm = TransferManagerBuilder
            .standard()
            .withS3Client(s3)
            .withMultipartUploadThreshold((long) (5 * 1024 * 1024))
            .withExecutorFactory(() -> Executors.newFixedThreadPool(maxUploadThreads))
            .build();

    PutObjectRequest request = new PutObjectRequest(existingBucketName, file.getName(), file);
    //request.putCustomRequestHeader("Access-Token", token);
    ProgressListener progressListener = progressEvent -> System.out.println("Transferred bytes: " + progressEvent.getBytesTransferred());
    request.setGeneralProgressListener(progressListener);
    Upload upload = tm.upload(request);

    LocalDateTime uploadStartedAt = LocalDateTime.now();
    log.info("Starting upload at: " + uploadStartedAt);

    try {
        upload.waitForCompletion();
        //upload.waitForUploadResult();
        log.info("Upload completed. " + strSize);

    } catch (Exception e) {//AmazonClientException
        log.error("Error occurred while uploading file - " + strSize);
        e.printStackTrace();
    }
}

2 个答案:

答案 0 :(得分:1)

找到解决方案!

我找到了一种方法来使此工作正常进行,老实说,我对结果感到非常满意,我已经使用大文件(50gd.zip)进行了许多测试,并且在每种情况下都表现良好

我的解决方法是,删除该行: BasicAWSCredentials sessionCredentials = new BasicAWSCredentials(token, "NOT_USED");

AWSCredentials是一个接口,因此我们可以用动态的东西覆盖它, token 过期且需要新的新令牌的逻辑保存在 getToken中()方法,意味着您每次都可以通话而不会造成伤害

AWSCredentials sessionCredentials = new AWSCredentials() {
    @Override
    public String getAWSAccessKeyId() {
        try {
            return getToken(); //getToken() method return a string 
        } catch (Exception e) {
            return null;
        }
    }

    @Override
    public String getAWSSecretKey() {
        return "NOT_USED";
    }
};

答案 1 :(得分:0)

上载文件(或多部分文件的一部分)时,您使用的凭据必须持续足够长的时间才能完成上载。您无法刷新凭证,因为没有方法可以更新AWS S3,因为您正在为已签名的请求使用新凭证。

您可以将上传文件分成较小的文件,以便更快地上传。然后,仅上传X个部分。刷新您的凭据并上传Y部分。重复直到所有部分都上传。然后,您需要通过合并各个部分来完成操作(这是一个单独的命令)。这不是一个完美的解决方案,因为无法精确地控制传输速度,这意味着您将必须编写自己的上传代码(这并不困难)。