package org.com.gather;

/**
*  author Adam
*  yangjh@tarena.com
*/

import java.util.*;
import java.io.*;
import java.sql.Timestamp;
import java.net.InetAddress;

public class GatherImpl implements Serializable,Gather {
  private Properties pro;
  //存放記錄為7的集合!
  private ArrayList list = new ArrayList();
  private RandomAccessFile raf;
  private Log log = null;

  public GatherImpl(){}

  public GatherImpl(Properties pro) {
    this.pro = pro;
  }

  public Collection getData(){
    ArrayList arrayList = new ArrayList();
    ArrayList aList = new ArrayList();

	//先看一下上次是否有客戶還沒有下線,如有,則把
	//該信息載入內存,繼續比較!
    if(this.loadTempFile()!=null){
		
      list = loadTempFile();
    }
    try{
	//獲得上次標志位!
	  long point = this.getPoint();
	  System.out.println("當前指針是--- "+point);
		
	//獲取解析文件路徑!
      String wtmpxPath = pro.getProperty("wtmpx_path");
	  

      File file = new File(wtmpxPath);

	//為了防止出現找不到文件的異常,用file.createNewFile();
	//來創建!
      if (file.createNewFile()) {
        System.out.println("wtmpx file is first creating");
      }

	  //用RandomAccessFile對File進行包裝!
      raf = new RandomAccessFile(file, "r");

	//將指針移到上次的標志位!
      raf.seek(point);

	//布爾值為真,直接進入循環體!
      boolean run = true;

	//循環讀取該文件
      while(run){
        String data = raf.readLine();
		//如果讀到空行就退出!
        if(data == null){
          run = false;
          break;
        }
		
		//先檢驗該記錄是否為有效記錄,如果無效則不做
		//任何處理!
          if (validate(data)) {
            TempBIDR obj = this.parseWtmpx(data);
			//先判斷一下該狀態位置是否為空!
            if (obj.getStatus() != null) {
              if (obj.getStatus().equals("7")) {
				//如果讀到的記錄狀態位為7的,就把該記錄
				//一起放到記錄為7的集合里!
                list.add(obj);
              }
			  //如果讀到的記錄狀態位為8,則直接和記錄為7
			  //的集合進行匹配,如果有匹配的則進行合并!
              else if (obj.getStatus().equals("8")) {
                Iterator iter = list.iterator();
                while (iter.hasNext()) {
				//把記錄放到一個另一個的TempBIDR對象里!
                  TempBIDR temp = (TempBIDR) iter.next();
				 
				  //如果線程號相同,則認為是同一個用戶!
                  if (obj.getThreadID().equals(temp.getThreadID()) && obj.getUserName().equals(obj.getUserName())) {

					//得到登陸時間和登出時間,然后登出時間
					//減去登陸時間,得出上網時長!
                    long loginTime = Long.parseLong(temp.getTime());
                    long logoutTime = Long.parseLong(obj.getTime());
					
                    long timeDuration = logoutTime - loginTime;

                    BIDR bidr = new BIDR();

					bidr.setLoginName(obj.getUserName());
					bidr.setLoginIP(obj.getLocalIP());
					bidr.setLoginDate(new Timestamp(loginTime * 1000));
					bidr.setLogoutDate(new Timestamp(logoutTime * 1000));
					bidr.setLabIP(InetAddress.getLocalHost().
                                         getHostAddress());
					bidr.setTimeDuration(timeDuration * 1000);
					//把成功合并的記錄存放放到集合!
                    arrayList.add(bidr);
					bidr = null;
					//把已經成功配對的7記錄從集合里刪除!
                    iter.remove();
                  }
                }
              }
            }
          }
      }
	 
	  //把狀態為7的記錄存放到臨時文件里,下次再比較!
      this.storeTempFile(list);
	  //獲取文件尾的指針,保存起來,做為下次讀文件的標志為位!
      long filePoint = raf.getFilePointer();

	  //把當前的指針位置保存起來!
      this.storePoint(filePoint);

//該方法是對讀出來的集合再一次過濾,把交叉時間給屏蔽調!
//      aList = this.parseArrayList(arrayList);

	  log.writeDebug("采集成功!");
	  if(arrayList.isEmpty()){
		System.out.println("=="+arrayList);
	  }
	  
      return arrayList;

    }catch(Exception e){
      e.printStackTrace();
      return null;
    }
  }

/**
	對幾個進行過濾,如果出現"變態"上網,那么應該把
	出現沖突的時間過濾掉.
*/
  public ArrayList parseArrayList(ArrayList aList){
    ArrayList list = new ArrayList();
    list = aList;
    for(int i = 0; i < list.size(); i++){
      for(int j = i+1; j < list.size(); j++){
        if(!(list.get(i)==null) && !(list.get(j)==null)){
          BIDR bidr_I = (BIDR) list.get(i);
          BIDR bidr_J = (BIDR) list.get(j);
          //如果兩個對象的名字一樣,而且有交錯時間的話就整合
          //把i 改變掉,然后與他比較的那個對象就刪除.
          if(bidr_I.getLoginName().equals(bidr_J.getLoginName())&&testBIDR(bidr_I,bidr_J)){
            list.set(i,this.combineBIDR(bidr_I,bidr_J));

            list.set(j,null);
          }
        }
      }
    }

    Iterator iter = list.iterator();
    while(iter.hasNext()){
      if(iter.next()==null){
        iter.remove();
      }
    }

    if(list==null){
      list = aList;
    }
    return list;
  }

/**
	對出現有交叉時間的JavaBean進行合并,取最短的和最長的上網
	時間
*/
  public BIDR combineBIDR(BIDR bidr_I,BIDR bidr_J){
    long s1Login_date = bidr_I.getLoginDate().getTime();
    long s2Login_date = bidr_J.getLoginDate().getTime();

    long s1Logout_date = bidr_I.getLogoutDate().getTime();
    long s2Logout_date = bidr_J.getLogoutDate().getTime();

    long login_date = 0l;
    long logout_date = 0l;

    login_date = (s1Login_date <= s2Login_date ? s1Login_date : s2Login_date);
    logout_date = (s1Logout_date >= s2Logout_date ? s1Logout_date : s2Logout_date);

    long timeduration = logout_date - login_date;

    Timestamp login_Date = new Timestamp(login_date);
    Timestamp logout_Date = new Timestamp(logout_date);

	BIDR bidr = new BIDR();

	bidr.setLoginName(bidr_I.getLoginName());
	bidr.setLoginIP(bidr_I.getLoginIP());
	bidr.setLoginDate(login_Date);
	bidr.setLogoutDate(logout_Date);
	bidr.setLabIP(bidr_I.getLabIP());
	bidr.setTimeDuration(timeduration);

    return bidr;
  }


/**
	測試該JavaBean的上網時間是否有交叉,如果有交叉
	則返回true,否則返回false
*/
  public boolean testBIDR(BIDR bidr_I,BIDR bidr_J){
    boolean status = false;

    long ILogin_date = bidr_I.getLoginDate().getTime();
    long JLogin_date = bidr_J.getLoginDate().getTime();

    long ILogout_date = bidr_I.getLogoutDate().getTime();
    long JLogout_date = bidr_J.getLogoutDate().getTime();

    if(ILogin_date >= JLogin_date && ILogin_date <= JLogout_date ||
       JLogin_date >= ILogin_date && JLogin_date <= ILogout_date){
      status = true;
    }
    return status;
  }


/*
	保存指針
*/
  public void storePoint(long point){
    String data = Long.toString(point);
    try{
		OutputStream os = new FileOutputStream(pro.getProperty("point_file"));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(os));
		bw.write(data);
		bw.flush();
		os.close();
		bw.close();
    }catch(Exception e){
      e.printStackTrace();
    }
  }

