net.oletalk.hellospringboot.dao.S3Dao.java Source code

Introduction

Here is the source code for net.oletalk.hellospringboot.dao.S3Dao.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package net.oletalk.hellospringboot.dao;

import com.amazonaws.AmazonClientException;
import com.amazonaws.AmazonServiceException;
import com.amazonaws.auth.ClasspathPropertiesFileCredentialsProvider;
import com.amazonaws.services.s3.AmazonS3;
import com.amazonaws.services.s3.AmazonS3Client;
import com.amazonaws.services.s3.model.AmazonS3Exception;
import com.amazonaws.services.s3.model.GetObjectRequest;
import com.amazonaws.services.s3.model.PutObjectRequest;
import com.amazonaws.services.s3.model.S3Object;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.Charset;
import net.oletalk.hellospringboot.exception.S3Exception;
import org.apache.log4j.Logger;
import org.apache.tomcat.util.http.fileupload.IOUtils;
import org.springframework.stereotype.Repository;

/**
 *
 * @author colin
 */
@Repository
public class S3Dao {
    /*
     * Sample bucket 'oletalk' and key is your filename
    */
    private static final Logger LOG = Logger.getLogger(S3Dao.class);

    public void uploadFile(String bucketName, String key, File file) throws S3Exception {
        AmazonS3 s3client = new AmazonS3Client(new ClasspathPropertiesFileCredentialsProvider());
        try {
            LOG.info("Uploading file to S3");
            s3client.putObject(new PutObjectRequest(bucketName, key, file));
        } catch (AmazonServiceException ase) {
            LOG.error("Problem uploading file to S3: " + ase.getMessage() + " (status code " + ase.getStatusCode()
                    + ")");
            throw new S3Exception("Problem uploading file to S3");
        } catch (AmazonClientException ace) {
            LOG.error("Internal error uploading file to S3: " + ace.getMessage());
            throw new S3Exception("Problem uploading file to S3");
        }
        LOG.info("Upload complete");

    }

    public void getObjectData(String bucketName, String key, OutputStream out) throws S3Exception {
        AmazonS3 s3client = new AmazonS3Client(new ClasspathPropertiesFileCredentialsProvider());
        LOG.info("fetching requested object " + key);
        try {
            S3Object object = s3client.getObject(new GetObjectRequest(bucketName, key));
            LOG.info("fetched, serving up");
            try {
                IOUtils.copy(object.getObjectContent(), out);
            } catch (IOException ioe) {
                LOG.error("Problem writing message: " + ioe.getMessage());
            }

        } catch (AmazonS3Exception e) {
            LOG.error("Problem fetching from S3: ", e);
            String errorMsg = "Error fetching document";
            try {
                out.write(errorMsg.getBytes(Charset.defaultCharset()));
                throw new S3Exception("Error fetching document");
            } catch (IOException ioe) {
                LOG.error("Problem writing message: " + ioe.getMessage());
            }
        }
    }
}