|
|
@@ -0,0 +1,245 @@
|
|
|
+package cn.com.qmth.stmms.ms.admin.api;
|
|
|
+
|
|
|
+import org.apache.commons.lang.StringUtils;
|
|
|
+import org.springframework.beans.factory.annotation.Autowired;
|
|
|
+import org.springframework.beans.factory.annotation.Value;
|
|
|
+import org.springframework.jdbc.core.JdbcTemplate;
|
|
|
+import org.springframework.web.bind.annotation.PostMapping;
|
|
|
+import org.springframework.web.bind.annotation.RequestMapping;
|
|
|
+import org.springframework.web.bind.annotation.RestController;
|
|
|
+
|
|
|
+import java.io.*;
|
|
|
+import java.sql.*;
|
|
|
+import java.util.regex.Matcher;
|
|
|
+import java.util.regex.Pattern;
|
|
|
+
|
|
|
+/**
|
|
|
+ * Created by yuanpan on 2017/11/17.
|
|
|
+ */
|
|
|
+@RestController
|
|
|
+@RequestMapping("api/snapshot")
|
|
|
+public class SnapshotController {
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ private JdbcTemplate jdbcTemplate;
|
|
|
+
|
|
|
+
|
|
|
+ @Value("${app.snapshot.fdsql}")
|
|
|
+ private String fdsql;
|
|
|
+
|
|
|
+ @Value("${app.snapshot.dfsql}")
|
|
|
+ private String dfsql;
|
|
|
+
|
|
|
+ @Value("${app.snapshot.fpsql}")
|
|
|
+ private String fpsql;
|
|
|
+
|
|
|
+
|
|
|
+ @PostMapping("/fendang")
|
|
|
+ public void fendang() {
|
|
|
+ execSql(fdsql);
|
|
|
+ }
|
|
|
+
|
|
|
+ @PostMapping("/dafen")
|
|
|
+ public void dafen() {
|
|
|
+ execSql(dfsql);
|
|
|
+ }
|
|
|
+
|
|
|
+ @PostMapping("/fuping")
|
|
|
+ public void fuping() {
|
|
|
+ execSql(fpsql);
|
|
|
+ }
|
|
|
+
|
|
|
+ private void execSql(String file) {
|
|
|
+ try {
|
|
|
+ PrintWriter logWriter = new PrintWriter(System.out);
|
|
|
+ PrintWriter errorLogWriter = new PrintWriter(System.err);
|
|
|
+
|
|
|
+ Connection conn = jdbcTemplate.getDataSource().getConnection();
|
|
|
+ SqlRunner sqlRunner = new SqlRunner(conn, logWriter, errorLogWriter, false, true);
|
|
|
+
|
|
|
+ sqlRunner.runScript(new FileReader(file));
|
|
|
+ } catch (SQLException e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ } catch (FileNotFoundException e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ }
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ private static class SqlRunner {
|
|
|
+ public static final String DELIMITER_LINE_REGEX = "(?i)DELIMITER.+", DELIMITER_LINE_SPLIT_REGEX = "(?i)DELIMITER", DEFAULT_DELIMITER = ";";
|
|
|
+ private final boolean autoCommit, stopOnError;
|
|
|
+ private final Connection connection;
|
|
|
+ private String delimiter = SqlRunner.DEFAULT_DELIMITER;
|
|
|
+ private final PrintWriter out, err;
|
|
|
+
|
|
|
+ public SqlRunner(final Connection connection, final PrintWriter out, final PrintWriter err, final boolean autoCommit, final boolean stopOnError) {
|
|
|
+ if (connection == null) {
|
|
|
+ throw new RuntimeException("SqlRunner requires an SQL Connection");
|
|
|
+ }
|
|
|
+ if (err == null || out == null) {
|
|
|
+ throw new RuntimeException("SqlRunner requires both out and err PrintWriters");
|
|
|
+ }
|
|
|
+ this.connection = connection;
|
|
|
+ this.autoCommit = autoCommit;
|
|
|
+ this.stopOnError = stopOnError;
|
|
|
+ this.out = out;
|
|
|
+ this.err = err;
|
|
|
+ }
|
|
|
+
|
|
|
+ public void runScript(final Reader reader) throws SQLException {
|
|
|
+ final boolean originalAutoCommit = this.connection.getAutoCommit();
|
|
|
+ try {
|
|
|
+ if (originalAutoCommit != this.autoCommit) {
|
|
|
+ this.connection.setAutoCommit(this.autoCommit);
|
|
|
+ }
|
|
|
+ this.runScript(this.connection, reader);
|
|
|
+ } finally {
|
|
|
+ this.connection.setAutoCommit(originalAutoCommit);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ private void runScript(final Connection conn, final Reader reader) {
|
|
|
+ StringBuffer command = null;
|
|
|
+ try {
|
|
|
+ final LineNumberReader lineReader = new LineNumberReader(reader);
|
|
|
+ String line = null;
|
|
|
+ while ((line = lineReader.readLine()) != null) {
|
|
|
+ if (command == null) {
|
|
|
+ command = new StringBuffer();
|
|
|
+ }
|
|
|
+ String trimmedLine = line.trim();
|
|
|
+
|
|
|
+ if (trimmedLine.startsWith("--") || trimmedLine.startsWith("//") || trimmedLine.startsWith("#")) {
|
|
|
+
|
|
|
+ // Line is a comment
|
|
|
+ out.println(trimmedLine);
|
|
|
+ out.flush();
|
|
|
+
|
|
|
+ } else if (trimmedLine.endsWith(this.delimiter)) {
|
|
|
+
|
|
|
+ // Line is end of statement
|
|
|
+
|
|
|
+ // Support new delimiter
|
|
|
+ final Pattern pattern = Pattern.compile(SqlRunner.DELIMITER_LINE_REGEX);
|
|
|
+ final Matcher matcher = pattern.matcher(trimmedLine);
|
|
|
+ if (matcher.matches()) {
|
|
|
+ delimiter = trimmedLine.split(SqlRunner.DELIMITER_LINE_SPLIT_REGEX)[1].trim();
|
|
|
+
|
|
|
+ // New delimiter is processed, continue on next
|
|
|
+ // statement
|
|
|
+ line = lineReader.readLine();
|
|
|
+ if (line == null) {
|
|
|
+ break;
|
|
|
+ }
|
|
|
+ trimmedLine = line.trim();
|
|
|
+ }
|
|
|
+
|
|
|
+ // Append
|
|
|
+ command.append(line.substring(0, line.lastIndexOf(this.delimiter)));
|
|
|
+ command.append(" ");
|
|
|
+
|
|
|
+ Statement stmt = null;
|
|
|
+ ResultSet rs = null;
|
|
|
+ try {
|
|
|
+ stmt = conn.createStatement();
|
|
|
+ out.println();
|
|
|
+ out.println(command);
|
|
|
+ out.flush();
|
|
|
+ boolean hasResults = false;
|
|
|
+ if (this.stopOnError) {
|
|
|
+ hasResults = stmt.execute(command.toString());
|
|
|
+ } else {
|
|
|
+ try {
|
|
|
+ stmt.execute(command.toString());
|
|
|
+ } catch (final SQLException e) {
|
|
|
+ e.fillInStackTrace();
|
|
|
+ err.println("Error on command: " + command);
|
|
|
+ err.println(e);
|
|
|
+ err.flush();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if (this.autoCommit && !conn.getAutoCommit()) {
|
|
|
+ conn.commit();
|
|
|
+ }
|
|
|
+ rs = stmt.getResultSet();
|
|
|
+ if (hasResults && rs != null) {
|
|
|
+
|
|
|
+ // Print result column names
|
|
|
+ final ResultSetMetaData md = rs.getMetaData();
|
|
|
+ final int cols = md.getColumnCount();
|
|
|
+ for (int i = 0; i < cols; i++) {
|
|
|
+ final String name = md.getColumnLabel(i + 1);
|
|
|
+ out.print(name + "\t");
|
|
|
+ }
|
|
|
+ out.println("");
|
|
|
+ out.println(StringUtils.repeat("---------", md.getColumnCount()));
|
|
|
+ out.flush();
|
|
|
+
|
|
|
+ // Print result rows
|
|
|
+ while (rs.next()) {
|
|
|
+ for (int i = 1; i <= cols; i++) {
|
|
|
+ final String value = rs.getString(i);
|
|
|
+ out.print(value + "\t");
|
|
|
+ }
|
|
|
+ out.println("");
|
|
|
+ }
|
|
|
+ out.flush();
|
|
|
+ } else {
|
|
|
+ out.println("Updated: " + stmt.getUpdateCount());
|
|
|
+ out.flush();
|
|
|
+ }
|
|
|
+ command = null;
|
|
|
+ } finally {
|
|
|
+ if (rs != null)
|
|
|
+ try {
|
|
|
+ rs.close();
|
|
|
+ } catch (final Exception e) {
|
|
|
+ err.println("Failed to close result: " + e.getMessage());
|
|
|
+ err.flush();
|
|
|
+ }
|
|
|
+ if (stmt != null)
|
|
|
+ try {
|
|
|
+ stmt.close();
|
|
|
+ } catch (final Exception e) {
|
|
|
+ err.println("Failed to close statement: " + e.getMessage());
|
|
|
+ err.flush();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+
|
|
|
+ // Line is middle of a statement
|
|
|
+
|
|
|
+ // Support new delimiter
|
|
|
+ final Pattern pattern = Pattern.compile(SqlRunner.DELIMITER_LINE_REGEX);
|
|
|
+ final Matcher matcher = pattern.matcher(trimmedLine);
|
|
|
+ if (matcher.matches()) {
|
|
|
+ delimiter = trimmedLine.split(SqlRunner.DELIMITER_LINE_SPLIT_REGEX)[1].trim();
|
|
|
+ line = lineReader.readLine();
|
|
|
+ if (line == null) {
|
|
|
+ break;
|
|
|
+ }
|
|
|
+ trimmedLine = line.trim();
|
|
|
+ }
|
|
|
+ command.append(line);
|
|
|
+ command.append(" ");
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if (!this.autoCommit) {
|
|
|
+ conn.commit();
|
|
|
+ }
|
|
|
+ } catch (final SQLException e) {
|
|
|
+ e.fillInStackTrace();
|
|
|
+ err.println("Error on command: " + command);
|
|
|
+ err.println(e);
|
|
|
+ err.flush();
|
|
|
+ } catch (final IOException e) {
|
|
|
+ e.fillInStackTrace();
|
|
|
+ err.println("Error on command: " + command);
|
|
|
+ err.println(e);
|
|
|
+ err.flush();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
YuanPan