/**
	把有7沒8的暫時保存起來
*/
  public void storeTempFile(ArrayList list){
    File file = new File(pro.getProperty("log_backup"));
    try{
      OutputStream os = new FileOutputStream(file);
      ObjectOutputStream oos = new ObjectOutputStream(os);
      oos.writeObject(list);
    }catch(Exception e){
      e.printStackTrace();
    }
  }

/**
	對讀到的數據進行解析,封裝成JavaBean
*/
  public TempBIDR parseWtmpx(String data){
    StringTokenizer st = new StringTokenizer(data,":");
    String[]record = new String[11];
    int i = 0;
    while(st.hasMoreTokens()){
      record[i] = st.nextToken();
      i++;
    }
    TempBIDR tempBIDR = new TempBIDR(record[0], record[3], record[4],
                                       record[5], record[8]);

    return tempBIDR;
  }


/**
	驗證數據的合法性,刷選一部分不合法數據
*/
  public boolean validate(String data){
    boolean status = true;
    if(data!=null){
      StringTokenizer st = new StringTokenizer(data, ":");
      String da = null;

      while (st.hasMoreTokens()) {
        da = st.nextToken();
        break;
      }
      if (da == null) {
        status = false;
      }
      else {
        String[] stringSet = {
            "root", "oracle","dba","LOGIN", "system boot", "run-level", "rc2", "rc3", "sac",
            "ttymon", "zsmon", "shutdown"};

        for (int i = 0; i < stringSet.length; i++) {
          if (da != null) {
            if (da.equals(stringSet[i])) {
              status = false;
              break;
            }
          }
        }
      }
    }
    return status;
  }


/**
	獲取上次文件讀到的指針
*/
  public long getPoint(){
    long point = 0l;
    File file = new File(pro.getProperty("point_file"));
    try{
      if (file.createNewFile()) {}
      if(file.length()>0){
        InputStream is = new FileInputStream(file);
        BufferedReader br = new BufferedReader(new InputStreamReader(is));

        String p = br.readLine();
        point = Long.parseLong(p);
        br.close();
      }
      return point;
    }catch(Exception e){
      e.printStackTrace();
      return 0;
    }
  }

/**
	載入上次還沒下線的用戶信息
*/
  public ArrayList loadTempFile(){
    File file = new File(pro.getProperty("log_backup"));
    ArrayList list =  null;
    try{
      if (file.createNewFile()) {
        
      }
	  //如果不判斷一下,會出現異常,因為如果該文件長度為零的話,
	  //會返回一個XX異常.
      if(file.length()>0){
        InputStream is = new FileInputStream(file);
        ObjectInputStream ois = new ObjectInputStream(is);
        list = (ArrayList) ois.readObject();
		is.close();
		ois.close();
      }
      return list;
    }catch(Exception e){
      e.printStackTrace();
      return null;
    }
  }

/**
	提供依賴注入的入口
*/
  public void setLog(Log log){
		this.log = log;
  }

}