jpayne@68: package tax;
jpayne@68: 
jpayne@68: import java.io.IOException;
jpayne@68: import java.io.InputStream;
jpayne@68: import java.net.MalformedURLException;
jpayne@68: import java.net.URL;
jpayne@68: import java.util.Arrays;
jpayne@68: import java.util.Locale;
jpayne@68: 
jpayne@68: import shared.Shared;
jpayne@68: import shared.Timer;
jpayne@68: 
jpayne@68: public class Query {
jpayne@68: 	
jpayne@68: 	/** For testing throughput */
jpayne@68: 	public static void main(String[] args){
jpayne@68: 		
jpayne@68: //		sun.net.http.errorstream.enableBuffering=true;
jpayne@68: 
jpayne@68: //		System.setProperty("sun.net.http.errorstream.enableBuffering", "true");
jpayne@68: //		System.setProperty("http.errorstream.enableBuffering", "true");
jpayne@68: //////		System.setProperty("", "");
jpayne@68: //		System.setProperty("http.keepAlive", "true");
jpayne@68: //		System.setProperty("http.maxConnections", "50");
jpayne@68: ////		System.setProperty("", "");
jpayne@68: ////		System.setProperty("", "");
jpayne@68: //
jpayne@68: //		System.out.println("sun.net.http.errorstream.enableBuffering="+System.getProperty("sun.net.http.errorstream.enableBuffering"));
jpayne@68: //		System.out.println("sun.net.http.errorstream.timeout="+System.getProperty("sun.net.http.errorstream.timeout"));
jpayne@68: //		System.out.println("sun.net.http.errorstream.bufferSize="+System.getProperty("sun.net.http.errorstream.bufferSize"));
jpayne@68: //		System.out.println("http.keepAlive="+System.getProperty("http.keepAlive"));
jpayne@68: //		System.out.println("http.maxConnections="+System.getProperty("http.maxConnections"));
jpayne@68: //		System.out.println(System.getProperties());
jpayne@68: //		System.out.println("="+System.getProperty(""));
jpayne@68: //		System.out.println("="+System.getProperty(""));
jpayne@68: //		System.out.println("="+System.getProperty(""));
jpayne@68: 		
jpayne@68: 		String x=args[0];
jpayne@68: 		int requests=(args.length>1 ? Integer.parseInt(args[1]) : 1);
jpayne@68: 
jpayne@68: 		Timer t=new Timer();
jpayne@68: 		Timer t2=new Timer();
jpayne@68: 		Timer t3=new Timer();
jpayne@68: 		
jpayne@68: //		URL url=toUrl(x);
jpayne@68: 		for(int i=0; i<requests; i++){
jpayne@68: 			String s=request(x);
jpayne@68: //			System.out.println(s);
jpayne@68: 			if(i==0){t3.start();}
jpayne@68: 			//t2.stop("");
jpayne@68: 			//t2.start();
jpayne@68: 		}
jpayne@68: 		
jpayne@68: 		t3.stop();
jpayne@68: 		
jpayne@68: 		t.stop("Time:  \t");
jpayne@68: 		double qps=requests*1000000000.0/(t.elapsed);
jpayne@68: 		System.err.println(String.format(Locale.ROOT, "Qps:   \t%.2f", qps));
jpayne@68: 		
jpayne@68: 		System.err.println("Time2: \t"+t3);
jpayne@68: 		double qps2=(requests-1)*1000000000.0/(t3.elapsed);
jpayne@68: 		System.err.println(String.format(Locale.ROOT, "Qps2:  \t%.2f", qps2));
jpayne@68: 		
jpayne@68: //		System.out.println(s);
jpayne@68: 		
jpayne@68: //		BufferedReader reader;
jpayne@68: //		try {
jpayne@68: //			reader = new BufferedReader(new InputStreamReader(url.openStream(), "UTF-8"));
jpayne@68: //
jpayne@68: //			for (String line; (line = reader.readLine()) != null;) {
jpayne@68: //				System.out.println(line);
jpayne@68: //			}
jpayne@68: //
jpayne@68: //		} catch (UnsupportedEncodingException e) {
jpayne@68: //			// TODO Auto-generated catch block
jpayne@68: //			e.printStackTrace();
jpayne@68: //		} catch (IOException e) {
jpayne@68: //			// TODO Auto-generated catch block
jpayne@68: //			e.printStackTrace();
jpayne@68: //		}
jpayne@68: 	}
jpayne@68: 	
jpayne@68: 	public static int[] getGi(String...args){
jpayne@68: 		return get("pt_gi", args);
jpayne@68: 	}
jpayne@68: 	
jpayne@68: 	public static int[] getAccession(String...args){
jpayne@68: 		return get("pt_accession", args);
jpayne@68: 	}
jpayne@68: 	
jpayne@68: 	public static int[] get(String type, String...args){
jpayne@68: 		String host=Shared.taxServer();
jpayne@68: 		StringBuilder sb=new StringBuilder(host.length()+32);
jpayne@68: 		sb.append(host);
jpayne@68: 		sb.append(type);
jpayne@68: 		sb.append('\t');
jpayne@68: 		String comma="";
jpayne@68: 		for(String arg : args){
jpayne@68: 			sb.append(comma);
jpayne@68: 			sb.append(arg);
jpayne@68: 			comma=",";
jpayne@68: 		}
jpayne@68: 		String result=request(sb.toString());
jpayne@68: 		String[] split=result.split(",");
jpayne@68: 		int[] ret=new int[split.length];
jpayne@68: 		for(int i=0; i<split.length; i++){
jpayne@68: 			ret[i]=Integer.parseInt(split[i]);
jpayne@68: 		}
jpayne@68: 		return ret;
jpayne@68: 	}
jpayne@68: 	
jpayne@68: 	public static String request(String x){
jpayne@68: 		InputStream is=toStream(x);
jpayne@68: 		if(is==null){return null;}
jpayne@68: 		try {
jpayne@68: 			byte[] buffer=new byte[256];
jpayne@68: 			int count=is.read(buffer);
jpayne@68: 			int next=0;
jpayne@68: 			while(count>-1){
jpayne@68: 				next+=count;
jpayne@68: 				if(next>=buffer.length){
jpayne@68: 					buffer=Arrays.copyOf(buffer, buffer.length*2);
jpayne@68: 				}
jpayne@68: 				count=is.read(buffer, next, buffer.length-next);
jpayne@68: 			}
jpayne@68: 			is.close(); //TODO: Why am I not closing this stream?
jpayne@68: 			return new String(buffer, 0, next);
jpayne@68: 		} catch (IOException e) {
jpayne@68: 			// TODO Auto-generated catch block
jpayne@68: 			e.printStackTrace();
jpayne@68: 		}
jpayne@68: 		try {
jpayne@68: 			is.close();
jpayne@68: 		} catch (IOException e) {
jpayne@68: 			// TODO Auto-generated catch block
jpayne@68: 			e.printStackTrace();
jpayne@68: 		}
jpayne@68: 		return null;
jpayne@68: 	}
jpayne@68: 	
jpayne@68: 	public static InputStream toStream(String x){
jpayne@68: 		URL url=null;
jpayne@68: 		
jpayne@68: 		try {
jpayne@68: 			url = new URL(x);
jpayne@68: 		} catch (MalformedURLException e1) {
jpayne@68: 			// TODO Auto-generated catch block
jpayne@68: 			e1.printStackTrace();
jpayne@68: 			return null;
jpayne@68: 		}
jpayne@68: 		
jpayne@68: 		try {
jpayne@68: 			InputStream is=url.openStream();
jpayne@68: 			return is;
jpayne@68: 		} catch (IOException e) {
jpayne@68: 			// TODO Auto-generated catch block
jpayne@68: 			e.printStackTrace();
jpayne@68: 		}
jpayne@68: 		return null;
jpayne@68: 	}
jpayne@68: 	
jpayne@68: //	public static URL toUrl(String x){
jpayne@68: //		URL url=null;
jpayne@68: //		try {
jpayne@68: //			url = new URL(x);
jpayne@68: //		} catch (MalformedURLException e1) {
jpayne@68: //			// TODO Auto-generated catch block
jpayne@68: //			e1.printStackTrace();
jpayne@68: //		}
jpayne@68: //
jpayne@68: //		return url;
jpayne@68: //	}
jpayne@68: //
jpayne@68: //	public static URLConnection toUrlConnection(URL url, String x){
jpayne@68: //		URLConnection urlc=null;
jpayne@68: //
jpayne@68: //		try {
jpayne@68: //			urlc = url.openConnection();
jpayne@68: //		} catch (IOException e) {
jpayne@68: //			// TODO Auto-generated catch block
jpayne@68: //			e.printStackTrace();
jpayne@68: //		}
jpayne@68: //		return urlc;
jpayne@68: //	}
jpayne@68: 	
jpayne@68: }