hr.zip_HR项目_core hr_hr_hr管理平台

package cn.xt.util; import java.io.BufferedReader; import java.io.DataInputStream; import java.io.DataOutputStream; import java.io.FileInputStream; import java.io.IOException; import java.io.InputStream; import java.io.InputStreamReader; import java.io.OutputStream; import java.io.OutputStreamWriter; import java.net.Socket; import java.sql.Connection; import java.sql.DatabaseMetaData; import java.sql.DriverManager; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.ResultSetMetaData; import java.sql.SQLException; import java.sql.Statement; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.Properties; import java.util.concurrent.Callable; import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ExecutionException; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import java.util.concurrent.Future; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; public class DbFH{ private static Log logger = LogFactory.getLog(DbFH.class); private static Properties pros = getPprVue(); public static Map<String, String> backUpTableList = new ConcurrentHashMap<String, String>(); public static Map<String, String> recoverTableList = new ConcurrentHashMap<String, String>(); private static DbFH dbFH = new DbFH(); public static void main(String[] arg){ try { String str = DbFH.getDbFH().backup("").toString();//璋冪敤鏁版嵁搴撳浠? System.out.println(FileUtil.getFilesize(str)); } catch (InterruptedException e) { e.printStackTrace(); } catch (ExecutionException e) { e.printStackTrace(); } } public static DbFH getDbFH(){ return dbFH; } /**鎵ц鏁版嵁搴撳浠藉叆鍙? * @param tableName 琛ㄥ悕 * @return * @throws InterruptedException * @throws ExecutionException */ public Object backup(String tableName) throws InterruptedException, ExecutionException { if(null != backUpTableList.get(tableName)) return null; backUpTableList.put(tableName, tableName); // 鏍囪宸茬粡鐢ㄤ簬澶囦唤(闃叉鍚屾椂閲嶅澶囦唤,姣斿澶囦唤涓?釜琛ㄧ殑绾跨▼姝ｅ湪杩愯锛屽張鍙戞潵涓?釜澶囦唤姝よ〃鐨勫懡浠? ExecutorService pool = Executors.newFixedThreadPool(2); Callable<Object> fhc = new DbBackUpCallable(tableName); //鍒涘缓涓?釜鏈夎繑鍥炲?鐨勭嚎绋? Future<Object> f1 = pool.submit(fhc); //鍚姩绾跨▼ String backstr = f1.get().toString(); //鑾峰彇绾跨▼鎵ц瀹屾瘯鐨勮繑鍥炲? pool.shutdown(); //鍏抽棴绾跨▼ return backstr; } /**鎵ц鏁版嵁搴撹繕鍘熷叆鍙? * @param tableName 琛ㄥ悕 * @param sqlFilePath 澶囦唤鏂囦欢瀛樻斁瀹屾暣璺緞 * @return * @throws InterruptedException * @throws ExecutionException */ public Object recover(String tableName,String sqlFilePath) throws InterruptedException, ExecutionException { if(null != recoverTableList.get(tableName)) return null; recoverTableList.put(tableName, tableName); // 鏍囪宸茬粡鐢ㄤ簬杩樺師(闃叉鍚屾椂閲嶅杩樺師,姣斿杩樺師涓?釜琛ㄧ殑绾跨▼姝ｅ湪杩愯锛屽張鍙戞潵涓?釜杩樺師姝よ〃鐨勫懡浠? ExecutorService pool = Executors.newFixedThreadPool(2); Callable<Object> fhc = new DbRecoverCallable(tableName,sqlFilePath); //鍒涘缓涓?釜鏈夎繑鍥炲?鐨勭嚎绋? Future<Object> f1 = pool.submit(fhc); //鍚姩绾跨▼ String backstr = f1.get().toString(); //鑾峰彇绾跨▼鎵ц瀹屾瘯鐨勮繑鍥炲? pool.shutdown(); //鍏抽棴绾跨▼ return backstr; } /**鑾峰彇鏈暟鎹簱鐨勬墍鏈夎〃鍚? * @return * @throws SQLException * @throws ClassNotFoundException */ public static Object[] getTables() throws ClassNotFoundException, SQLException{ String dbtype = pros.getProperty("dbtype"); //鏁版嵁搴撶被鍨? String username = pros.getProperty("username"); //鐢ㄦ埛鍚? String password = pros.getProperty("password"); //瀵嗙爜 String address = pros.getProperty("dbAddress"); //鏁版嵁搴撹繛鎺ュ湴鍧? String dbport = pros.getProperty("dbport"); //绔彛 String databaseName = pros.getProperty("databaseName"); //鏁版嵁搴撳悕 Connection conn = DbFH.getCon(dbtype,username,password,address+":"+dbport,databaseName); Object[] arrOb = {databaseName,DbFH.getTablesByCon(conn, "sqlserver".equals(dbtype)?null:databaseName),dbtype}; return arrOb; } /** * @return 鑾峰彇conn瀵硅薄 * @throws ClassNotFoundException * @throws SQLException */ public static Connection getFHCon() throws ClassNotFoundException, SQLException{ String dbtype = pros.getProperty("dbtype"); //鏁版嵁搴撶被鍨? String username = pros.getProperty("username"); //鐢ㄦ埛鍚? String password = pros.getProperty("password"); //瀵嗙爜 String address = pros.getProperty("dbAddress"); //鏁版嵁搴撹繛鎺ュ湴鍧? String dbport = pros.getProperty("dbport"); //绔彛 String databaseName = pros.getProperty("databaseName"); //鏁版嵁搴撳悕 return DbFH.getCon(dbtype,username,password,address+":"+dbport,databaseName); } /** * @param dbtype 鏁版嵁搴撶被鍨? * @param username 鐢ㄦ埛鍚? * @param password 瀵嗙爜 * @param dburl 鏁版嵁搴撹繛鎺ュ湴鍧?绔彛 * @param databaseName 鏁版嵁搴撳悕 * @return * @throws SQLException * @throws ClassNotFoundException */ public static Connection getCon(String dbtype,String username,String password,String dburl,String databaseName) throws SQLException, ClassNotFoundException{ if("mysql".equals(dbtype)){ Class.forName("com.mysql.jdbc.Driver"); return DriverManager.getConnection("jdbc:mysql://"+dburl+"/"+databaseName+"?user="+username+"&password="+password); }else if("oracle".equals(dbtype)){ Class.forName("oracle.jdbc.driver.OracleDriver"); return DriverManager.getConnection("jdbc:oracle:thin:@"+dburl+":"+username, databaseName, password); }else if("sqlserver".equals(dbtype)){ Class.forName("com.microsoft.sqlserver.jdbc.SQLServerDriver"); return DriverManager.getConnection("jdbc:sqlserver://"+dburl+"; DatabaseName="+databaseName, username, password); }else{ return null; } } /**鑾峰彇鏌愪釜conn涓嬬殑鎵?湁琛? * @param conn 鏁版嵁搴撹繛鎺ュ璞? * @param schema mysql:鏁版嵁搴撳悕; oracle:鐢ㄦ埛鍚?sqlserver:null * @return */ public static List<String> getTablesByCon(Connection conn, String schema) { try { List<String> listTb = new ArrayList<String>(); DatabaseMetaData meta = conn.getMetaData(); ResultSet rs = meta.getTables(null, schema, null, new String[] { "TABLE" }); while (rs.next()) { listTb.add(rs.getString(3)); } return listTb; } catch (Exception e) { e.printStackTrace(); } finally { try { conn.close(); } catch (SQLException e) { e.printStackTrace(); } } return null; } /**鐢ㄤ簬鎵ц鏌愯〃鐨勫浠?鍐呴儴绫?绾跨▼ * Callable 鏈夎繑鍥炲?鐨勭嚎绋嬫帴鍙? */ class DbBackUpCallable implements Callable<Object>{ String tableName = null; public DbBackUpCallable(String tableName){ this.tableName = tableName; } public Object call() { try { String remoteDB = pros.getProperty("remoteDB"); //鏄惁杩滅▼澶囦唤鏁版嵁搴?yes or no String DBSeverport = pros.getProperty("DBSeverport"); //杩滅▼鏈嶅姟鍣ㄥ浠界▼搴忕鍙? String dbtype = pros.getProperty("dbtype"); //鏁版嵁搴撶被鍨? String username = pros.getProperty("username"); //鐢ㄦ埛鍚? String password = pros.getProperty("password"); //瀵嗙爜 String address = pros.getProperty("dbAddress"); //鏁版嵁搴撹繛鎺ュ湴鍧? String databaseName = pros.getProperty("databaseName"); //鏁版嵁搴撳悕 String dbpath = pros.getProperty("dbpath"); //鏁版嵁搴撶殑瀹夎璺緞 String sqlpath = pros.getProperty("sqlFilePath"); //瀛樺偍璺緞 String ffilename = DateUtil.getSdfTimes(); String commandStr = ""; if(!"sqlserver".equals(dbtype))