using System.Collections.Generic;
|
|
namespace IStation.DAL
|
{
|
/// <summary>
|
///
|
/// </summary>
|
public class ScheduleCombine
|
{
|
public SqlSugarClient Connection
|
{
|
get { return ConfigHelper.GetSqlSugarClient(); }
|
}
|
|
private readonly string _tableNamePrefix = "ScheduleCombine_Cb_";
|
|
private string GetTableName(string runFlag)
|
{
|
return $"{_tableNamePrefix}{runFlag}";
|
}
|
|
private readonly string _createTable = "(\r\n\r\n `ID` Int64,\r\n\r\n `RunFlag` String,\r\n\r\n `RunCount` Int32,\r\n\r\n `Pump1` Float64,\r\n\r\n `Pump2` Float64,\r\n\r\n `Pump3` Float64,\r\n\r\n `CurveQH` String,\r\n\r\n `CurveQP` String,\r\n\r\n `MaxFlow` Float64,\r\n\r\n `MinFlow` Float64,\r\n\r\n `MaxHead` Float64,\r\n\r\n `MinHead` Float64,\r\n\r\n `AnaStatus` Bool\r\n)\r\nENGINE = MergeTree\r\nPRIMARY KEY ID\r\nORDER BY ID\r\nSETTINGS index_granularity = 8192;";
|
|
/// <summary>
|
/// 获取全部表名
|
/// </summary>
|
public List<string> GetAllTableName()
|
{
|
var tables = new List<string>();
|
//using (var connection = new SQLiteConnection(ConfigHelper.ConnectionString))
|
//{
|
// connection.Open();
|
// var sql = $"select name from sqlite_master where type='table' and name like '%{_tableNamePrefix}%';";
|
// using (var command = new SQLiteCommand(sql, connection))
|
// {
|
// using (var reader = command.ExecuteReader())
|
// {
|
// while (reader.Read())
|
// {
|
// var tableName = reader["name"].ToString();
|
// tables.Add(tableName);
|
// }
|
// }
|
// }
|
//}
|
using (SqlSugarClient db = Connection)
|
{
|
|
var exist_sql = $"SELECT name FROM system.tables WHERE database = 'default' AND name like '%{_tableNamePrefix}%';";
|
using (var reader = db.Ado.GetDataReader(exist_sql))
|
{
|
while (reader.Read())
|
{
|
var tableName = reader["name"].ToString();
|
tables.Add(tableName);
|
}
|
}
|
}
|
return tables;
|
}
|
|
|
/// <summary>
|
/// 通过表名获取所有未分析组合
|
/// </summary>
|
public List<Entity.ScheduleCombine> GetAllUnanalyzedAnaByTableName(string tableName)
|
{
|
using (SqlSugarClient db = Connection)
|
{
|
var sql = $"select * from {tableName} where AnaStatus=False";
|
var list = db.Ado.SqlQuery<Entity.ScheduleCombine>(sql);
|
return list;
|
}
|
}
|
|
|
|
/// <summary>
|
/// 大批量插入
|
/// </summary>
|
public bool BulkInserts_Create(string runFlag, List<Entity.ScheduleCombine> list)
|
{
|
if (list == null || list.Count < 1)
|
return default;
|
var tableName = GetTableName(runFlag);
|
using (SqlSugarClient db = Connection)
|
{
|
|
var exist_sql = $"SELECT COUNT(*) FROM system.tables WHERE database = 'default' AND name = '{tableName}'";
|
if (db.Ado.GetInt(exist_sql) < 1)
|
{
|
var sql_create_table = $"CREATE TABLE default.{tableName} {_createTable}";
|
db.Ado.ExecuteCommand(sql_create_table);
|
}
|
//大数据写入
|
return db.Fastest<Entity.ScheduleCombine>().AS(tableName).BulkCopy(list) > 0;
|
}
|
}
|
|
|
/// <summary>
|
/// 设置已分析状态
|
/// </summary>
|
public bool SetAnaStatus(string tableName, long id)
|
{
|
using (SqlSugarClient db = Connection)
|
{
|
var sql = $"update {tableName} set AnaStatus=True where ID={id}";
|
var bol = db.Ado.ExecuteCommand(sql) > 0;
|
return bol;
|
}
|
}
|
|
|
/// <summary>
|
/// 设置已分析状态
|
/// </summary>
|
public bool SetAnaStatus(string tableName, List<long> ids)
|
{
|
using (SqlSugarClient db = Connection)
|
{
|
var sql = $"update {tableName} set AnaStatus=True where ID in(@ids)";
|
var obj = new { ids = ids.ToArray() };
|
var bol = db.Ado.ExecuteCommand(sql, obj) > 0;
|
return bol;
|
}
|
}
|
|
|
/// <summary>
|
/// 判断数据库是否存在表
|
/// </summary>
|
public bool ExistTable(string runFlag)
|
{
|
using (SqlSugarClient db = Connection)
|
{
|
var exist_sql = $"SELECT COUNT(*) FROM system.tables WHERE database = 'default' AND name = '{GetTableName(runFlag)}'";
|
var bol = db.Ado.GetInt(exist_sql) > 0;
|
return bol;
|
}
|
}
|
|
/// <summary>
|
/// 获取全部表名
|
/// </summary>
|
public bool DeleteAllTable()
|
{
|
var tables = GetAllTableName();
|
if (tables == null || !tables.Any())
|
return false;
|
|
using (SqlSugarClient db = Connection)
|
{
|
db.BeginTran();
|
|
foreach (var table in tables)
|
{
|
var sql = $"drop table {table};";
|
db.Ado.ExecuteCommand(sql);
|
}
|
db.CommitTran();
|
}
|
return true;
|
}
|
|
|
}
|
}
|
