rbatch 1.6.3 → 1.6.4

data/sample/moto/src/main/java/jp/ossc/imortAuditlog/MysqlController.java

@@ -0,0 +1,315 @@
+package jp.ossc.imortAuditlog;
+
+import java.net.InetAddress;
+import java.net.UnknownHostException;
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.SQLException;
+import java.sql.Timestamp;
+import java.util.Map;
+import java.util.PropertyResourceBundle;
+import java.util.ResourceBundle;
+
+import java.sql.PreparedStatement;
+
+import org.apache.log4j.Logger;
+
+public class MysqlController {
+	private static Logger logger = Logger.getLogger(MysqlController.class);
+
+	private static Connection con = null;
+	private static InetAddress addr = null;
+	private static PreparedStatement pstmtAuthBatch = null;
+	private static PreparedStatement pstmtAgentAuthBatch = null;
+	private static PreparedStatement pstmtBatch = null;
+	ResourceBundle sysResource = PropertyResourceBundle.getBundle("sysparam");
+
+	//処理件数
+	public int excuteAuthCount =0;
+	public int excuteAgentAuthCount =0;
+	public int excuteCount =0;
+
+	/*
+	 * MySqlへ接続を行う。
+	 */
+	public Boolean connect(){
+
+		if(con == null){
+			try {
+				// JDBCドライバの登録
+				String driver   = "com.mysql.jdbc.Driver";
+				// Host名
+				String server = sysResource.getString("mysql.host");
+				// port番号
+				String port = sysResource.getString("mysql.port");
+				// ユーザー
+				String user     = sysResource.getString("mysql.user");
+				// パスワード
+				String password = sysResource.getString("mysql.password");
+				// データベース名
+				String dbname =sysResource.getString("mysql.dbname");
+				// オプション
+				String option =sysResource.getString("mysql.option");
+				// データベースの指定
+				String url = "jdbc:mysql://" + server +":" + port + "/"+ dbname + "?" + option;
+				Class.forName (driver);
+				// データベースとの接続
+				con = DriverManager.getConnection(url, user, password);
+				con.setAutoCommit(false);
+			} catch (SQLException e) {
+				logger.error("dbconect error");
+				e.printStackTrace ();
+			} catch (ClassNotFoundException ex) {
+				logger.error("driver class not found");
+				ex.printStackTrace ();
+			}
+		}
+		return true;
+	}
+
+	/*
+	 * MySqlへの接続を切断する。
+	 */
+	public void disconnect() {
+		try {
+			con.close();
+			if (con != null)
+				con = null;
+		} catch (SQLException e) {
+			//logger.warn("", e);
+		}
+	}
+
+	/*
+	 * コミットを行う。
+	 */
+	public void commit() {
+		try {
+			con.commit();
+		} catch (SQLException e) {
+			//logger.warn("", e);
+		}
+	}
+
+	/*
+	 * 認証ログ登録SQLのパラメータをセットする。
+	 */
+	public boolean insertAuth(Map authLog){
+		try {
+			if(addr == null){
+				addr = InetAddress.getLocalHost();
+			}
+			// 各カラムに登録するデータをセット
+			int companyId = Integer.valueOf(sysResource.getString("companyid.value"));
+			String date = authLog.get("DATE").toString();
+			java.util.Date dateTime = DateUtil.toCalendar(date).getTime();
+			Timestamp sqlDate = new Timestamp(dateTime.getTime());
+			pstmtAuthBatch.setInt(1,companyId);
+			pstmtAuthBatch.setTimestamp(2, sqlDate);
+			pstmtAuthBatch.setString(3, authLog.get("USERID").toString());
+			pstmtAuthBatch.setString(4, authLog.get("CLIENTIP").toString());
+			pstmtAuthBatch.setString(5, addr.getHostName());
+			pstmtAuthBatch.setString(6, authLog.get("STATUS").toString());
+
+			pstmtAuthBatch.addBatch();
+		} catch (SQLException e) {
+			e.printStackTrace();
+			try {
+				con.rollback();
+			} catch (SQLException e1) {
+				e1.printStackTrace();
+			}
+		} catch (UnknownHostException e1) {
+			e1.printStackTrace();
+		}
+		return false;
+	}
+
+	/*
+	 * 認証ログ登録SQLをセットする。
+	 */
+	public void setAuthQuery(){
+		String query = "INSERT INTO authentication_log(companyId,date,login_id,access_ip,host_name,status) VALUES (?,?,?,?,?,?)";
+		try {
+			pstmtAuthBatch = con.prepareStatement(query);
+		} catch (SQLException e) {
+			try {
+				con.rollback();
+			} catch (SQLException e1) {
+				e1.printStackTrace();
+			}
+			e.printStackTrace();
+		}
+	}
+
+	/*
+	 * 認証ログ登録SQLを実行する。
+	 */
+	public boolean executeAuthBatch(){
+		// SQL文を発行し、結果を取得
+		try {
+			int[] numInsert = pstmtAuthBatch.executeBatch();
+			excuteAuthCount = excuteAuthCount + numInsert.length;
+		} catch (SQLException e) {
+			try {
+				con.rollback();
+			} catch (SQLException e1) {
+				e1.printStackTrace();
+			}
+			e.printStackTrace();
+		}
+		// データ作成が正常か判定
+		return true;
+	}
+
+	/*
+	 * アクセスログ登録SQLをセットする。
+	 */
+	public void setQuery(){
+		String query = "INSERT INTO access_log(companyId,date,login_id,access_ip,host_name,access_url,forward_time,httpstatus,user_agent,referer) VALUES (?,?,?,?,?,?,?,?,?,?)";
+		try {
+			pstmtBatch = con.prepareStatement(query);
+		} catch (SQLException e) {
+			try {
+				con.rollback();
+			} catch (SQLException e1) {
+				e1.printStackTrace();
+			}
+			e.printStackTrace();
+		}
+	}
+
+	/*
+	 * アクセスログ登録SQLのパラメータをセットする。
+	 */
+	public boolean insertAccessBatch(Map accessLog){
+		// ステートメント生成
+		try {
+			if(addr == null){
+				addr = InetAddress.getLocalHost();
+			}
+			// 各カラムに登録するデータをセット
+			int companyId = Integer.valueOf(sysResource.getString("companyid.value"));
+			String date = accessLog.get("DATE").toString();
+			java.util.Date dateTime = DateUtil.toCalendar(date).getTime();
+			Timestamp sqlDate = new Timestamp(dateTime.getTime());
+			pstmtBatch.setInt(1,companyId);
+			pstmtBatch.setObject(2, sqlDate);
+			pstmtBatch.setObject(3, accessLog.get("USERID"));
+			pstmtBatch.setObject(4, accessLog.get("CLIENTIP"));
+			pstmtBatch.setObject(5, addr.getHostName());
+			pstmtBatch.setObject(6, accessLog.get("REQUEST"));
+			pstmtBatch.setObject(7, accessLog.get("PROCESSINGTIME"));
+			pstmtBatch.setObject(8, accessLog.get("STATUS"));
+			pstmtBatch.setObject(9, accessLog.get("AGENT"));
+			pstmtBatch.setObject(10, accessLog.get("REFERER"));
+
+			pstmtBatch.addBatch();
+
+		} catch (SQLException e) {
+			try {
+				con.rollback();
+			} catch (SQLException e1) {
+				e1.printStackTrace();
+			}
+			e.printStackTrace();
+		} catch (UnknownHostException e1) {
+			e1.printStackTrace();
+		}
+		return false;
+	}
+
+	/*
+	 * アクセスログ登録SQLを実行する。
+	 */
+	public boolean executeBatch() throws SQLException{
+		// SQL文を発行し、結果を取得
+		try {
+			int [] numInsert = pstmtBatch.executeBatch();
+			excuteCount = excuteCount + numInsert.length;
+		} catch (SQLException e) {
+			try {
+				con.rollback();
+			} catch (SQLException e1) {
+				e1.printStackTrace();
+				throw e;
+			}
+			e.printStackTrace();
+			throw e;
+		}
+		// データ作成が正常か判定
+		return true;
+	}
+	
+	/*
+	 * 認可ログ登録SQLのパラメータをセットする。
+	 */
+	public boolean insertAgentAuth(Map agentAuthLog){
+		try {
+			if(addr == null){
+				addr = InetAddress.getLocalHost();
+			}
+			// 各カラムに登録するデータをセット
+			int companyId = Integer.valueOf(sysResource.getString("companyid.value"));
+			String date = agentAuthLog.get("DATE").toString();
+			java.util.Date dateTime = DateUtil.toCalendar(date).getTime();
+			Timestamp sqlDate = new Timestamp(dateTime.getTime());
+			pstmtAgentAuthBatch.setInt(1,companyId);
+			pstmtAgentAuthBatch.setTimestamp(2, sqlDate);
+			pstmtAgentAuthBatch.setString(3, agentAuthLog.get("USERID").toString());
+			pstmtAgentAuthBatch.setString(4, addr.getHostName());
+			pstmtAgentAuthBatch.setObject(5, agentAuthLog.get("REQUEST"));
+			pstmtAgentAuthBatch.setString(6, agentAuthLog.get("STATUS").toString());
+
+			pstmtAgentAuthBatch.addBatch();
+		} catch (SQLException e) {
+			e.printStackTrace();
+			try {
+				con.rollback();
+			} catch (SQLException e1) {
+				e1.printStackTrace();
+			}
+		} catch (UnknownHostException e1) {
+			e1.printStackTrace();
+		}
+		return false;
+	}
+	
+	/*
+	 * 認可ログ登録SQLをセットする。
+	 */
+	public void setAgentAuthQuery(){
+		String query = "INSERT INTO authorization_log(companyId,date,login_id,host_name,access_url,status) VALUES (?,?,?,?,?,?)";
+		try {
+			pstmtAgentAuthBatch = con.prepareStatement(query);
+		} catch (SQLException e) {
+			try {
+				con.rollback();
+			} catch (SQLException e1) {
+				e1.printStackTrace();
+			}
+			e.printStackTrace();
+		}
+	}
+	
+	/*
+	 * 認可ログ登録SQLを実行する。
+	 */
+	public boolean executeAgentAuthBatch(){
+		// SQL文を発行し、結果を取得
+		try {
+			int[] numInsert = pstmtAgentAuthBatch.executeBatch();
+			excuteAgentAuthCount = excuteAgentAuthCount + numInsert.length;
+		} catch (SQLException e) {
+			try {
+				con.rollback();
+			} catch (SQLException e1) {
+				e1.printStackTrace();
+			}
+			e.printStackTrace();
+		}
+		// データ作成が正常か判定
+		return true;
+	}
+
+}

