通過Java程序抽取日志中的sql語句

今天在翻看以前的筆記時，發現自己在很早之前寫過一個java程序，能夠解析日志中的sql語句。當時使用的環境是weblogic,日志目錄下總是有幾十上百個日志文件，有時候排查問題的時候只需要找到對應的DML語句即可。

使用Linux命令固然也可以，但是解析的時候還是比較被動，不能夠正確地解析出sql語句來。比如日志中出現insert的字樣可能只是日志中的一段信息，不是insert語句。

這些通過Linux命令來完成還是有一定的難度，記得當時問題比較多，自己也飽受這種困擾。於是寫了一個java程序來。

代碼如下：

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;

public class LogToSqlMain {
private static String SELECT = "SELECT";
private static String UPDATE = "UPDATE";
private static String DELETE = "DELETE";
private static String INSERT = "INSERT";
private static String ALL = "ALL";

public static void main(String[] args) {
new LogToSqlMain().parse(args);
}

public void test(File logFile) {
// get file
// initialized io
// parse log to sql

// format sql
// generate sql file
// invoke jdbc
}

public void parse(String[] args) {
String args0 = null;
String args1 = null;
if (args == null) {
return;
}
if (args != null && args.length == 1) {
args0 = args[0];
}
if (args != null && args.length == 2) {
args0 = args[0];
args1 = args[1];
if (!args1.equalsIgnoreCase(ALL) && !args1.equalsIgnoreCase(SELECT)
&& !args1.equalsIgnoreCase(UPDATE) && !args1.equalsIgnoreCase(DELETE)
&& !args1.equalsIgnoreCase(INSERT)) {
return;
}
}

BufferedReader buffer_reader = null;
String sql_type = null;
try {
File file = new File(args0);
File[] filesOfDirs = file.listFiles();
if (!file.isDirectory() || filesOfDirs.length == 0) {
System.out.println("invalid path or io error");
return;
}
String temp_read = null;
String strTemp = null;
String strTimeStamp = null;
Long log_Line_Num = null;
for (int i = 0; i < filesOfDirs.length; i++) {
if (getSqlMode(args1, filesOfDirs[i].getName())) {
File tmp_File = filesOfDirs[i].getAbsoluteFile();
sql_type = getSQLType(tmp_File);
buffer_reader = new BufferedReader(new FileReader(tmp_File));
temp_read = buffer_reader.readLine();
while (temp_read != null) {
char sep_str = ':';
// validate every line should be SQL
if (validateFileLine(temp_read, sql_type)) {
log_Line_Num = Long.parseLong(temp_read.substring(
0, temp_read.indexOf(sep_str)));
strTemp = temp_read.substring(temp_read
.indexOf(':') + 1);
strTimeStamp = strTemp.substring(
strTemp.indexOf('[') + 1,
strTemp.indexOf(']'));
strTemp = strTemp
.substring(strTemp.indexOf(']') + 1);
String temp_Sql = strTemp.substring(strTemp
.indexOf(sql_type + " "));
System.out.println(sql_type + log_Line_Num + ","
+ strTimeStamp + "," + temp_Sql);
}
temp_read = buffer_reader.readLine();
}
buffer_reader.close();
}
}
} catch (NumberFormatException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
} finally {
if (buffer_reader != null) {
try {
buffer_reader.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}

}

private boolean validateFileLine(String str_Line, String sql_type) {
if (sql_type.equals(INSERT)) {
if (str_Line.toUpperCase().contains("INTO")
&& str_Line.toUpperCase().contains("VALUES"))
return true;
} else if (sql_type.equals(SELECT)) {
if (str_Line.toUpperCase().contains("FROM")
&& str_Line.toUpperCase().contains("WHERE")) {
return true;
}
} else if (sql_type.equals(UPDATE)) {
if (str_Line.toUpperCase().contains("SET")
&& str_Line.toUpperCase().contains("WHERE")) {
return true;
}
} else if (sql_type.equals(DELETE)) {
if (str_Line.toUpperCase().contains("WHERE")) {
return true;
}
}
return false;
}

private String getSQLType(File file) {
if (file.getName().toUpperCase().contains(SELECT)) {
return SELECT;
} else if (file.getName().toUpperCase().contains(UPDATE)) {
return UPDATE;
} else if (file.getName().toUpperCase().contains(DELETE)) {
return DELETE;
} else if (file.getName().toUpperCase().contains(INSERT)) {
return INSERT;
}
return null;
}

public boolean getSqlMode(String args, String fileName) {
if (args == null || ALL.equalsIgnoreCase(args))
return (fileName.startsWith(INSERT) || fileName.startsWith(DELETE)
|| fileName.startsWith(UPDATE) || fileName
.startsWith(SELECT));
if ((SELECT).equalsIgnoreCase(args)) {
return (fileName.startsWith(SELECT));
} else if ((UPDATE).equalsIgnoreCase(args)) {
return (fileName.startsWith(UPDATE));
} else if ((INSERT).equalsIgnoreCase(args)) {
return (fileName.startsWith(INSERT));
} else if ((DELETE).equalsIgnoreCase(args)) {
return (fileName.startsWith(DELETE));
}
return false;
}
}

如果需要得到a.log中的sql語句，
就可以這樣調用java程序 java LogToSqlMain a.log INSET
如果想得到所有的sql語句 java LogToSqlMain a.log ALL