
org.apache.hadoop.gateway.shell.job.Pig Maven / Gradle / Ivy
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.hadoop.gateway.shell.job;
import com.jayway.jsonpath.JsonPath;
import org.apache.hadoop.gateway.shell.AbstractRequest;
import org.apache.hadoop.gateway.shell.BasicResponse;
import org.apache.hadoop.gateway.shell.Hadoop;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.utils.URIBuilder;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.Callable;
public class Pig {
static class Request extends AbstractRequest {
private String group;
private String file;
private String arg;
private String statusDir;
public Request( Hadoop session ) {
super( session );
}
public Request group( String group ) {
this.group = group;
return this;
}
public Request file( String file ) {
this.file = file;
return this;
}
public Request arg( String arg ) {
this.arg = arg;
return this;
}
public Request statusDir( String dir ) {
this.statusDir = dir;
return this;
}
protected Callable callable() {
return new Callable() {
@Override
public Response call() throws Exception {
URIBuilder uri = uri( Job.SERVICE_PATH, "/pig" );
List params = new ArrayList();
addParam( params, "group", group );
addParam( params, "file", file );
addParam( params, "arg", arg );
addParam( params, "statusdir", statusDir );
UrlEncodedFormEntity form = new UrlEncodedFormEntity( params );
HttpPost request = new HttpPost( uri.build() );
request.setEntity( form );
return new Response( execute( request ) );
}
};
}
}
static class Response extends BasicResponse {
protected Response( HttpResponse response ) {
super( response );
}
public String getJobId() throws IOException {
return JsonPath.read( getString(), "$.id" );
}
}
}