下面列出了怎么用com.amazonaws.services.s3.model.OutputSerialization的API类实例代码及写法,或者点击链接到github查看源代码。
private static SelectObjectContentRequest generateBaseCSVRequest(String bucket, String key, String query) {
SelectObjectContentRequest request = new SelectObjectContentRequest();
request.setBucketName(bucket);
request.setKey(key);
request.setExpression(query);
request.setExpressionType(ExpressionType.SQL);
InputSerialization inputSerialization = new InputSerialization();
inputSerialization.setCsv(new CSVInput());
inputSerialization.setCompressionType(CompressionType.NONE);
request.setInputSerialization(inputSerialization);
OutputSerialization outputSerialization = new OutputSerialization();
outputSerialization.setCsv(new CSVOutput());
request.setOutputSerialization(outputSerialization);
return request;
}
private static SelectObjectContentRequest generateBaseCSVRequest(String bucket, String key, String query) {
SelectObjectContentRequest request = new SelectObjectContentRequest();
request.setBucketName(bucket);
request.setKey(key);
request.setExpression(query);
request.setExpressionType(ExpressionType.SQL);
InputSerialization inputSerialization = new InputSerialization();
inputSerialization.setCsv(new CSVInput());
inputSerialization.setCompressionType(CompressionType.NONE);
request.setInputSerialization(inputSerialization);
OutputSerialization outputSerialization = new OutputSerialization();
outputSerialization.setCsv(new CSVOutput());
request.setOutputSerialization(outputSerialization);
return request;
}
@Override
public SelectObjectContentRequest buildSelectObjectRequest(Properties schema, String query, Path path)
{
SelectObjectContentRequest selectObjectRequest = new SelectObjectContentRequest();
URI uri = path.toUri();
selectObjectRequest.setBucketName(PrestoS3FileSystem.extractBucketName(uri));
selectObjectRequest.setKey(PrestoS3FileSystem.keyFromPath(path));
selectObjectRequest.setExpression(query);
selectObjectRequest.setExpressionType(ExpressionType.SQL);
String fieldDelimiter = getFieldDelimiter(schema);
String quoteChar = schema.getProperty(QUOTE_CHAR, null);
String escapeChar = schema.getProperty(ESCAPE_CHAR, null);
CSVInput selectObjectCSVInputSerialization = new CSVInput();
selectObjectCSVInputSerialization.setRecordDelimiter(lineDelimiter);
selectObjectCSVInputSerialization.setFieldDelimiter(fieldDelimiter);
selectObjectCSVInputSerialization.setComments(COMMENTS_CHAR_STR);
selectObjectCSVInputSerialization.setQuoteCharacter(quoteChar);
selectObjectCSVInputSerialization.setQuoteEscapeCharacter(escapeChar);
InputSerialization selectObjectInputSerialization = new InputSerialization();
selectObjectInputSerialization.setCompressionType(getCompressionType(path));
selectObjectInputSerialization.setCsv(selectObjectCSVInputSerialization);
selectObjectRequest.setInputSerialization(selectObjectInputSerialization);
OutputSerialization selectObjectOutputSerialization = new OutputSerialization();
CSVOutput selectObjectCSVOutputSerialization = new CSVOutput();
selectObjectCSVOutputSerialization.setRecordDelimiter(lineDelimiter);
selectObjectCSVOutputSerialization.setFieldDelimiter(fieldDelimiter);
selectObjectCSVOutputSerialization.setQuoteCharacter(quoteChar);
selectObjectCSVOutputSerialization.setQuoteEscapeCharacter(escapeChar);
selectObjectOutputSerialization.setCsv(selectObjectCSVOutputSerialization);
selectObjectRequest.setOutputSerialization(selectObjectOutputSerialization);
return selectObjectRequest;
}
/**
* Generates the {@link SelectObjectContentRequest} object from
* the request context.
*
* @param context the request context
* @return a {@link SelectObjectContentRequest}
*/
SelectObjectContentRequest generateBaseCSVRequest(RequestContext context) {
InputSerialization inputSerialization = getInputSerialization(context);
String fileHeaderInfo = context.getOption(FILE_HEADER_INFO);
boolean usePositionToIdentifyColumn = inputSerialization.getCsv() != null &&
(StringUtils.isBlank(fileHeaderInfo) ||
!StringUtils.equalsIgnoreCase(FILE_HEADER_INFO_USE, fileHeaderInfo));
String query = null;
try {
S3SelectQueryBuilder queryBuilder = new S3SelectQueryBuilder(context, usePositionToIdentifyColumn);
query = queryBuilder.buildSelectQuery();
} catch (SQLException e) {
LOG.error("Unable to build select query for filter string {}", context.getFilterString());
}
LOG.trace("Select query: {}", query);
SelectObjectContentRequest request = new SelectObjectContentRequest();
request.setBucketName(name.getHost());
request.setKey(StringUtils.removeStart(name.getPath(), "/"));
request.setExpression(query);
request.setExpressionType(ExpressionType.SQL);
LOG.debug("With bucket name '{}'", request.getBucketName());
LOG.debug("With key '{}'", request.getKey());
LOG.debug("With expression query '{}'", query);
request.setInputSerialization(inputSerialization);
OutputSerialization outputSerialization = getOutputSerialization(context);
request.setOutputSerialization(outputSerialization);
return request;
}
/**
* Returns a {@link com.amazonaws.services.s3.model.OutputSerialization}
* object with parsed values from the request context.
*
* @param context the request context
* @return a {@link OutputSerialization} object
*/
private OutputSerialization getOutputSerialization(RequestContext context) {
GreenplumCSV csv = context.getGreenplumCSV();
OutputSerialization outputSerialization = new OutputSerialization();
CSVOutput csvOutput = new CSVOutput();
csvOutput.setFieldDelimiter(csv.getDelimiter());
csvOutput.setQuoteCharacter(csv.getQuote());
csvOutput.setQuoteEscapeCharacter(csv.getEscape());
csvOutput.setRecordDelimiter(csv.getNewline());
outputSerialization.setCsv(csvOutput);
return outputSerialization;
}