data/sample/moto/src/main/logrotate/delete_auditlog

@@ -0,0 +1,6 @@
+/home/job/delete_auditlog/log/delete_auditlog.log {
+    copytruncate
+    rotate 31
+    daily
+    missingok
+}

data/sample/moto/src/main/logrotate/delete_workflowinstanceslog

@@ -0,0 +1,6 @@
+/root/delete_auditlog/log/delete_workflowinstances.log {
+    copytruncate
+    rotate 31
+    daily
+    missingok
+}

data/sample/moto/src/main/sql/tables.sql

@@ -0,0 +1,36 @@
+CREATE  TABLE IF NOT EXISTS `authentication_log` (
+	`id` int(10) NOT NULL AUTO_INCREMENT,
+	`companyId` bigint(20) NOT NULL,
+	`date` datetime NOT NULL,
+	`login_id` varchar(75) NOT NULL,
+	`access_ip` varchar(15) NOT NULL,
+	`host_name` varchar(75) NOT NULL,
+	`status` varchar(50) NOT NULL,
+	PRIMARY KEY (`id`)
+);
+
+CREATE  TABLE IF NOT EXISTS `access_log` (
+	`id` int(10) NOT NULL AUTO_INCREMENT,
+	`companyId` bigint(20) NOT NULL,
+	`date` datetime NOT NULL,
+	`login_id` varchar(75) NOT NULL,
+	`access_ip` varchar(15) NOT NULL,
+	`host_name` varchar(75) NOT NULL,
+	`access_url` varchar(2048) NOT NULL,
+	`forward_time` int(5) NOT NULL,
+	`httpstatus` int(3) NOT NULL,
+	`user_agent` varchar(512) NOT NULL,
+	`referer` varchar(2048),
+	PRIMARY KEY (`id`)
+);
+
+CREATE  TABLE IF NOT EXISTS `authorization_log` (
+	`id` int(10) NOT NULL AUTO_INCREMENT,
+	`companyId` bigint(20) NOT NULL,
+	`date` datetime NOT NULL,
+	`login_id` varchar(75) NOT NULL,
+	`host_name` varchar(75) NOT NULL,
+	`access_url` varchar(2048) NOT NULL,
+	`status` varchar(50) NOT NULL,
+	PRIMARY KEY (`id`)
+);

