/*
 * Copyright 1999-2006 University of Chicago
 *
 * Licensed 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 java.sql.ResultSet;

import java.util.HashMap;
import java.util.Map;
import java.util.Iterator;
import java.util.Date;
import java.util.Locale;

public class ByteReport{
    
    public static void main (String [] args) throws Exception{
	String USAGE = "Usage: java JobFlagReport [options] <date (YYYY-MM-DD)> Enter -help for a list of options\n";

        String HELP = "Where [options] are:\n"+
            " -help                 Displays help\n"+
            " -step <day|month>     Specifies size of step (day by default)\n"+
            " -n <steps>            Specifies number of steps to do\n";

	if (args.length == 0){
	    System.err.println(USAGE);
	    System.exit(1);
	}
	else if (args.length == 1 && args[0].equalsIgnoreCase("-help")){
	    System.err.println(USAGE);
	    System.err.println(HELP);
	    System.exit(1);
	}

	int n = 1;
        String stepStr = "day";

        for (int i=0;i<args.length-1;i++){
            if (args[i].equals("-n")) {
                n = Integer.parseInt(args[++i]);
            }
            else if (args[i].equals("-step")) {
                stepStr = args[++i];
            } else if (args[i].equalsIgnoreCase("-help")) {
                System.err.println(USAGE);
                System.err.println(HELP);
                System.exit(1);
	    } else {
                System.err.println("Unknown argument: " + args[i]);
                System.exit(1);
            }
        }

	String inputDate = args[args.length-1];

        TimeStep ts = new TimeStep (stepStr, n, inputDate);	

	HistogramParser byteHist = new HistogramParser("Total number of megabytes transfered", "GFTPbytehistogram", "Number of Bytes transfered by GFTP", n);

	HistogramParser bandwidthHist = new HistogramParser("Average Bandwidth in mb/s", "GFTPbandwidthhistogram", "Average Bandwidth of GFTP users (megabytes/sec)", n); 

	HistogramParser packetHist = new HistogramParser("Transfer Number", "GFTPpackethistogram", "Number of GFTP Transfers", n);

	DatabaseRetriever dbr = new DatabaseRetriever();

	long byteNumber;
	long packetNumber;
	double bandWidth;
	long timeDiff;

	while(ts.next()){
	    byteNumber = 0;
	    bandWidth = 0;
	    packetNumber=0;
	    timeDiff = 0;
 
	    Date startD = ts.getTime();
	    String startS = ts.getFormattedTime();
	    ts.stepTime();

	    bandwidthHist.nextEntry(startS, ts.getFormattedTime());
	    byteHist.nextEntry(startS, ts.getFormattedTime());
	    packetHist.nextEntry(startS, ts.getFormattedTime());

	    String startDate = ts.getFormattedTime();
	    ResultSet rs = dbr.retrieve("gftp_packets", new String [] {"num_bytes","start_time", "end_time"}, startD, ts.getTime());
	    while(rs.next()){
		packetNumber++;
		byteNumber+=rs.getLong(1);
		timeDiff = rs.getTimestamp(3).getTime() - rs.getTimestamp(2).getTime();
		if (timeDiff == 0){
		    timeDiff = 1;
		}
		bandWidth+=(double)rs.getLong(1)/(timeDiff/1000.0);
	    }
	    rs.close();
	    ResultSet rs2 = dbr.retrieve("mcs_internal_gftp_packets", new String[] {"num_bytes","start_time", "end_time"}, startD, ts.getTime());
	    while (rs2.next()){
		packetNumber++;
		byteNumber+=rs2.getLong(1);
		timeDiff = rs2.getTimestamp(3).getTime() - rs2.getTimestamp(2).getTime();
                if (timeDiff == 0){
                    timeDiff = 1;
                }
		bandWidth+=(double)rs2.getLong(1)/(timeDiff/1000.0);
	    }
	    byteHist.addData("MegaBytes", byteNumber/1048576.0);
	    packetHist.addData("Number of Transfers", packetNumber);
	    bandwidthHist.addData("Average Bandwidth (mb/s)",(bandWidth/packetNumber)/(1048576.0));
	    rs2.close();	    
	}
	dbr.close();
	System.out.println("<report>");
	packetHist.output(System.out);
	byteHist.output(System.out);
	bandwidthHist.output(System.out);
	System.out.println("</report>");
    }
}