com.nflabs.zeppelin.cli.ZeppelinCli Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zeppelin-cli Show documentation
There is a newer version: 0.3.3
package com.nflabs.zeppelin.cli;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.LinkedList;
import java.util.List;

import jline.ConsoleReader;

import org.apache.commons.cli.BasicParser;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.HelpFormatter;
import org.apache.commons.cli.Option;
import org.apache.commons.cli.OptionBuilder;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;

import com.nflabs.zeppelin.result.Result;
import com.nflabs.zeppelin.result.ResultDataException;
import com.nflabs.zeppelin.result.ResultDataStream;
import com.nflabs.zeppelin.zengine.Z;
import com.nflabs.zeppelin.zengine.ZException;
import com.nflabs.zeppelin.zengine.ZQL;
import com.nflabs.zeppelin.zengine.ZQLException;

public class ZeppelinCli {
	public static void main(String args[]) throws ParseException, ZException, IOException, ZQLException, SQLException, ResultDataException{
	
		Options options = new Options();
		options.addOption(OptionBuilder.withArgName("File")
									   .withDescription("Run ZQL in given text file")
									   .hasArg(true)
									   .create("f"));
		options.addOption(OptionBuilder.withArgName("ZQLStatement")
				   .withDescription("Run ZQL statement")
				   .hasArg(true)
				   .create("e"));		
		options.addOption(OptionBuilder.withDescription("See this messages")
									   .withLongOpt("help")
									   .create("h"));
									 
				
		
		CommandLineParser parser = new BasicParser();
		CommandLine cmd = parser.parse(options, args);
		
		if(cmd.hasOption("f")){
			ZQL zql = new ZQL();
			zql.load(new File(cmd.getOptionValue("f")));
			List zs = zql.compile();
			
			
			for(Z z : zs){
				z.execute().result().write(System.out);
			}
		} else if(cmd.hasOption("e")){ 			
			ZQL zql = new ZQL(cmd.getOptionValue("e"));
			List zs = zql.compile();			
			
			for(Z z : zs){
				z.execute().result().write(System.out);
			}
		} else if(cmd.hasOption("h")){
			HelpFormatter formatter = new HelpFormatter();
			formatter.printHelp("ZeppelinCli", options);
		} else {
			ZeppelinCli cli = new ZeppelinCli();
			cli.run();
		}
		

	}
	
	public ZeppelinCli(){
		
	}
	
	 private String readLine(ConsoleReader reader, String promtMessage)
             throws IOException {
	     String line = reader.readLine(promtMessage + "\nzeppelin> ");
	     if(line==null) return null;
	     else return line.trim();
	 }
	 
	 public void run() throws IOException{
		 printWelcomeMessage();
         ConsoleReader reader = new ConsoleReader();
         reader.setBellEnabled(false);
         String line;

         while ((line = readLine(reader, "")) != null) {
        	 ZQL zql;
			try {
				zql = new ZQL(line);
	 			List zs = zql.compile();			
				
	 			for(Z z : zs){
	 				Result result = z.execute().result();
	 				if(result!=null){
	 					result.write(System.out);
	 				}
	 			}
	 			
	 			for(Z z : zs){
	 				z.release();
	 			}
			} catch (ZException e) {
				e.printStackTrace();
			} catch (ZQLException e) {
				e.printStackTrace();
			}

         }
	 }

	 private void printWelcomeMessage() {
		System.out.println("Welcome to Zeppelin.");
	 }
}