data/test/cases/test_cmd.rb

@@ -0,0 +1,120 @@
+require 'test/unit'
+require 'rbatch'
+
+class RuncherTest < Test::Unit::TestCase
+  def setup
+    @config_dir =  File.join(File.dirname(RBatch.program_name), "..", "conf")
+    Dir::mkdir(@config_dir) if ! Dir.exists? @config_dir
+    confstr = ""
+    open( RBatch.common_config_path  , "w" ){|f| f.write(confstr)}
+  end
+
+  def test_cmd_exists
+    result = RBatch::cmd "ruby -e 'STDOUT.print 1; STDERR.print 2; exit 0;'"
+    assert_equal "1", result.stdout.chomp
+    assert_equal "2", result.stderr.chomp
+    assert_equal 0, result.status
+  end
+  def test_cmd_does_not_exist
+    assert_raise(Errno::ENOENT){
+      RBatch::cmd "not_exist_command"
+    }
+  end
+  def test_stdout_size_greater_than_65534
+    result = RBatch::cmd "ruby -e '100000.times{print 0}'"
+    assert_equal 100000, result.stdout.chomp.size
+    assert_equal "", result.stderr.chomp
+    assert_equal 0, result.status
+  end
+  def test_stdout_size_greater_than_65534_with_status_1
+    result = RBatch::cmd "ruby -e '100000.times{print 0}; exit 1'"
+    assert_equal 100000, result.stdout.chomp.size
+    assert_equal "", result.stderr.chomp
+    assert_equal 1, result.status
+  end
+  def test_status_code_is_1
+    result = RBatch::cmd "ruby -e 'STDOUT.print 1; STDERR.print 2; exit 1;'"
+    assert_equal "1", result.stdout.chomp
+    assert_equal "2", result.stderr.chomp
+    assert_equal 1, result.status
+  end
+  def test_status_code_is_greater_than_256
+    returncode = 300
+    result = RBatch::cmd  "ruby -e 'STDOUT.print 1; STDERR.print 2; exit #{returncode};'"
+    assert_equal "1", result.stdout.chomp
+    assert_equal "2", result.stderr.chomp
+    case RUBY_PLATFORM
+    when /mswin|mingw/
+      assert_equal returncode, result.status
+    when /cygwin|linux/
+      assert_equal returncode % 256, result.status
+    end
+  end
+  def test_to_h
+    result = RBatch::cmd "ruby -e 'STDOUT.print 1; STDERR.print 2; exit 1;'"
+    assert_equal "1", result.to_h[:stdout]
+    assert_equal "2", result.to_h[:stderr]
+    assert_equal 1  , result.to_h[:status]
+  end
+  def test_to_s
+    result = RBatch::cmd "ruby -e 'STDOUT.print 1; STDERR.print 2; exit 1;'"
+    assert_equal "{:cmd_str=>\"ruby -e 'STDOUT.print 1; STDERR.print 2; exit 1;'\", :stdout=>\"1\", :stderr=>\"2\", :status=>1}", result.to_s
+  end
+  def test_nil_command
+    assert_raise(RBatch::CmdException){
+      RBatch::Cmd.new(nil)
+    }
+  end
+  def test_fullcmd_exists
+    result = RBatch::Cmd.new("ruby -e 'STDOUT.print 1; STDERR.print 2; exit 0;'").run
+    assert_equal "1", result.stdout.chomp
+    assert_equal "2", result.stderr.chomp
+    assert_equal 0, result.status
+  end
+  def test_opt_raise_true_status_1
+    cmd_str = "ruby -e 'STDOUT.print 1; STDERR.print 2; exit 1;'"
+    opt = {:raise => true}
+    assert_raise(RBatch::CmdException){
+      RBatch::Cmd.new(cmd_str,opt).run
+    }
+  end
+  def test_opt_raise_false_status_1
+    cmd_str = "ruby -e 'STDOUT.print 1; STDERR.print 2; exit 1;'"
+    opt = {:raise => false}
+    result = RBatch::Cmd.new(cmd_str,opt).run
+    assert_equal "1", result.stdout.chomp
+    assert_equal "2", result.stderr.chomp
+    assert_equal 1, result.status
+  end
+  def test_opt_raise_true_status_0
+    cmd_str = "ruby -e 'STDOUT.print 1; STDERR.print 2; exit 0;'"
+    opt = {:raise => true}
+    result = RBatch::Cmd.new(cmd_str,opt).run
+    assert_equal "1", result.stdout.chomp
+    assert_equal "2", result.stderr.chomp
+    assert_equal 0, result.status
+  end
+  def test_opt_raise_false_status_0
+    cmd_str = "ruby -e 'STDOUT.print 1; STDERR.print 2; exit 0;'"
+    opt = {:raise => false}
+    result = RBatch::Cmd.new(cmd_str,opt).run
+    assert_equal "1", result.stdout.chomp
+    assert_equal "2", result.stderr.chomp
+    assert_equal 0, result.status
+  end
+  def test_opt_raise_true_status_1_by_conf
+    confstr = "cmd_raise: true "
+    open( RBatch.common_config_path  , "a" ){|f| f.write(confstr)}
+    cmd_str = "ruby -e 'STDOUT.print 1; STDERR.print 2; exit 1;'"
+    assert_raise(RBatch::CmdException){
+      RBatch::Cmd.new(cmd_str).run
+    }
+  end
+  def test_opt_raise_true_status_1_shortcut
+    cmd_str = "ruby -e 'STDOUT.print 1; STDERR.print 2; exit 1;'"
+    opt = {:raise => true}
+    assert_raise(RBatch::CmdException){
+      RBatch::cmd(cmd_str,opt)
+    }
+  end
+end

