001/**
002 * Licensed to the Apache Software Foundation (ASF) under one
003 * or more contributor license agreements.  See the NOTICE file
004 * distributed with this work for additional information
005 * regarding copyright ownership.  The ASF licenses this file
006 * to you under the Apache License, Version 2.0 (the
007 * "License"); you may not use this file except in compliance
008 * with the License.  You may obtain a copy of the License at
009 *
010 *     http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018package org.apache.oozie.action.hadoop;
019
020import static org.apache.oozie.action.hadoop.LauncherMapper.CONF_OOZIE_ACTION_MAIN_CLASS;
021
022import java.io.BufferedReader;
023import java.io.IOException;
024import java.io.InputStream;
025import java.io.InputStreamReader;
026import java.io.StringReader;
027import java.net.URISyntaxException;
028import java.util.List;
029import java.util.Properties;
030
031import org.apache.hadoop.conf.Configuration;
032import org.apache.hadoop.fs.FileSystem;
033import org.apache.hadoop.fs.Path;
034import org.apache.hadoop.mapred.RunningJob;
035import org.apache.oozie.action.ActionExecutorException;
036import org.apache.oozie.client.WorkflowAction;
037import org.apache.oozie.client.XOozieClient;
038import org.apache.oozie.service.HadoopAccessorException;
039import org.apache.oozie.util.IOUtils;
040import org.jdom.Element;
041import org.jdom.JDOMException;
042import org.jdom.Namespace;
043
044public class HiveActionExecutor extends ScriptLanguageActionExecutor {
045
046    private static final String HIVE_MAIN_CLASS_NAME = "org.apache.oozie.action.hadoop.HiveMain";
047    static final String HIVE_SCRIPT = "oozie.hive.script";
048    static final String HIVE_PARAMS = "oozie.hive.params";
049    static final String HIVE_ARGS = "oozie.hive.args";
050
051    public HiveActionExecutor() {
052        super("hive");
053    }
054
055    @Override
056    protected List<Class> getLauncherClasses() {
057        List<Class> classes = super.getLauncherClasses();
058        try {
059            classes.add(Class.forName(HIVE_MAIN_CLASS_NAME));
060        }
061        catch (ClassNotFoundException e) {
062            throw new RuntimeException("Class not found", e);
063        }
064        return classes;
065    }
066
067    @Override
068    protected String getLauncherMain(Configuration launcherConf, Element actionXml) {
069        return launcherConf.get(CONF_OOZIE_ACTION_MAIN_CLASS, HIVE_MAIN_CLASS_NAME);
070    }
071
072    @Override
073    @SuppressWarnings("unchecked")
074    Configuration setupActionConf(Configuration actionConf, Context context, Element actionXml,
075                                  Path appPath) throws ActionExecutorException {
076        Configuration conf = super.setupActionConf(actionConf, context, actionXml, appPath);
077
078        Namespace ns = actionXml.getNamespace();
079        String script = actionXml.getChild("script", ns).getTextTrim();
080        String scriptName = new Path(script).getName();
081        String hiveScriptContent = context.getProtoActionConf().get(XOozieClient.HIVE_SCRIPT);
082
083        if (hiveScriptContent == null){
084            addToCache(conf, appPath, script + "#" + scriptName, false);
085        }
086
087        List<Element> params = (List<Element>) actionXml.getChildren("param", ns);
088        String[] strParams = new String[params.size()];
089        for (int i = 0; i < params.size(); i++) {
090            strParams[i] = params.get(i).getTextTrim();
091        }
092        String[] strArgs = null;
093        List<Element> eArgs = actionXml.getChildren("argument", ns);
094        if (eArgs != null && eArgs.size() > 0) {
095            strArgs = new String[eArgs.size()];
096            for (int i = 0; i < eArgs.size(); i++) {
097                strArgs[i] = eArgs.get(i).getTextTrim();
098            }
099        }
100
101        setHiveScript(conf, scriptName, strParams, strArgs);
102        return conf;
103    }
104
105    public static void setHiveScript(Configuration conf, String script, String[] params, String[] args) {
106        conf.set(HIVE_SCRIPT, script);
107        MapReduceMain.setStrings(conf, HIVE_PARAMS, params);
108        MapReduceMain.setStrings(conf, HIVE_ARGS, args);
109    }
110
111    @Override
112    protected boolean getCaptureOutput(WorkflowAction action) throws JDOMException {
113        return true;
114    }
115
116    @Override
117    protected void getActionData(FileSystem actionFs, RunningJob runningJob, WorkflowAction action, Context context)
118            throws HadoopAccessorException, JDOMException, IOException, URISyntaxException {
119        super.getActionData(actionFs, runningJob, action, context);
120
121        // Load stored Hadoop jobs ids and promote them as external child ids on job success
122        Properties props = new Properties();
123        props.load(new StringReader(action.getData()));
124        context.setExternalChildIDs((String) props.get(LauncherMain.HADOOP_JOBS));
125    }
126
127    @Override
128    protected void setActionCompletionData(Context context, FileSystem actionFs) throws IOException,
129            HadoopAccessorException, URISyntaxException {
130        super.setActionCompletionData(context, actionFs);
131
132        // Load stored Hadoop jobs ids and promote them as external child ids on job failure
133        Path externalChildIDs = LauncherMapperHelper.getExternalChildIDsDataPath(context.getActionDir());
134        if (actionFs.exists(externalChildIDs)) {
135            InputStream is = actionFs.open(externalChildIDs);
136            BufferedReader reader = new BufferedReader(new InputStreamReader(is));
137            context.setExternalChildIDs(IOUtils.getReaderAsString(reader, -1));
138            reader.close();
139        }
140    }
141
142    /**
143     * Return the sharelib name for the action.
144     *
145     * @return returns <code>hive</code>.
146     * @param actionXml
147     */
148    @Override
149    protected String getDefaultShareLibName(Element actionXml) {
150        return "hive";
151    }
152
153    protected String getScriptName() {
154        return XOozieClient.HIVE_SCRIPT;
155    }
156
157}