AstralTransRocketries/MCGalaxy
1
0
Fork 0
mirror of https://github.com/ClassiCube/MCGalaxy.git synced 2025-09-22 03:55:18 -04:00
Code Issues Packages Projects Releases Wiki Activity

Can retrieve columns name for a database table now.

Browse Source
This commit is contained in:
UnknownShadow200 2017-08-26 11:53:28 +10:00
parent 2b245672d4
commit 26e2483c66
4 changed files with 34 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
MCGalaxy/Database/Backends/MySQL.cs
View File

@ -78,6 +78,17 @@ namespace MCGalaxy.SQL {
}
}
public override List<string> ColumnNames(string table) {
ValidateTable(table);
using (DataTable results = Database.Fill("DESCRIBE `" + table + "`")) {
List<string> columns = new List<string>(results.Rows.Count);
foreach (DataRow row in results.Rows) {
columns.Add(row["Field"].ToString());
}
return columns;
}
}
public override void RenameTable(string srcTable, string dstTable) {
ValidateTable(srcTable);
ValidateTable(dstTable);

11
MCGalaxy/Database/Backends/SQLite.cs
View File

@ -75,6 +75,17 @@ namespace MCGalaxy.SQL {
}
}
public override List<string> ColumnNames(string table) {
ValidateTable(table);
using (DataTable results = Database.Fill("PRAGMA table_info(`" + table + "`)")) {
List<string> columns = new List<string>(results.Rows.Count);
foreach (DataRow row in results.Rows) {
columns.Add(row["name"].ToString());
}
return columns;
}
}
public override void RenameTable(string srcTable, string dstTable) {
ValidateTable(srcTable);
ValidateTable(dstTable);

3
MCGalaxy/Database/IDatabaseBackend.cs
View File

@ -63,6 +63,9 @@ namespace MCGalaxy.SQL {
/// <summary> Returns a list of all tables in this database. </summary>
public abstract List<string> AllTables();
/// <summary> Returns a list of the column names in the given table. </summary>
public abstract List<string> ColumnNames(string table);
/// <summary> Renames the source table to the given name. </summary>
public abstract void RenameTable(string srcTable, string dstTable);

24
MCGalaxy/Server/Server.DB.cs
View File

@ -16,6 +16,7 @@
permissions and limitations under the Licenses.
*/
using System;
using System.Collections.Generic;
using System.Data;
using System.IO;
using MCGalaxy.SQL;
@ -81,29 +82,22 @@ namespace MCGalaxy {
// Here, since SQLite is a NEW thing from 5.3.0.0, we do not have to check for existing tables in SQLite.
if (!ServerConfig.UseMySQL) return;
// Check if the color column exists.
DataTable colorExists = Database.Fill("SHOW COLUMNS FROM Players WHERE `Field`='color'");
if (colorExists.Rows.Count == 0) {
List<string> columns = Database.Backend.ColumnNames("Players");
if (columns.Count == 0) return;
if (!columns.CaselessContains("Color")) {
Database.Backend.AddColumn("Players", new ColumnDesc("color", ColumnType.VarChar, 6), "totalKicked");
}
colorExists.Dispose();
DataTable tcolorExists = Database.Fill("SHOW COLUMNS FROM Players WHERE `Field`='title_color'");
if (tcolorExists.Rows.Count == 0) {
if (!columns.CaselessContains("Title_Color")) {
Database.Backend.AddColumn("Players", new ColumnDesc("title_color", ColumnType.VarChar, 6), "color");
}
tcolorExists.Dispose();
DataTable timespent = Database.Fill("SHOW COLUMNS FROM Players WHERE `Field`='TimeSpent'");
if (timespent.Rows.Count == 0) {
if (!columns.CaselessContains("TimeSpent")) {
Database.Backend.AddColumn("Players", new ColumnDesc("TimeSpent", ColumnType.VarChar, 20), "totalKicked");
}
timespent.Dispose();
DataTable totalCuboided = Database.Fill("SHOW COLUMNS FROM Players WHERE `Field`='totalCuboided'");
if (totalCuboided.Rows.Count == 0) {
if (!columns.CaselessContains("TotalCuboided")) {
Database.Backend.AddColumn("Players", new ColumnDesc("totalCuboided", ColumnType.Int64), "totalBlocks");
}
totalCuboided.Dispose();
}
}
}