data/test/cases/test_config.rb

@@ -0,0 +1,41 @@
+require 'test/unit'
+require 'rbatch'
+
+class RuncherTest < Test::Unit::TestCase
+  def setup
+    @config_dir =  File.join(File.dirname(RBatch.program_name), "..", "conf")
+    @config_file = File.join(@config_dir , "test_config.yaml")
+    Dir::mkdir(@config_dir) if ! Dir.exists? @config_dir
+  end
+
+  def teardown
+    File.delete @config_file if File.exist? @config_file
+  end
+
+  def test_require
+  end
+
+  def test_config
+    open( @config_file  , "w" ){|f| f.write("key: value")}
+    assert_equal "value",  RBatch.config["key"]
+  end
+
+  def test_read_error
+    assert_raise(Errno::ENOENT){
+      RBatch.config
+    }
+  end
+
+  def test_double_read
+    open( @config_file  , "w" ){|f| f.write("key: value")}
+    assert_equal "value",  RBatch.config["key"]
+    assert_equal "value",  RBatch.config["key"]
+  end
+
+  def test_not_exist_key
+    open( @config_file  , "w" ){|f| f.write("key: value")}
+    assert_raise(RBatch::Config::Exception){
+      RBatch.config["not_exist"]
+    }
+  end
+end