package logToMySQL;

import java.sql.*;

public class MySQL_Writer {
	static long newLog = 0;
	public void write(MessageSTL sqlMessage) {
		try {
			//加載驅動
			Class.forName("com.mysql.jdbc.Driver").newInstance();
            //建立連接
			Connection conn = DriverManager.getConnection("jdbc:mysql://localhost/syslog2idmef?user=root&password=password");
            Statement stmt = conn.createStatement();

            StringBuilder str = new StringBuilder();
            for (IDMEF i:IDMEF.values()) {
            	String attrib = sqlMessage.getMessage(i);
                str.append("'"+attrib+"'"+",");
            }
            str.deleteCharAt(str.length()-1);
            int i = stmt.executeUpdate("insert into idmef_message values("+str+");");
            System.out.println(""+newLog+":"+str);
            if (i != 0) newLog +=i;
        } catch (SQLException ex) {
        	System.out.println("SQLException: " + ex.getMessage());
            System.out.println("SQLState: " + ex.getSQLState());
            System.out.println("VendorError: " + ex.getErrorCode());
        } catch (Exception ex) {
        	ex.printStackTrace();
        }
	}
}