Project update.
This commit is contained in:
MobiusDev
@@ -0,0 +1,214 @@
|
||||
/*
|
||||
* This file is part of the L2J Mobius project.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
* General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
package com.l2jmobius.tools.dbinstaller.util.mysql;
|
||||
|
||||
import java.io.File;
|
||||
import java.io.FileWriter;
|
||||
import java.sql.Connection;
|
||||
import java.sql.ResultSet;
|
||||
import java.sql.Statement;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Formatter;
|
||||
import java.util.GregorianCalendar;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import com.l2jmobius.tools.dbinstaller.DBOutputInterface;
|
||||
import com.l2jmobius.tools.dbinstaller.util.FileWriterStdout;
|
||||
|
||||
/**
|
||||
* @author mrTJO
|
||||
*/
|
||||
public class DBDumper
|
||||
{
|
||||
DBOutputInterface _frame;
|
||||
String _db;
|
||||
|
||||
public DBDumper(DBOutputInterface frame, String db)
|
||||
{
|
||||
_frame = frame;
|
||||
_db = db;
|
||||
createDump();
|
||||
}
|
||||
|
||||
public void createDump()
|
||||
{
|
||||
try (Formatter form = new Formatter())
|
||||
{
|
||||
final Connection con = _frame.getConnection();
|
||||
try (Statement s = con.createStatement();
|
||||
ResultSet rset = s.executeQuery("SHOW TABLES"))
|
||||
{
|
||||
final File dump = new File("dumps", form.format("%1$s_dump_%2$tY%2$tm%2$td-%2$tH%2$tM%2$tS.sql", _db, new GregorianCalendar().getTime()).toString());
|
||||
new File("dumps").mkdir();
|
||||
dump.createNewFile();
|
||||
|
||||
_frame.appendToProgressArea("Writing dump " + dump.getName());
|
||||
if (rset.last())
|
||||
{
|
||||
final int rows = rset.getRow();
|
||||
rset.beforeFirst();
|
||||
if (rows > 0)
|
||||
{
|
||||
_frame.setProgressIndeterminate(false);
|
||||
_frame.setProgressMaximum(rows);
|
||||
}
|
||||
}
|
||||
|
||||
try (FileWriter fileWriter = new FileWriter(dump);
|
||||
FileWriterStdout fws = new FileWriterStdout(fileWriter))
|
||||
{
|
||||
while (rset.next())
|
||||
{
|
||||
_frame.setProgressValue(rset.getRow());
|
||||
_frame.appendToProgressArea("Dumping Table " + rset.getString(1));
|
||||
fws.println("CREATE TABLE `" + rset.getString(1) + "`");
|
||||
fws.println("(");
|
||||
try (Statement desc = con.createStatement();
|
||||
ResultSet dset = desc.executeQuery("DESC " + rset.getString(1)))
|
||||
{
|
||||
final Map<String, List<String>> keys = new HashMap<>();
|
||||
boolean isFirst = true;
|
||||
while (dset.next())
|
||||
{
|
||||
if (!isFirst)
|
||||
{
|
||||
fws.println(",");
|
||||
}
|
||||
fws.print("\t`" + dset.getString(1) + "`");
|
||||
fws.print(" " + dset.getString(2));
|
||||
if (dset.getString(3).equals("NO"))
|
||||
{
|
||||
fws.print(" NOT NULL");
|
||||
}
|
||||
if (!dset.getString(4).isEmpty())
|
||||
{
|
||||
if (!keys.containsKey(dset.getString(4)))
|
||||
{
|
||||
keys.put(dset.getString(4), new ArrayList<String>());
|
||||
}
|
||||
keys.get(dset.getString(4)).add(dset.getString(1));
|
||||
}
|
||||
if (dset.getString(5) != null)
|
||||
{
|
||||
fws.print(" DEFAULT '" + dset.getString(5) + "'");
|
||||
}
|
||||
if (!dset.getString(6).isEmpty())
|
||||
{
|
||||
fws.print(" " + dset.getString(6));
|
||||
}
|
||||
isFirst = false;
|
||||
}
|
||||
if (keys.containsKey("PRI"))
|
||||
{
|
||||
fws.println(",");
|
||||
fws.print("\tPRIMARY KEY (");
|
||||
isFirst = true;
|
||||
for (String key : keys.get("PRI"))
|
||||
{
|
||||
if (!isFirst)
|
||||
{
|
||||
fws.print(", ");
|
||||
}
|
||||
fws.print("`" + key + "`");
|
||||
isFirst = false;
|
||||
}
|
||||
fws.print(")");
|
||||
}
|
||||
if (keys.containsKey("MUL"))
|
||||
{
|
||||
fws.println(",");
|
||||
isFirst = true;
|
||||
for (String key : keys.get("MUL"))
|
||||
{
|
||||
if (!isFirst)
|
||||
{
|
||||
fws.println(", ");
|
||||
}
|
||||
fws.print("\tKEY `key_" + key + "` (`" + key + "`)");
|
||||
isFirst = false;
|
||||
}
|
||||
}
|
||||
fws.println();
|
||||
fws.println(");");
|
||||
fws.flush();
|
||||
}
|
||||
|
||||
try (Statement desc = con.createStatement();
|
||||
ResultSet dset = desc.executeQuery("SELECT * FROM " + rset.getString(1)))
|
||||
{
|
||||
boolean isFirst = true;
|
||||
int cnt = 0;
|
||||
while (dset.next())
|
||||
{
|
||||
if ((cnt % 100) == 0)
|
||||
{
|
||||
fws.println("INSERT INTO `" + rset.getString(1) + "` VALUES ");
|
||||
}
|
||||
else
|
||||
{
|
||||
fws.println(",");
|
||||
}
|
||||
|
||||
fws.print("\t(");
|
||||
boolean isInFirst = true;
|
||||
for (int i = 1; i <= dset.getMetaData().getColumnCount(); i++)
|
||||
{
|
||||
if (!isInFirst)
|
||||
{
|
||||
fws.print(", ");
|
||||
}
|
||||
|
||||
if (dset.getString(i) == null)
|
||||
{
|
||||
fws.print("NULL");
|
||||
}
|
||||
else
|
||||
{
|
||||
fws.print("'" + dset.getString(i).replace("\'", "\\\'") + "'");
|
||||
}
|
||||
isInFirst = false;
|
||||
}
|
||||
fws.print(")");
|
||||
isFirst = false;
|
||||
|
||||
if ((cnt % 100) == 99)
|
||||
{
|
||||
fws.println(";");
|
||||
}
|
||||
cnt++;
|
||||
}
|
||||
if (!isFirst && ((cnt % 100) != 0))
|
||||
{
|
||||
fws.println(";");
|
||||
}
|
||||
fws.println();
|
||||
fws.flush();
|
||||
}
|
||||
}
|
||||
fws.flush();
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
e.printStackTrace();
|
||||
}
|
||||
_frame.appendToProgressArea("Dump Complete!");
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,112 @@
|
||||
/*
|
||||
* This file is part of the L2J Mobius project.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
* General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
package com.l2jmobius.tools.dbinstaller.util.mysql;
|
||||
|
||||
import java.sql.Connection;
|
||||
import java.sql.DriverManager;
|
||||
import java.sql.SQLException;
|
||||
import java.sql.Statement;
|
||||
import java.util.Formatter;
|
||||
|
||||
import javax.swing.JOptionPane;
|
||||
|
||||
/**
|
||||
* @author mrTJO
|
||||
*/
|
||||
public class MySqlConnect
|
||||
{
|
||||
Connection con = null;
|
||||
|
||||
public MySqlConnect(String host, String port, String user, String password, String db, boolean console)
|
||||
{
|
||||
try (Formatter form = new Formatter())
|
||||
{
|
||||
Class.forName("com.mysql.jdbc.Driver").newInstance();
|
||||
final String formattedText = form.format("jdbc:mysql://%1$s:%2$s", host, port).toString();
|
||||
con = DriverManager.getConnection(formattedText, user, password);
|
||||
|
||||
try (Statement s = con.createStatement())
|
||||
{
|
||||
s.execute("CREATE DATABASE IF NOT EXISTS `" + db + "`");
|
||||
s.execute("USE `" + db + "`");
|
||||
}
|
||||
}
|
||||
catch (SQLException e)
|
||||
{
|
||||
if (console)
|
||||
{
|
||||
e.printStackTrace();
|
||||
}
|
||||
else
|
||||
{
|
||||
JOptionPane.showMessageDialog(null, "MySQL Error: " + e.getMessage(), "Connection Error", JOptionPane.ERROR_MESSAGE);
|
||||
}
|
||||
}
|
||||
catch (InstantiationException e)
|
||||
{
|
||||
if (console)
|
||||
{
|
||||
e.printStackTrace();
|
||||
}
|
||||
else
|
||||
{
|
||||
JOptionPane.showMessageDialog(null, "Instantiation Exception: " + e.getMessage(), "Connection Error", JOptionPane.ERROR_MESSAGE);
|
||||
}
|
||||
}
|
||||
catch (IllegalAccessException e)
|
||||
{
|
||||
if (console)
|
||||
{
|
||||
e.printStackTrace();
|
||||
}
|
||||
else
|
||||
{
|
||||
JOptionPane.showMessageDialog(null, "Illegal Access: " + e.getMessage(), "Connection Error", JOptionPane.ERROR_MESSAGE);
|
||||
}
|
||||
}
|
||||
catch (ClassNotFoundException e)
|
||||
{
|
||||
if (console)
|
||||
{
|
||||
e.printStackTrace();
|
||||
}
|
||||
else
|
||||
{
|
||||
JOptionPane.showMessageDialog(null, "Cannot find MySQL Connector: " + e.getMessage(), "Connection Error", JOptionPane.ERROR_MESSAGE);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public Connection getConnection()
|
||||
{
|
||||
return con;
|
||||
}
|
||||
|
||||
public Statement getStatement()
|
||||
{
|
||||
try
|
||||
{
|
||||
return con.createStatement();
|
||||
}
|
||||
catch (SQLException e)
|
||||
{
|
||||
e.printStackTrace();
|
||||
System.out.println("Statement Null");
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,134 @@
|
||||
/*
|
||||
* This file is part of the L2J Mobius project.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
* General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
package com.l2jmobius.tools.dbinstaller.util.mysql;
|
||||
|
||||
import java.awt.HeadlessException;
|
||||
import java.io.File;
|
||||
import java.io.FileNotFoundException;
|
||||
import java.sql.Connection;
|
||||
import java.sql.SQLException;
|
||||
import java.sql.Statement;
|
||||
import java.util.Arrays;
|
||||
import java.util.Scanner;
|
||||
|
||||
import javax.swing.JOptionPane;
|
||||
|
||||
import com.l2jmobius.tools.dbinstaller.DBOutputInterface;
|
||||
import com.l2jmobius.util.file.filter.SQLFilter;
|
||||
|
||||
/**
|
||||
* @author mrTJO
|
||||
*/
|
||||
public class ScriptExecutor
|
||||
{
|
||||
DBOutputInterface _frame;
|
||||
|
||||
public ScriptExecutor(DBOutputInterface frame)
|
||||
{
|
||||
_frame = frame;
|
||||
}
|
||||
|
||||
public void execSqlBatch(File dir)
|
||||
{
|
||||
execSqlBatch(dir, false);
|
||||
}
|
||||
|
||||
public void execSqlBatch(File dir, boolean skipErrors)
|
||||
{
|
||||
final File[] files = dir.listFiles(new SQLFilter());
|
||||
if (files != null)
|
||||
{
|
||||
Arrays.sort(files);
|
||||
_frame.setProgressIndeterminate(false);
|
||||
_frame.setProgressMaximum(files.length - 1);
|
||||
for (int i = 0; i < files.length; i++)
|
||||
{
|
||||
_frame.setProgressValue(i);
|
||||
execSqlFile(files[i], skipErrors);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public void execSqlFile(File file)
|
||||
{
|
||||
execSqlFile(file, false);
|
||||
}
|
||||
|
||||
public void execSqlFile(File file, boolean skipErrors)
|
||||
{
|
||||
_frame.appendToProgressArea("Installing " + file.getName());
|
||||
String line = "";
|
||||
final Connection con = _frame.getConnection();
|
||||
try (Statement stmt = con.createStatement();
|
||||
Scanner scn = new Scanner(file))
|
||||
{
|
||||
StringBuilder sb = new StringBuilder();
|
||||
while (scn.hasNextLine())
|
||||
{
|
||||
line = scn.nextLine();
|
||||
if (line.startsWith("--"))
|
||||
{
|
||||
continue;
|
||||
}
|
||||
else if (line.contains("--"))
|
||||
{
|
||||
line = line.split("--")[0];
|
||||
}
|
||||
|
||||
line = line.trim();
|
||||
if (!line.isEmpty())
|
||||
{
|
||||
sb.append(line + System.getProperty("line.separator"));
|
||||
}
|
||||
|
||||
if (line.endsWith(";"))
|
||||
{
|
||||
stmt.execute(sb.toString());
|
||||
sb = new StringBuilder();
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (FileNotFoundException e)
|
||||
{
|
||||
JOptionPane.showMessageDialog(null, "File Not Found!: " + e.getMessage(), "Installer Error", JOptionPane.ERROR_MESSAGE);
|
||||
}
|
||||
catch (SQLException e)
|
||||
{
|
||||
if (!skipErrors)
|
||||
{
|
||||
try
|
||||
{
|
||||
final Object[] options =
|
||||
{
|
||||
"Continue",
|
||||
"Abort"
|
||||
};
|
||||
|
||||
final int n = JOptionPane.showOptionDialog(null, "MySQL Error: " + e.getMessage(), "Script Error", JOptionPane.YES_NO_OPTION, JOptionPane.WARNING_MESSAGE, null, options, options[0]);
|
||||
if (n == 1)
|
||||
{
|
||||
System.exit(0);
|
||||
}
|
||||
}
|
||||
catch (HeadlessException h)
|
||||
{
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user