| /** |
| * 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. |
| */ |
| import com.jayway.jsonpath.JsonPath |
| import groovy.json.JsonSlurper |
| import org.apache.knox.gateway.shell.KnoxSession |
| import org.apache.knox.gateway.shell.hdfs.Hdfs |
| import org.apache.knox.gateway.shell.job.Job |
| |
| import static java.util.concurrent.TimeUnit.SECONDS |
| import org.apache.knox.gateway.shell.Credentials |
| |
| gateway = "https://localhost:8443/gateway/sandbox" |
| |
| credentials = new Credentials() |
| credentials.add("ClearInput", "Enter username: ", "user") |
| .add("HiddenInput", "Enter pas" + "sword: ", "pass") |
| credentials.collect() |
| |
| username = credentials.get("user").string() |
| pass = credentials.get("pass").string() |
| |
| jobDir = "/user/" + username + "/test" |
| |
| session = KnoxSession.login( gateway, username, pass ) |
| |
| println "Delete " + jobDir + ": " + Hdfs.rm( session ).file( jobDir ).recursive().now().statusCode |
| println "Create " + jobDir + ": " + Hdfs.mkdir( session ).dir( jobDir ).now().statusCode |
| |
| id_pig = ''' |
| A = load 'test/input/$filename' using PigStorage(':'); |
| B = foreach A generate $0 as id; |
| dump B; |
| ''' |
| |
| fake_passwd = '''ctdean:Chris Dean:secret |
| pauls:Paul Stolorz:good |
| carmas:Carlos Armas:evil |
| dra:Deirdre McClure:marvelous |
| ''' |
| |
| Hdfs.put(session).text( id_pig ).to( jobDir + "/input/id.pig" ).now() |
| Hdfs.put(session).text( fake_passwd ).to( jobDir + "/input/passwd" ).now() |
| |
| jobId = Job.submitPig(session) \ |
| .file("${jobDir}/input/id.pig") \ |
| .arg("-v") \ |
| .arg("-p").arg("filename=passwd") \ |
| .statusDir("${jobDir}/output") \ |
| .now().jobId |
| |
| println "Submitted job: " + jobId |
| |
| println "Polling up to 60s for job completion..." |
| done = false |
| count = 0 |
| while( !done && count++ < 60 ) { |
| sleep( 1000 ) |
| json = Job.queryStatus(session).jobId(jobId).now().string |
| done = JsonPath.read( json, "\$.status.jobComplete" ) |
| print "."; System.out.flush(); |
| } |
| println "" |
| println "Job status: " + done |
| |
| text = Hdfs.ls( session ).dir( jobDir + "/output" ).now().string |
| json = (new JsonSlurper()).parseText( text ) |
| println json.FileStatuses.FileStatus.pathSuffix |
| |
| println "Session closed: " + session.shutdown( 10, SECONDS ) |