package com;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;

import ch.ethz.ssh2.Connection;
import ch.ethz.ssh2.StreamGobbler;

public class MemoryChecker {

	public static void main(String[] args) throws Exception {
		runCheck();
	}
	public static void runCheck() throws Exception {
		String serverIp, command, usernameString, password ;

		 // 서버 ip
		 serverIp = "127.0.0.1";
		 // id
		 usernameString = "ubuntu";
		 // password
		 password = "password";
		 
		 
		 // 메모리 체크 명령어
		 command = "ps -Ao rss,pcpu,pmem,cmd --sort -rss";
		 
		 // 접속 객체 선언
		 Connection conn = null;
		 // 세션 선언
		 ch.ethz.ssh2.Session sess = null;
		 // 결과값을 읽어들일 버퍼
		 BufferedReader br = null;
		 
		 System.out.println("inside the ssh function");
	     try
	     {
	    	 
	    	 // ip 에 접속
	         conn = new Connection(serverIp);
	         conn.connect();
	         // 권한획득
	         boolean isAuthenticated = conn.authenticateWithPassword(usernameString, password);
	         if (isAuthenticated == false)
	        	 // 권한 획득 실패시 오류 print 후 종료
	             throw new IOException("Authentication failed.");
	         // 권한 획득 시 세션 획득
	         sess = conn.openSession();
	         // 명령어 전달
	         sess.execCommand(command);
	         // 결과 값 가져오기
	         InputStream stdout = new StreamGobbler(sess.getStdout());
	         // 결과 값 버퍼에 저장
	         br = new BufferedReader(new InputStreamReader(stdout));
	         System.out.println("the output of the command is");
	         List<MemoryVO> pList = new ArrayList<MemoryVO>();
	         
	         // 결과 값 VO에 삽입
	         // 머리행 제거, 메모리 미사용 프로세스 제거
	         while (true)
	         {
	             String line = br.readLine();
	             
	             
	             if (line == null) {break;}
	             MemoryVO mVO = new MemoryVO();
	             mVO = printStr(line);
	             if(mVO.getCmd() != null && !mVO.getCmd().substring(0,1).equals("R") && !mVO.getCmd().substring(0,1).equals("0")) { 
	            	 // 명령어 문자열 trim 처리
	            	 pList.add(printStr(line));
	             }

	         }

	         // VO 에 삽입 된 데이터 insert 
	         insertDB(pList);
	         System.out.println("ExitCode: " + sess.getExitStatus());

	     }
	     catch (IOException e)
	     {
	         e.printStackTrace(System.err);

	     }	
	     finally {
	         if(br != null) br.close();
	         if(sess != null) sess.close();
	         if(conn != null) conn.close(); 
	     }
	}
	

    // 명령어 trim 처리 후 객체에 담아 return
	public static MemoryVO printStr(String param) {
		
		MemoryVO rtnVO = new MemoryVO();
		
		String line = "";
		line = param.trim();
		
		rtnVO.setCmd(line);
		
        return rtnVO;
	}
	
	public static void insertDB(List<MemoryVO> param) throws Exception{
		
		// sql connection 선언
		java.sql.Connection con = null;
		// 쿼리 전달을 위한 statement
		PreparedStatement pstmt = null;
		
		Statement stmt = null;
		
		String insertSql = "INSERT INTO COM_24_SERVER_MEM_CHECK(RNO, CMD, CREATE_TIME ) "
				+ "VALUES ((SELECT NVL(MAX(RNO),0) + 1 FROM COM_24_SERVER_MEM_CHECK) , ? , SYSDATE )";

		
		try {
			Class.forName("oracle.jdbc.driver.OracleDriver");
			con = DriverManager.getConnection("jdbc:oracle:thin:@127.0.0.1:1521:mrstest","orcl","1234");
			stmt = con.createStatement();

			pstmt = con.prepareStatement(insertSql);
			
			int insertCnt = 0;
			for ( MemoryVO pVal : param) {
				
				pstmt.setString(1, pVal.getCmd());

				insertCnt ++;
				
				pstmt.addBatch();
				
				pstmt.clearParameters();
				
				if(  ( insertCnt % 10000) == 0  ) {
		    		pstmt.executeBatch();
		    		pstmt.clearBatch();
		    		con.commit();
		    	}
				
			}
			
			int[] result = pstmt.executeBatch();
			
			pstmt.clearBatch();
			con.commit();
			System.out.print(result.toString());
		} catch(Exception e) {
			e.printStackTrace();
		} finally {
			if(stmt != null ) { stmt.close(); }
			if(con  != null ) { con .close(); }
		}
		
	}
	
}