写了一批Lua脚本,要放到数据库里面,调用的时候进行计算,由于有太多lua脚本,就写了个程序来录入。
下面主要分三个部分:
public static String readToString(String fileName) public static void wirteToMYSQL(String string) throws FileNotFoundException public static List<String> getFileList(String strPath)
readToString,用作从lua脚本,也就是文件中读取script出来,传送一个参数就是文件夹的绝对路径。
writeToMYSQL主要进行数据库写的函数,main函数里面也就调用了这一个函数,也就是说这个函数调用了另外两个函数。
getFileList,对指定路径的文件夹进行遍历,找遍所有的lua脚本,并将lua脚本的绝对路径处理(加上\,因为java处理绝对路径需要加上)放到一个list里面。
writeToMYSQL本来是打算传个参数作为路径进去的,想想就用一次,就直接在里面写死了绝对路径。
上完整代码:
import java.io.File; import java.io.FileInputStream; import java.io.FileNotFoundException; import java.io.IOException; import java.io.UnsupportedEncodingException; import java.sql.Connection; import java.sql.DriverManager; import java.sql.SQLException; import java.sql.Statement; import java.util.LinkedList; import java.util.List; import com.mysql.jdbc.PreparedStatement; public class Test { public static void main(String[] args) throws IOException { wirteToMYSQL(""); } public static String readToString(String fileName) { String encoding = "UTF-8"; File file = new File(fileName); Long filelength = file.length(); byte[] filecontent = new byte[filelength.intValue()]; try { FileInputStream in = new FileInputStream(file); in.read(filecontent); in.close(); } catch (FileNotFoundException e) { e.printStackTrace(); } catch (IOException e) { e.printStackTrace(); } try { return new String(filecontent, encoding); } catch (UnsupportedEncodingException e) { System.err.println("The OS does not support " + encoding); e.printStackTrace(); return null; } } public static void wirteToMYSQL(String string) throws FileNotFoundException { final String JDBC_DRIVER = "com.mysql.jdbc.Driver"; final String DB_URL = "jdbc:mysql://192.168.**.***:3306/cqc"; final String USER = "***"; final String PASS = "*****"; Connection conn = null; Statement stmt = null; List<String> id = getFileList("C:\\Users\\yefeifan\\Desktop\\lua"); List<String> fileList = new LinkedList<>(); for (int i = 0; i < id.size(); i++) { fileList.add(id.get(i)); } // 得到 文件名list也就是id list C001002000_001 for (int i = 0; i < id.size(); i++) { id.set(i, id.get(i).substring(35, 49)); } // 得到script内容 List<String> scripts = new LinkedList<>(); for (int i = 0; i < fileList.size(); i++) { // System.out.println(readToString(fileList.get(i))); scripts.add(readToString(fileList.get(i))); } try { Class.forName("com.mysql.jdbc.Driver"); System.out.println("连接数据库..."); conn = DriverManager.getConnection(DB_URL, USER, PASS); PreparedStatement pstmt; // 执行查询 for (int i = 0; i < scripts.size(); i++) { String sql = "update capability set function_script=‘ " + scripts.get(i) + "‘ where id=‘" + id.get(i) + "‘"; pstmt = (PreparedStatement) conn.prepareStatement(sql); pstmt.executeUpdate(); pstmt.close(); } conn.close(); } catch (SQLException se) { // 处理 JDBC 错误 se.printStackTrace(); } catch (Exception e) { // 处理 Class.forName 错误 e.printStackTrace(); } finally { // 关闭资源 try { if (stmt != null) stmt.close(); } catch (SQLException se2) { } // 什么都不做 try { if (conn != null) conn.close(); } catch (SQLException se) { se.printStackTrace(); } } } public static List<String> getFileList(String strPath) { List<String> filelist = new LinkedList<String>(); File dir = new File(strPath); File[] files = dir.listFiles(); // 该文件目录下文件全部放入数组 if (files != null) { for (int i = 0; i < files.length; i++) { String fileName = files[i].getName(); if (files[i].isDirectory()) { // 判断是文件还是文件夹 return null; // 获取文件绝对路径 } else if (fileName.endsWith("lua")) { // 判断文件名是否以.avi结尾 StringBuffer strFileName = new StringBuffer(files[i].getAbsolutePath()); strFileName.insert(2, ‘\\‘); strFileName.insert(9, ‘\\‘); strFileName.insert(19, ‘\\‘); strFileName.insert(28, ‘\\‘); strFileName.insert(33, ‘\\‘); filelist.add(strFileName.toString()); } else { continue; } } } return filelist; } }