mirror of
https://github.com/ClassiCube/MCGalaxy.git
synced 2025-09-25 22:30:52 -04:00
Use much faster method of backing up sql tables (still need to fix dates though)
This commit is contained in:
UnknownShadow200
parent
413990e3ca
commit
9be0b0691d
@ -29,62 +29,70 @@ namespace MCGalaxy.SQL {
|
|||||||
}
|
}
|
||||||
|
|
||||||
[Obsolete("Use Execute() method instead.")]
|
[Obsolete("Use Execute() method instead.")]
|
||||||
public static void executeQuery(string queryString, bool createDB = false) {
|
public static void executeQuery(string sql, bool createDB = false) {
|
||||||
ParameterisedQuery query = Backend.GetStaticParameterised();
|
ParameterisedQuery query = Backend.GetStaticParameterised();
|
||||||
Execute(query, queryString, createDB, null);
|
Execute(query, sql, createDB, null);
|
||||||
}
|
}
|
||||||
|
|
||||||
public static void Execute(string queryString) {
|
public static void Execute(string sql) {
|
||||||
ParameterisedQuery query = Backend.GetStaticParameterised();
|
ParameterisedQuery query = Backend.GetStaticParameterised();
|
||||||
Execute(query, queryString, false, null);
|
Execute(query, sql, false, null);
|
||||||
}
|
}
|
||||||
|
|
||||||
public static void Execute(string queryString, params object[] args) {
|
public static void Execute(string sql, params object[] args) {
|
||||||
ParameterisedQuery query = Backend.CreateParameterised();
|
ParameterisedQuery query = Backend.CreateParameterised();
|
||||||
Execute(query, queryString, false, args);
|
Execute(query, sql, false, args);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void ExecuteReader(string sql, ReaderCallback callback, params object[] args) {
|
||||||
|
ParameterisedQuery query = Backend.CreateParameterised();
|
||||||
|
DoDatabaseCall(query, sql, false, null, callback, args);
|
||||||
}
|
}
|
||||||
|
|
||||||
[Obsolete("Use Fill() method instead.")]
|
[Obsolete("Use Fill() method instead.")]
|
||||||
public static DataTable fillData(string queryString, bool skipError = false) {
|
public static DataTable fillData(string sql, bool skipError = false) {
|
||||||
ParameterisedQuery query = Backend.GetStaticParameterised();
|
ParameterisedQuery query = Backend.GetStaticParameterised();
|
||||||
return Fill(query, queryString, null);
|
return Fill(query, sql, null);
|
||||||
}
|
}
|
||||||
|
|
||||||
public static DataTable Fill(string queryString) {
|
public static DataTable Fill(string sql) {
|
||||||
ParameterisedQuery query = Backend.GetStaticParameterised();
|
ParameterisedQuery query = Backend.GetStaticParameterised();
|
||||||
return Fill(query, queryString, null);
|
return Fill(query, sql, null);
|
||||||
}
|
}
|
||||||
|
|
||||||
public static DataTable Fill(string queryString, params object[] args) {
|
public static DataTable Fill(string sql, params object[] args) {
|
||||||
ParameterisedQuery query = Backend.CreateParameterised();
|
ParameterisedQuery query = Backend.CreateParameterised();
|
||||||
return Fill(query, queryString, args);
|
return Fill(query, sql, args);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
internal static void Execute(ParameterisedQuery query, string sql, bool createDB, params object[] args) {
|
internal static void Execute(ParameterisedQuery query, string sql, bool createDB, params object[] args) {
|
||||||
DoDatabaseCall(query, sql, createDB, null, args);
|
DoDatabaseCall(query, sql, createDB, null, null, args);
|
||||||
}
|
}
|
||||||
|
|
||||||
internal static DataTable Fill(ParameterisedQuery query, string sql, params object[] args) {
|
internal static DataTable Fill(ParameterisedQuery query, string sql, params object[] args) {
|
||||||
using (DataTable results = new DataTable("toReturn")) {
|
using (DataTable results = new DataTable("toReturn")) {
|
||||||
DoDatabaseCall(query, sql, false, results, args);
|
DoDatabaseCall(query, sql, false, results, null, args);
|
||||||
return results;
|
return results;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static void DoDatabaseCall(ParameterisedQuery query, string sql, bool createDB,
|
static void DoDatabaseCall(ParameterisedQuery query, string sql, bool createDB,
|
||||||
DataTable results, params object[] args) {
|
DataTable results, ReaderCallback callback, params object[] args) {
|
||||||
BindParams(query, args);
|
BindParams(query, args);
|
||||||
string connString = Backend.ConnectionString;
|
string connString = Backend.ConnectionString;
|
||||||
Exception e = null;
|
Exception e = null;
|
||||||
|
|
||||||
for (int i = 0; i < 10; i++) {
|
for (int i = 0; i < 10; i++) {
|
||||||
try {
|
try {
|
||||||
if (results == null) {
|
if (callback != null) {
|
||||||
|
query.ExecuteReader(sql, connString, callback);
|
||||||
|
} else if (results == null) {
|
||||||
query.Execute(sql, connString, createDB);
|
query.Execute(sql, connString, createDB);
|
||||||
} else {
|
} else {
|
||||||
query.Fill(sql, connString, results);
|
query.Fill(sql, connString, results);
|
||||||
}
|
}
|
||||||
|
|
||||||
query.ClearParams();
|
query.ClearParams();
|
||||||
return;
|
return;
|
||||||
} catch (Exception ex) {
|
} catch (Exception ex) {
|
||||||
|
|||||||
@ -22,7 +22,7 @@ using System.Data.Common;
|
|||||||
|
|
||||||
namespace MCGalaxy.SQL {
|
namespace MCGalaxy.SQL {
|
||||||
|
|
||||||
public delegate void ReaderCallback(DataTable schema, IDataReader reader);
|
public delegate void ReaderCallback(IDataReader reader);
|
||||||
|
|
||||||
public abstract class ParameterisedQuery {
|
public abstract class ParameterisedQuery {
|
||||||
|
|
||||||
@ -77,7 +77,7 @@ namespace MCGalaxy.SQL {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public void ExecuteReader(string query, string connString) {
|
public void ExecuteReader(string query, string connString, ReaderCallback callback) {
|
||||||
using (IDbConnection conn = CreateConnection(connString)) {
|
using (IDbConnection conn = CreateConnection(connString)) {
|
||||||
conn.Open();
|
conn.Open();
|
||||||
if (MultipleSchema)
|
if (MultipleSchema)
|
||||||
@ -86,9 +86,7 @@ namespace MCGalaxy.SQL {
|
|||||||
using (IDbCommand cmd = CreateCommand(query, conn)) {
|
using (IDbCommand cmd = CreateCommand(query, conn)) {
|
||||||
FillParams(cmd);
|
FillParams(cmd);
|
||||||
using (IDataReader reader = cmd.ExecuteReader()) {
|
using (IDataReader reader = cmd.ExecuteReader()) {
|
||||||
while (reader.Read()) {
|
while (reader.Read()) { callback(reader); }
|
||||||
// TODO: do callback here
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
conn.Close();
|
conn.Close();
|
||||||
@ -103,6 +101,5 @@ namespace MCGalaxy.SQL {
|
|||||||
cmd.Parameters.Add(dParam);
|
cmd.Parameters.Add(dParam);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@ -595,8 +595,9 @@
|
|||||||
<Compile Include="Database\Database.cs" />
|
<Compile Include="Database\Database.cs" />
|
||||||
<Compile Include="Games\CTF\CtfGame.cs" />
|
<Compile Include="Games\CTF\CtfGame.cs" />
|
||||||
<Compile Include="Levels\BlockQueue.cs" />
|
<Compile Include="Levels\BlockQueue.cs" />
|
||||||
<Compile Include="Server\Backup.cs" />
|
<Compile Include="Server\Backup\Backup.cs" />
|
||||||
<Compile Include="Server\BackupDB.cs" />
|
<Compile Include="Server\Backup\BackupDB.cs" />
|
||||||
|
<Compile Include="Server\Backup\TableDumper.cs" />
|
||||||
<Compile Include="Server\Extra\UPnP.cs" />
|
<Compile Include="Server\Extra\UPnP.cs" />
|
||||||
<Compile Include="Network\IBeat.cs" />
|
<Compile Include="Network\IBeat.cs" />
|
||||||
<Compile Include="Levels\Level.cs" />
|
<Compile Include="Levels\Level.cs" />
|
||||||
@ -738,6 +739,7 @@
|
|||||||
<Folder Include="CorePlugin" />
|
<Folder Include="CorePlugin" />
|
||||||
<Folder Include="Levels\IO\Importers" />
|
<Folder Include="Levels\IO\Importers" />
|
||||||
<Folder Include="Levels\IO\Exporters" />
|
<Folder Include="Levels\IO\Exporters" />
|
||||||
|
<Folder Include="Server\Backup" />
|
||||||
<Folder Include="Server\Tasks" />
|
<Folder Include="Server\Tasks" />
|
||||||
<Folder Include="util\Math" />
|
<Folder Include="util\Math" />
|
||||||
<Folder Include="Player\Group" />
|
<Folder Include="Player\Group" />
|
||||||
|
|||||||
@ -1,165 +0,0 @@
|
|||||||
/*
|
|
||||||
Copyright 2011 MCForge
|
|
||||||
|
|
||||||
Dual-licensed under the Educational Community License, Version 2.0 and
|
|
||||||
the GNU General Public License, Version 3 (the "Licenses"); you may
|
|
||||||
not use this file except in compliance with the Licenses. You may
|
|
||||||
obtain a copy of the Licenses at
|
|
||||||
|
|
||||||
http://www.opensource.org/licenses/ecl2.php
|
|
||||||
http://www.gnu.org/licenses/gpl-3.0.html
|
|
||||||
|
|
||||||
Unless required by applicable law or agreed to in writing,
|
|
||||||
software distributed under the Licenses are distributed on an "AS IS"
|
|
||||||
BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
|
|
||||||
or implied. See the Licenses for the specific language governing
|
|
||||||
permissions and limitations under the Licenses.
|
|
||||||
*/
|
|
||||||
using System;
|
|
||||||
using System.Collections.Generic;
|
|
||||||
using System.IO;
|
|
||||||
using System.IO.Packaging;
|
|
||||||
|
|
||||||
namespace MCGalaxy {
|
|
||||||
public static partial class Backup {
|
|
||||||
|
|
||||||
const string path = "MCGalaxy.zip";
|
|
||||||
public class BackupArgs {
|
|
||||||
public Player p;
|
|
||||||
public bool Files, Database, Lite;
|
|
||||||
}
|
|
||||||
|
|
||||||
public static void CreatePackage(Player p, bool files, bool db, bool lite) {
|
|
||||||
if (db) {
|
|
||||||
Server.s.Log("Backing up the database...");
|
|
||||||
using (StreamWriter sql = new StreamWriter("SQL.sql"))
|
|
||||||
BackupDatabase(sql,lite);
|
|
||||||
Server.s.Log("Backed up the database to SQL.sql");
|
|
||||||
}
|
|
||||||
|
|
||||||
List<Uri> filesList = null;
|
|
||||||
if (files) {
|
|
||||||
Server.s.Log("Determining which files to backup...");
|
|
||||||
string dir = Directory.GetCurrentDirectory() + "\\";
|
|
||||||
filesList = GetAllFiles(new DirectoryInfo("./"), new Uri(dir), lite);
|
|
||||||
Server.s.Log("Finished determining included files");
|
|
||||||
}
|
|
||||||
|
|
||||||
Server.s.Log("Creating compressed backup...");
|
|
||||||
using (ZipPackage package = (ZipPackage)ZipPackage.Open(path, FileMode.Create)) {
|
|
||||||
if (files) {
|
|
||||||
Server.s.Log("Compressing files...");
|
|
||||||
SaveFiles(package, filesList);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (db) SaveDatabase(package);
|
|
||||||
Server.s.Log("Compressed all data!");
|
|
||||||
}
|
|
||||||
Player.Message(p, "Backup of (" + (files ? "everything" + (db ? "" : " but database") : "database") + ") complete!");
|
|
||||||
Server.s.Log("Server backed up!");
|
|
||||||
}
|
|
||||||
|
|
||||||
const string undo1 = "extra/undo/", undo2 = @"extra\undo\";
|
|
||||||
const string prev1 = "extra/undoPrevious/", prev2 = @"extra\undoPrevious\";
|
|
||||||
const string levelBackup1 = "levels/backups/", levelBackup2 = @"levels\backups\";
|
|
||||||
const string levelPrev1 = "levels/prev/", levelPrev2 = @"levels\prev\";
|
|
||||||
const string blockDB1 = "blockdb/", blockDB2 = @"blockdb\";
|
|
||||||
|
|
||||||
static List<Uri> GetAllFiles(DirectoryInfo dir, Uri baseUri, bool lite) {
|
|
||||||
List<Uri> list = new List<Uri>();
|
|
||||||
foreach (FileSystemInfo entry in dir.GetFileSystemInfos()) {
|
|
||||||
if (entry is FileInfo) {
|
|
||||||
string path = ((FileInfo)entry).FullName;
|
|
||||||
if (lite && (path.Contains(undo1) || path.Contains(undo2))) continue;
|
|
||||||
if (lite && (path.Contains(prev1) || path.Contains(prev2))) continue;
|
|
||||||
if (lite && (path.Contains(levelBackup1) || path.Contains(levelBackup2))) continue;
|
|
||||||
if (lite && (path.Contains(levelPrev1) || path.Contains(levelPrev2))) continue;
|
|
||||||
if (lite && (path.Contains(blockDB1) || path.Contains(blockDB2))) continue;
|
|
||||||
|
|
||||||
// Make a relative URI
|
|
||||||
Uri uri = baseUri.MakeRelativeUri(new Uri(path));
|
|
||||||
if (uri.ToString().IndexOfAny("/\\".ToCharArray()) > 0)
|
|
||||||
list.Add(PackUriHelper.CreatePartUri(uri));
|
|
||||||
} else {
|
|
||||||
list.AddRange(GetAllFiles((DirectoryInfo)entry, baseUri, lite));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return list;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void SaveFiles(ZipPackage package, List<Uri> partURIs) {
|
|
||||||
foreach (Uri loc in partURIs) {
|
|
||||||
string file = Uri.UnescapeDataString(loc.ToString());
|
|
||||||
if (file.Contains(path)) continue;
|
|
||||||
|
|
||||||
try {
|
|
||||||
PackagePart part = package.CreatePart(loc, "");
|
|
||||||
using (Stream src = new FileStream("./" + file, FileMode.Open, FileAccess.Read))
|
|
||||||
CopyStream(src, part.GetStream());
|
|
||||||
} catch (Exception ex) {
|
|
||||||
Server.s.Log("Failed to save file: " + file);
|
|
||||||
Server.ErrorLog(ex);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void SaveDatabase(ZipPackage package) {
|
|
||||||
Server.s.Log("Compressing Database...");
|
|
||||||
Uri uri = new Uri("/SQL.sql", UriKind.Relative);
|
|
||||||
|
|
||||||
PackagePart part = package.CreatePart(uri, "", CompressionOption.Normal);
|
|
||||||
CopyStream(File.OpenRead("SQL.sql"), part.GetStream());
|
|
||||||
Server.s.Log("Database compressed");
|
|
||||||
}
|
|
||||||
|
|
||||||
static void CopyStream(Stream source, Stream target) {
|
|
||||||
const int bufSize = 0x1000;
|
|
||||||
byte[] buf = new byte[bufSize];
|
|
||||||
int bytesRead = 0;
|
|
||||||
while ((bytesRead = source.Read(buf, 0, bufSize)) > 0)
|
|
||||||
target.Write(buf, 0, bytesRead);
|
|
||||||
}
|
|
||||||
|
|
||||||
public static void ExtractPackage(Player p) {
|
|
||||||
int errors = 0;
|
|
||||||
using (FileStream src = File.OpenRead(path))
|
|
||||||
using (ZipPackage zip = (ZipPackage)ZipPackage.Open(src))
|
|
||||||
{
|
|
||||||
PackagePartCollection parts = zip.GetParts();
|
|
||||||
foreach (ZipPackagePart item in parts) {
|
|
||||||
ExtractItem(item, ref errors);
|
|
||||||
if (item.Uri.ToString().ToLower().Contains("sql.sql")) {
|
|
||||||
// If it's in there, they backed it up, meaning they want it restored
|
|
||||||
Backup.ReplaceDatabase(item.GetStream());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// To make life easier, we reload settings now, to maker it less likely to need restart
|
|
||||||
Command.all.Find("server").Use(null, "reload"); // Reload, as console
|
|
||||||
Player.Message(p, "Server restored" + (errors > 0 ? " with errors. May be a partial restore" : "") + ". Restart is reccommended, though not required.");
|
|
||||||
}
|
|
||||||
|
|
||||||
static void ExtractItem(ZipPackagePart item, ref int errors) {
|
|
||||||
string entry = item.Uri.ToString();
|
|
||||||
string file = "./" + Uri.UnescapeDataString(entry);
|
|
||||||
|
|
||||||
using (Stream src = item.GetStream()) {
|
|
||||||
try {
|
|
||||||
using (Stream dst = File.Create(file))
|
|
||||||
CopyStream(src, dst);
|
|
||||||
} catch {
|
|
||||||
try {
|
|
||||||
Directory.CreateDirectory("./" + entry.Substring(0, entry.LastIndexOfAny("\\/".ToCharArray())));
|
|
||||||
using (Stream dst = File.Create(file))
|
|
||||||
CopyStream(src, dst);
|
|
||||||
} catch (IOException e) {
|
|
||||||
Server.ErrorLog(e);
|
|
||||||
Server.s.Log("Caught ignoreable Error. See log for more details. Will continue with rest of files.");
|
|
||||||
errors++;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,234 +0,0 @@
|
|||||||
/*
|
|
||||||
Copyright 2011 MCForge
|
|
||||||
|
|
||||||
Dual-licensed under the Educational Community License, Version 2.0 and
|
|
||||||
the GNU General Public License, Version 3 (the "Licenses"); you may
|
|
||||||
not use this file except in compliance with the Licenses. You may
|
|
||||||
obtain a copy of the Licenses at
|
|
||||||
|
|
||||||
http://www.opensource.org/licenses/ecl2.php
|
|
||||||
http://www.gnu.org/licenses/gpl-3.0.html
|
|
||||||
|
|
||||||
Unless required by applicable law or agreed to in writing,
|
|
||||||
software distributed under the Licenses are distributed on an "AS IS"
|
|
||||||
BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
|
|
||||||
or implied. See the Licenses for the specific language governing
|
|
||||||
permissions and limitations under the Licenses.
|
|
||||||
*/
|
|
||||||
using System;
|
|
||||||
using System.Collections.Generic;
|
|
||||||
using System.Data;
|
|
||||||
using System.IO;
|
|
||||||
using MCGalaxy.SQL;
|
|
||||||
|
|
||||||
namespace MCGalaxy {
|
|
||||||
public static partial class Backup {
|
|
||||||
|
|
||||||
public static void BackupDatabase(StreamWriter sql, bool lite) {
|
|
||||||
//We technically know all tables in the DB... But since this is MySQL, we can also get them all with a MySQL command
|
|
||||||
//So we show the tables, and store the result.
|
|
||||||
//Also output information data (Same format as phpMyAdmin's dump)
|
|
||||||
|
|
||||||
//Important note: This does NOT account for foreign keys, BLOB's etc. It only works for what we actually put in the db.
|
|
||||||
|
|
||||||
sql.WriteLine("-- MCGalaxy SQL Database Dump");
|
|
||||||
sql.WriteLine("-- version 1.5");
|
|
||||||
sql.WriteLine("-- http://mcgalaxy.ml");
|
|
||||||
sql.WriteLine("--");
|
|
||||||
sql.WriteLine("-- Host: {0}", Server.MySQLHost);
|
|
||||||
sql.WriteLine("-- Generation Time: {0:d} at {0:HH:mm:ss}", DateTime.Now, DateTime.Now);
|
|
||||||
sql.WriteLine("-- MCGalaxy Version: {0}", Server.Version);
|
|
||||||
sql.WriteLine();
|
|
||||||
sql.WriteLine();
|
|
||||||
|
|
||||||
List<string> sqlTables = Database.Backend.AllTables();
|
|
||||||
foreach (string name in sqlTables) {
|
|
||||||
if (lite && name.CaselessStarts("Block")) continue;
|
|
||||||
BackupTable(name, sql);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public static void BackupTable(string tableName, StreamWriter sql) {
|
|
||||||
//For each table, we iterate through all rows, (and save them)
|
|
||||||
sql.WriteLine();
|
|
||||||
sql.WriteLine("-- --------------------------------------------------------");
|
|
||||||
sql.WriteLine("-- Table structure for table `{0}`", tableName);
|
|
||||||
sql.WriteLine();
|
|
||||||
WriteTableSchema(tableName, sql);
|
|
||||||
|
|
||||||
using (DataTable data = Database.Backend.GetRows(tableName, "*")) {
|
|
||||||
if (data.Rows.Count == 0) {
|
|
||||||
sql.WriteLine("-- No data in table `{0}`!", tableName);
|
|
||||||
sql.WriteLine();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
sql.WriteLine("--");
|
|
||||||
sql.WriteLine("-- Dumping data for table `{0}`", tableName);
|
|
||||||
sql.WriteLine("--");
|
|
||||||
sql.WriteLine();
|
|
||||||
|
|
||||||
List<DataColumn> allCols = new List<DataColumn>();
|
|
||||||
foreach (DataColumn col in data.Columns) {
|
|
||||||
allCols.Add(col);
|
|
||||||
}
|
|
||||||
string insertCols = FormatInsertColumns(allCols, tableName);
|
|
||||||
|
|
||||||
foreach (DataRow row in data.Rows) { //We rely on the correct datatype being given here.
|
|
||||||
sql.WriteLine();
|
|
||||||
sql.WriteLine(insertCols);
|
|
||||||
|
|
||||||
sql.Write("(");
|
|
||||||
for (int col = 0; col < data.Columns.Count; col++) {
|
|
||||||
//The values themselves can be integers or strings, or null
|
|
||||||
Type type = allCols[col].DataType;
|
|
||||||
if (row.IsNull(col)) {
|
|
||||||
sql.Write("NULL");
|
|
||||||
} else if (type == typeof(DateTime)) { // special format
|
|
||||||
sql.Write("'{0:yyyy-MM-dd HH:mm:ss.ffff}'", (DateTime)row[col]);
|
|
||||||
} else if (type == typeof(bool)) {
|
|
||||||
sql.Write((bool)row[col] ? "1" : "0");
|
|
||||||
} else if (type == typeof(string)) {
|
|
||||||
string value = row[col].ToString();
|
|
||||||
if (value.IndexOf('\'') >= 0) // escape '
|
|
||||||
value = value.Replace("'", "''");
|
|
||||||
sql.Write("'{0}'", value);
|
|
||||||
} else {
|
|
||||||
sql.Write(row[col]); // We assume all other data is left as-is
|
|
||||||
//This includes numbers, blobs, etc. (As well as objects, but we don't save them into the database)
|
|
||||||
}
|
|
||||||
sql.Write((col < row.ItemArray.Length - 1 ? ", " : ");"));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
sql.WriteLine();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static string FormatInsertColumns(List<DataColumn> cols, string name) {
|
|
||||||
string sql = "INSERT INTO `" + name + "` (`";
|
|
||||||
for (int i = 0; i < cols.Count; i++) {
|
|
||||||
sql += cols[i].ColumnName + "`";
|
|
||||||
if (i < cols.Count - 1) sql += ", `";
|
|
||||||
else sql += ") VALUES";
|
|
||||||
}
|
|
||||||
return sql;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void WriteTableSchema(string tableName, StreamWriter sql) {
|
|
||||||
if (Server.useMySQL) {
|
|
||||||
string pri = "";
|
|
||||||
sql.WriteLine("CREATE TABLE IF NOT EXISTS `{0}` (", tableName);
|
|
||||||
|
|
||||||
using (DataTable schema = Database.Fill("DESCRIBE `" + tableName + "`")) {
|
|
||||||
string[] rowParams = new string[schema.Columns.Count];
|
|
||||||
foreach (DataRow row in schema.Rows) {
|
|
||||||
//Save the info contained to file
|
|
||||||
List<string> tmp = new List<string>();
|
|
||||||
for (int col = 0; col < schema.Columns.Count; col++)
|
|
||||||
tmp.Add(row[col].ToString());
|
|
||||||
|
|
||||||
rowParams = tmp.ToArray();
|
|
||||||
rowParams[2] = (rowParams[2].CaselessEq("no") ? "NOT " : "DEFAULT ") + "NULL";
|
|
||||||
pri += (rowParams[3].CaselessEq("pri") ? rowParams[0] + ";" : "");
|
|
||||||
sql.WriteLine("`{0}` {1} {2}" + (rowParams[5].Equals("") ? "" : " {5}") + (pri.Equals("") && row == schema.Rows[schema.Rows.Count - 1] ? "" : ","), rowParams);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (pri != "") {
|
|
||||||
string[] tmp = pri.Substring(0, pri.Length - 1).Split(';');
|
|
||||||
sql.Write("PRIMARY KEY (`");
|
|
||||||
foreach (string prim in tmp) {
|
|
||||||
sql.Write(prim);
|
|
||||||
sql.Write("`" + (tmp[tmp.Length - 1].Equals(prim) ? ")" : ", `"));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
sql.WriteLine(");");
|
|
||||||
} else {
|
|
||||||
using (DataTable tableSQL = Database.Fill("SELECT sql FROM sqlite_master" +
|
|
||||||
" WHERE tbl_name LIKE '" + tableName + "'" +
|
|
||||||
" AND type = 'table' AND name NOT LIKE 'sqlite_%'" +
|
|
||||||
" ORDER BY substr(type,2,1), name"))
|
|
||||||
{
|
|
||||||
//just print out the data in the table.
|
|
||||||
foreach (DataRow row in tableSQL.Rows) {
|
|
||||||
string tableSQLString = row[0].ToString();
|
|
||||||
sql.WriteLine(tableSQLString.Replace(" " + tableName, " `" + tableName + "`").Replace("CREATE TABLE `" + tableName + "`", "CREATE TABLE IF NOT EXISTS `" + tableName + "`") + ";");
|
|
||||||
//We parse this ourselves to find the actual types.
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
sql.WriteLine();
|
|
||||||
}
|
|
||||||
|
|
||||||
internal static void ReplaceDatabase(Stream sql) {
|
|
||||||
using (FileStream backup = File.Create("backup.sql"))
|
|
||||||
BackupDatabase(new StreamWriter(backup), false); // backup
|
|
||||||
|
|
||||||
List<string> tables = Database.Backend.AllTables();
|
|
||||||
foreach (string table in tables)
|
|
||||||
Database.Backend.DeleteTable(table); // drop all tables
|
|
||||||
|
|
||||||
ImportSql(sql);
|
|
||||||
}
|
|
||||||
|
|
||||||
internal static void ImportSql(Stream sql) {
|
|
||||||
// Import data (we only have CREATE TABLE and INSERT INTO statements)
|
|
||||||
using (StreamReader reader = new StreamReader(sql))
|
|
||||||
using (BulkTransaction helper = BulkTransaction.Create())
|
|
||||||
{
|
|
||||||
List<string> buffer = new List<string>();
|
|
||||||
while (!reader.EndOfStream) {
|
|
||||||
string cmd = NextStatement(reader, buffer);
|
|
||||||
if (cmd == null || cmd.Length == 0) continue;
|
|
||||||
|
|
||||||
int index = cmd.ToUpper().IndexOf("CREATE TABLE");
|
|
||||||
if (index > -1) ParseCreate(ref cmd, index);
|
|
||||||
|
|
||||||
//Run the command in the transaction.
|
|
||||||
helper.Execute(cmd);
|
|
||||||
}
|
|
||||||
helper.Commit();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static string NextStatement(StreamReader reader, List<string> buffer) {
|
|
||||||
buffer.Clear();
|
|
||||||
string line = null;
|
|
||||||
|
|
||||||
while ((line = reader.ReadLine()) != null) {
|
|
||||||
if (line.StartsWith("--")) continue; // comment
|
|
||||||
line = line.Trim();
|
|
||||||
if (line.Length == 0) continue; // whitespace
|
|
||||||
|
|
||||||
buffer.Add(line);
|
|
||||||
if (line[line.Length - 1] == ';') break;
|
|
||||||
}
|
|
||||||
return buffer.Join("");
|
|
||||||
}
|
|
||||||
|
|
||||||
static void ParseCreate(ref string cmd, int index) {
|
|
||||||
cmd = cmd.Remove(0, index);
|
|
||||||
cmd = cmd.Replace(" unsigned", " UNSIGNED");
|
|
||||||
if (!Server.useMySQL) return;
|
|
||||||
|
|
||||||
// MySQL does not support the format used by the SQLite backend for the primary key
|
|
||||||
const string priKey = " PRIMARY KEY AUTOINCREMENT";
|
|
||||||
int priIndex = cmd.ToUpper().IndexOf(priKey);
|
|
||||||
if (priIndex == -1) return;
|
|
||||||
|
|
||||||
// Find the name of this column
|
|
||||||
char[] sepChars = { '\t', ' ' }; // chars that separate part of a column definition
|
|
||||||
char[] startChars = { '`', '(', ' ', ',', '\t' }; // chars that can start a column definition
|
|
||||||
string before = cmd.Substring(0, priIndex);
|
|
||||||
before = before.Substring(0, before.LastIndexOfAny(sepChars)); // get rid of column type
|
|
||||||
int nameStart = before.LastIndexOfAny(startChars) + 1;
|
|
||||||
string name = before.Substring(nameStart);
|
|
||||||
|
|
||||||
// Replace the 'PRIMARY KEY AUTOINCREMENT' with just 'AUTO_INCREMENT';
|
|
||||||
cmd = cmd.Remove(priIndex, priKey.Length);
|
|
||||||
cmd = cmd.Insert(priIndex, " AUTO_INCREMENT");
|
|
||||||
// Insert 'PRIMARY KEY' at end of columns definition
|
|
||||||
cmd = cmd.Insert(cmd.LastIndexOf(")"), ", PRIMARY KEY (`" + name + "`)");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
x
Reference in New Issue
Block a user