/**
* Executes a query.
*
* @param cmd HiveQL query to execute
*/
public void execute(String cmd) throws HiveServerException, TException {
HiveServerHandler.LOG.info("Running the query: " + cmd);
SessionState ss = SessionState.get();
String cmd_trimmed = cmd.trim();
String[] tokens = cmd_trimmed.split("\\s");
String cmd_1 = cmd_trimmed.substring(tokens[0].length()).trim();
int ret = 0;
try {
CommandProcessor proc = CommandProcessorFactory.get(tokens[0]);
if (proc != null) {
if (proc instanceof Driver) {
isHiveQuery = true;
ret = driver.run(cmd);
} else {
isHiveQuery = false;
ret = proc.run(cmd_1);
}
}
} catch (Exception e) {
throw new HiveServerException("Error running query: " + e.toString());
}
if (ret != 0) {
throw new HiveServerException("Query returned non-zero code: " + ret);
}
}
public static void main(String[] args) {
try {
int port = 10000;
if (args.length >= 1) {
port = Integer.parseInt(args[0]);
}
TServerTransport serverTransport = new TServerSocket(port);
ThriftHiveProcessorFactory hfactory = new ThriftHiveProcessorFactory(null);
TThreadPoolServer.Options options = new TThreadPoolServer.Options();
TServer server =
new TThreadPoolServer(
hfactory,
serverTransport,
new TTransportFactory(),
new TTransportFactory(),
new TBinaryProtocol.Factory(),
new TBinaryProtocol.Factory(),
options);
HiveServerHandler.LOG.info("Starting hive server on port " + port);
server.serve();
} catch (Exception x) {
x.printStackTrace();
}
}
