using System;
using System.Collections;
using System.Collections.Generic;
using System.Data;
using System.Data.Common;
using System.Linq;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using NewLife;
using NewLife.Collections;
using NewLife.Data;
using NewLife.Reflection;
using XCode.Common;
namespace XCode.DataAccessLayer
{
class DB2 : RemoteDb
{
#region 属性
/// <summary>返回数据库类型。外部DAL数据库类请使用Other</summary>
public override DatabaseType Type => DatabaseType.DB2;
private static DbProviderFactory _Factory;
/// <summary>工厂</summary>
public override DbProviderFactory Factory
{
get
{
if (_Factory == null)
{
lock (typeof(DB2))
{
#if __CORE
_Factory = GetProviderFactory("IBM.Data.DB2.Core.dll", "IBM.Data.DB2.Core.DB2Factory");
#else
_Factory = GetProviderFactory("IBM.Data.DB2.dll", "IBM.Data.DB2.DB2Factory");
#endif
}
}
return _Factory;
}
}
protected override void OnSetConnectionString(ConnectionStringBuilder builder)
{
base.OnSetConnectionString(builder);
// 修正数据源
if (builder.TryGetAndRemove("Data Source", out var str) && !str.IsNullOrEmpty())
{
if (str.Contains("://"))
{
var uri = new Uri(str);
var type = uri.Scheme.IsNullOrEmpty() ? "TCP" : uri.Scheme.ToUpper();
var port = uri.Port > 0 ? uri.Port : 1521;
var name = uri.PathAndQuery.TrimStart("/");
if (name.IsNullOrEmpty()) name = "ORCL";
str = $"(DESCRIPTION=(ADDRESS_LIST=(ADDRESS=(PROTOCOL={type})(HOST={uri.Host})(PORT={port})))(CONNECT_DATA=(SERVICE_NAME={name})))";
}
builder.TryAdd("Data Source", str);
}
}
#endregion
#region 方法
/// <summary>创建数据库会话</summary>
/// <returns></returns>
protected override IDbSession OnCreateSession() => new DB2Session(this);
/// <summary>创建元数据对象</summary>
/// <returns></returns>
protected override IMetaData OnCreateMetaData() => new DB2Meta();
public override Boolean Support(String providerName)
{
providerName = providerName.ToLower();
if (providerName.Contains("db2")) return true;
return false;
}
#endregion
#region 分页
/// <summary>已重写。获取分页 2012.9.26 HUIYUE修正分页BUG</summary>
/// <param name="sql">SQL语句</param>
/// <param name="startRowIndex">开始行，0表示第一行</param>
/// <param name="maximumRows">最大返回行数，0表示所有行</param>
/// <param name="keyColumn">主键列。用于not in分页</param>
/// <returns></returns>
public override String PageSplit(String sql, Int64 startRowIndex, Int64 maximumRows, String keyColumn)
{
// 从第一行开始
if (startRowIndex <= 0)
{
if (maximumRows <= 0) return sql;
if (!sql.ToLower().Contains("order by")) return $"Select * From ({sql}) T0 Where rownum<={maximumRows}";
}
//if (maximumRows <= 0)
// sql = String.Format("Select * From ({1}) XCode_T0 Where rownum>={0}", startRowIndex + 1, sql);
//else
sql = $"Select * From (Select T0.*, rownum as rowNumber From ({sql}) T0) T1 Where rowNumber>{startRowIndex}";
if (maximumRows > 0) sql += $" And rowNumber<={startRowIndex + maximumRows}";
return sql;
}
/// <summary>构造分页SQL</summary>
/// <remarks>
/// 两个构造分页SQL的方法，区别就在于查询生成器能够构造出来更好的分页语句，尽可能的避免子查询。
/// MS体系的分页精髓就在于唯一键，当唯一键带有Asc/Desc/Unkown等排序结尾时，就采用最大最小值分页，否则使用较次的TopNotIn分页。
/// TopNotIn分页和MaxMin分页的弊端就在于无法完美的支持GroupBy查询分页，只能查到第一页，往后分页就不行了，因为没有主键。
/// </remarks>
/// <param name="builder">查询生成器</param>
/// <param name="startRowIndex">开始行，0表示第一行</param>
/// <param name="maximumRows">最大返回行数，0表示所有行</param>
/// <returns>分页SQL</returns>
public override SelectBuilder PageSplit(SelectBuilder builder, Int64 startRowIndex, Int64 maximumRows)
{
/*
* DB2的rownum分页，在内层有Order By非主键排序时，外层的rownum会优先生效，
* 导致排序字段有相同值时无法在多次查询中保持顺序，（DB2算法参数会改变）。
* 其一，可以在排序字段后加上主键，确保排序内容唯一；
* 其二，可以在第二层提前取得rownum，然后在第三层外使用；
*
* 原分页算法始于2005年，只有在特殊情况下遇到分页出现重复数据的BUG：
* 排序、排序字段不包含主键且不唯一、排序字段拥有相同数值的数据行刚好被分到不同页上
*/
// 从第一行开始，不需要分页
if (startRowIndex <= 0)
{
if (maximumRows <= 0) return builder;
//// 如果带有排序，需要生成完整语句
//if (builder.OrderBy.IsNullOrEmpty())
return builder.AsChild("T0", false).AppendWhereAnd("rownum<={0}", maximumRows);
}
else if (maximumRows < 1)
throw new NotSupportedException();
builder = builder.AsChild("T0", false).AppendWhereAnd("rownum<={0}", startRowIndex + maximumRows);
builder.Column = "T0.*, rownum as rowNumber";
builder = builder.AsChild("T1", false).AppendWhereAnd("rowNumber>{0}", startRowIndex);
//builder = builder.AsChild("T0", false);
//builder.Column = "T0.*, rownum as rowNumber";
//builder = builder.AsChild("T1", false);
//builder.AppendWhereAnd("rowNumber>{0}", startRowIndex);
//if (maximumRows > 0) builder.AppendWhereAnd("rowNumber<={0}", startRowIndex + maximumRows);
return builder;
}
#endregion
#region 数据库特性
/// <summary>已重载。格式化时间</summary>
/// <param name="dt"></param>
/// <returns></returns>
public override String FormatDateTime(DateTime dt)
{
if (dt.Hour == 0 && dt.Minute == 0 && dt.Second == 0) return $"To_Date('{dt:yyyy-MM-dd}', 'YYYY-MM-DD')";
return $"To_Date('{dt:yyyy-MM-dd HH:mm:ss}', 'YYYY-MM-DD HH24:MI:SS')";
}
public override String FormatValue(IDataColumn field, Object value)
{
var code = System.Type.GetTypeCode(field.DataType);
var isNullable = field.Nullable;
if (code == TypeCode.String)
{
if (value == null) return isNullable ? "null" : "''";
if (field.RawType.StartsWithIgnoreCase("n"))
return "N'" + value.ToString().Replace("'", "''") + "'";
else
return "'" + value.ToString().Replace("'", "''") + "'";
}
return base.FormatValue(field, value);
}
internal protected override String ParamPrefix => ":";
/// <summary>字符串相加</summary>
/// <param name="left"></param>
/// <param name="right"></param>
/// <returns></returns>
public override String StringConcat(String left, String right) => (!String.IsNullOrEmpty(left) ? left : "\'\'") + "||" + (!String.IsNullOrEmpty(right) ? right : "\'\'");
/// <summary>创建参数</summary>
/// <param name="name">名称</param>
/// <param name="value">值</param>
/// <param name="type">类型</param>
/// <returns></returns>
public override IDataParameter CreateParameter(String name, Object value, Type type = null)
{
//var type = field?.DataType;
if (type == null)
{
type = value?.GetType();
// 参数可能是数组
if (type != null && type != typeof(Byte[]) && type.IsArray) type = type.GetElementTypeEx();
}
if (type == typeof(Boolean))
{
if (value is IEnumerable<Object> list)
value = list.Select(e => e.ToBoolean() ? 1 : 0).ToArray();
else if (value is IEnumerable<Boolean> list2)
value = list2.Select(e => e.ToBoolean() ? 1 : 0).ToArray();
else
value = value.ToBoolean() ? 1 : 0;
//type = typeof(Int32);
var dp2 = Factory.CreateParameter();
dp2.ParameterName = FormatParameterName(name);
dp2.Direction = ParameterDirection.Input;
dp2.DbType = DbType.Int32;
dp2.Value = value;
return dp2;
}
var dp = base.CreateParameter(name, value, type);
// 修正时间映射
if (type == typeof(DateTime)) dp.DbType = DbType.Date;
return dp;
}
#endregion
#region 关键字
protected override String ReservedWordsStr
{
get
{
return "ALL,ALTER,AND,ANY,AS,ASC,BETWEEN,BY,CHAR,CHECK,CLUSTER,COMPRESS,CONNECT,CREATE,DATE,DECIMAL,DEFAULT,DELETE,DESC,DISTINCT,DROP,ELSE,EXCLUSIVE,EXISTS,FLOAT,FOR,FROM,GRANT,GROUP,HAVING,IDENTIFIED,IN,INDEX,INSERT,INTEGER,INTERSECT,INTO,IS,LIKE,LOCK,LONG,MINUS,MODE,NOCOMPRESS,NOT,NOWAIT,NULL,NUMBER,OF,ON,OPTION,OR,ORDER,PCTFREE,PRIOR,PUBLIC,RAW,RENAME,RESOURCE,REVOKE,SELECT,SET,SHARE,SIZE,SMALLINT,START,SYNONYM,TABLE,THEN,TO,TRIGGER,UNION,UNIQUE,UPDATE,VALUES,VARCHAR,VARCHAR2,VIEW,WHERE,WITH," +
"Sort,Level,User,Online";
}
}
/// <summary>格式化关键字</summary>
/// <param name="keyWord">表名</param>
/// <returns></returns>
public override String FormatKeyWord(String keyWord)
{
if (String.IsNullOrEmpty(keyWord)) return keyWord;
var pos = keyWord.LastIndexOf(".");
if (pos < 0) return "\"" + keyWord + "\"";
var tn = keyWord.Substring(pos + 1);
if (tn.StartsWith("\"")) return keyWord;
return keyWord.Substring(0, pos + 1) + "\"" + tn + "\"";
}
#endregion
}
/// <summary>DB2数据库</summary>
internal class DB2Session : RemoteDbSession
{
#region 构造函数
public DB2Session(IDatabase db) : base(db) { }
#endregion
#region 基本方法 查询/执行
protected override DbTable OnFill(DbDataReader dr)
{
var dt = new DbTable();
dt.ReadHeader(dr);
Int32[] fields = null;
// 干掉rowNumber
var idx = Array.FindIndex(dt.Columns, c => c.EqualIgnoreCase("rowNumber"));
if (idx >= 0)
{
var cs = dt.Columns.ToList();
var ts = dt.Types.ToList();
var fs = Enumerable.Range(0, cs.Count).ToList();
cs.RemoveAt(idx);
ts.RemoveAt(idx);
fs.RemoveAt(idx);
dt.Columns = cs.ToArray();
dt.Types = ts.ToArray();
fields = fs.ToArray();
}
dt.ReadData(dr, fields);
return dt;
}
/// <summary>快速查询单表记录数，稍有偏差</summary>
/// <param name="tableName"></param>
/// <returns></returns>
public override Int64 QueryCountFast(String tableName)
{
if (String.IsNullOrEmpty(tableName)) return 0;
var p = tableName.LastIndexOf(".");
if (p >= 0 && p < tableName.Length - 1) tableName = tableName.Substring(p + 1);
tableName = tableName.ToUpper();
var owner = (Database as DB2).Owner;
if (owner.IsNullOrEmpty()) owner = (Database as DB2).User;
//var owner = (Database as DB2).Owner.ToUpper();
owner = owner.ToUpper();
// 某些表没有聚集索引，导致查出来的函数为零
var sql = $"select NUM_ROWS from all_tables where OWNER='{owner}' and TABLE_NAME='{tableName}'";
return ExecuteScalar<Int64>(sql);
}
static readonly Regex reg_SEQ = new(@"\b(\w+)\.nextval\b", RegexOptions.IgnoreCase | RegexOptions.Compiled);
/// <summary>执行插入语句并返回新增行的自动编号</summary>
/// <param name="sql">SQL语句</param>
/// <param name="type">命令类型，默认SQL文本</param>
/// <param name="ps">命令参数</param>
/// <returns>新增行的自动编号</returns>
public override Int64 InsertAndGetIdentity(String sql, CommandType type = CommandType.Text, params IDataParameter[] ps)
{
BeginTransaction(IsolationLevel.Serializable);
try
{
Int64 rs = Execute(sql, type, ps);
if (rs > 0)
{
var m = reg_SEQ.Match(sql);
if (m != null && m.Success && m.Groups != null && m.Groups.Count > 0)
rs = ExecuteScalar<Int64>($"Select {m.Groups[1].Value}.currval From dual");
}
Commit();
return rs;
}
catch { Rollback(true); throw; }
}
#if !NET40
public override Task<Int64> QueryCountFastAsync(String tableName)
{
if (String.IsNullOrEmpty(tableName)) return Task.FromResult(0L);
var p = tableName.LastIndexOf(".");
if (p >= 0 && p < tableName.Length - 1) tableName = tableName.Substring(p + 1);
tableName = tableName.ToUpper();
var owner = (Database as DB2).Owner;
if (owner.IsNullOrEmpty()) owner = (Database as DB2).User;
//var owner = (Database as DB2).Owner.ToUpper();
owner = owner.ToUpper();
// 某些表没有聚集索引，导致查出来的函数为零
var sql = $"select NUM_ROWS from all_tables where OWNER='{owner}' and TABLE_NAME='{tableName}'";
return ExecuteScalarAsync<Int64>(sql);
}
public override async Task<Int64> InsertAndGetIdentityAsync(String sql, CommandType type = CommandType.Text, params IDataParameter[] ps)
{
BeginTransaction(IsolationLevel.Serializable);
try
{
Int64 rs = await ExecuteAsync(sql, type, ps);
if (rs > 0)
{
var m = reg_SEQ.Match(sql);
if (m != null && m.Success && m.Groups != null && m.Groups.Count > 0)
rs = await ExecuteScalarAsync<Int64>($"Select {m.Groups[1].Value}.currval From dual");
}
Commit();
return rs;
}
catch { Rollback(true); throw; }
}
#endif
/// <summary>重载支持批量操作</summary>
/// <param name="sql"></param>
/// <param name="type"></param>
/// <param name="ps"></param>
/// <returns></returns>
protected override DbCommand OnCreateCommand(String sql, CommandType type = CommandType.Text, params IDataParameter[] ps)
{
var cmd = base.OnCreateCommand(sql, type, ps);
if (cmd == null) return null;
// 如果参数Value都是数组，那么就是批量操作
if (ps != null && ps.Length > 0 && ps.All(p => p.Value is IList))
{
var arr = ps.First().Value as IList;
cmd.SetValue("ArrayBindCount", arr.Count);
cmd.SetValue("BindByName", true);
// 超时时间放大10倍
if (cmd.CommandTimeout > 0)
cmd.CommandTimeout *= 10;
else
cmd.CommandTimeout = 120;
}
return cmd;
}
#endregion
#region 批量操作
public override Int32 Insert(IDataTable table, IDataColumn[] columns, IEnumerable<IExtend> list)
{
var ps = new HashSet<String>();
var sql = GetInsertSql(table, columns, ps);
var dps = GetParameters(columns, ps, list);
return Execute(sql, CommandType.Text, dps);
}
private String GetInsertSql(IDataTable table, IDataColumn[] columns, ICollection<String> ps)
{
var sb = Pool.StringBuilder.Get();
var db = Database as DbBase;
// 字段列表
sb.AppendFormat("Insert Into {0}(", db.FormatName(table));
foreach (var dc in columns)
{
//if (dc.Identity) continue;
sb.Append(db.FormatName(dc));
sb.Append(',');
}
sb.Length--;
sb.Append(')');
// 值列表
sb.Append(" Values(");
foreach (var dc in columns)
{
//if (dc.Identity) continue;
sb.Append(db.FormatParameterName(dc.Name));
sb.Append(',');
if (!ps.Contains(dc.Name)) ps.Add(dc.Name);
}
sb.Length--;
sb.Append(')');
return sb.Put(true);
}
private IDataParameter[] GetParameters(IDataColumn[] columns, ICollection<String> ps, IEnumerable<IExtend> list)
{
var db = Database;
var dps = new List<IDataParameter>();
foreach (var dc in columns)
{
//if (dc.Identity) continue;
if (!ps.Contains(dc.Name)) continue;
//var vs = new List<Object>();
var type = dc.DataType;
if (!type.IsInt() && type.IsEnum) type = typeof(Int32);
var arr = Array.CreateInstance(type, list.Count());
var k = 0;
foreach (var entity in list)
{
//vs.Add(entity[dc.Name]);
arr.SetValue(entity[dc.Name], k++);
}
var dp = db.CreateParameter(dc.Name, arr, dc);
dps.Add(dp);
}
return dps.ToArray();
}
public override Int32 Upsert(IDataTable table, IDataColumn[] columns, ICollection<String> updateColumns, ICollection<String> addColumns, IEnumerable<IExtend> list)
{
var ps = new HashSet<String>();
var insert = GetInsertSql(table, columns, ps);
var update = GetUpdateSql(table, columns, updateColumns, addColumns, ps);
var sb = Pool.StringBuilder.Get();
sb.AppendLine("BEGIN");
sb.AppendLine(insert + ";");
sb.AppendLine("EXCEPTION");
// 没有更新时，直接返回，可用于批量插入且其中部分有冲突需要忽略的场景
if (!update.IsNullOrEmpty())
{
sb.AppendLine("WHEN DUP_VAL_ON_INDEX THEN");
sb.AppendLine(update + ";");
}
else
{
//sb.AppendLine("WHEN OTHERS THEN");
sb.AppendLine("WHEN DUP_VAL_ON_INDEX THEN");
sb.AppendLine("RETURN;");
}
sb.AppendLine("END;");
var sql = sb.Put(true);
var dps = GetParameters(columns, ps, list);
return Execute(sql, CommandType.Text, dps);
}
private String GetUpdateSql(IDataTable table, IDataColumn[] columns, ICollection<String> updateColumns, ICollection<String> addColumns, ICollection<String> ps)
{
if ((updateColumns == null || updateColumns.Count == 0)
&& (addColumns == null || addColumns.Count == 0)) return null;
var sb = Pool.StringBuilder.Get();
var db = Database as DbBase;
// 字段列表
sb.AppendFormat("Update {0} Set ", db.FormatName(table));
foreach (var dc in columns)
{
if (dc.Identity || dc.PrimaryKey) continue;
if (addColumns != null && addColumns.Contains(dc.Name))
{
sb.AppendFormat("{0}={0}+{1},", db.FormatName(dc), db.FormatParameterName(dc.Name));
if (!ps.Contains(dc.Name)) ps.Add(dc.Name);
}
else if (updateColumns != null && updateColumns.Contains(dc.Name))
{
sb.AppendFormat("{0}={1},", db.FormatName(dc), db.FormatParameterName(dc.Name));
if (!ps.Contains(dc.Name)) ps.Add(dc.Name);
}
}
sb.Length--;
// 条件
sb.Append(" Where ");
foreach (var dc in columns)
{
if (!dc.PrimaryKey) continue;
sb.AppendFormat("{0}={1}", db.FormatName(dc), db.FormatParameterName(dc.Name));
sb.Append(" And ");
if (!ps.Contains(dc.Name)) ps.Add(dc.Name);
}
sb.Length -= " And ".Length;
return sb.Put(true);
}
public override Int32 Update(IDataTable table, IDataColumn[] columns, ICollection<String> updateColumns, ICollection<String> addColumns, IEnumerable<IExtend> list)
{
var ps = new HashSet<String>();
var sql = GetUpdateSql(table, columns, updateColumns, addColumns, ps);
var dps = GetParameters(columns, ps, list);
return Execute(sql, CommandType.Text, dps);
}
#endregion
}
/// <summary>DB2元数据</summary>
class DB2Meta : RemoteDbMetaData
{
public DB2Meta() => Types = _DataTypes;
/// <summary>拥有者</summary>
public String Owner
{
get
{
var owner = Database.Owner;
if (owner.IsNullOrEmpty()) owner = (Database as DB2).User;
return owner.ToUpper();
}
}
/// <summary>用户名</summary>
public String UserID => (Database as DB2).User.ToUpper();
/// <summary>取得所有表构架</summary>
/// <returns></returns>
protected override List<IDataTable> OnGetTables(String[] names)
{
DataTable dt = null;
// 不缺分大小写，并且不是保留字，才转大写
if (names != null)
{
var db = Database as DB2;
/*if (db.IgnoreCase)*/
names = names.Select(e => db.IsReservedWord(e) ? e : e.ToUpper()).ToArray();
}
// 采用集合过滤，提高效率
String tableName = null;
if (names != null && names.Length == 1) tableName = names.FirstOrDefault();
if (tableName.IsNullOrEmpty()) tableName = null;
var owner = Owner;
//if (owner.IsNullOrEmpty()) owner = UserID;
dt = GetSchema(_.Tables, new String[] { owner, tableName });
if (!dt.Columns.Contains("TABLE_TYPE"))
{
dt.Columns.Add("TABLE_TYPE", typeof(String));
foreach (var dr in dt.Rows?.ToArray())
{
dr["TABLE_TYPE"] = "Table";
}
}
var dtView = GetSchema(_.Views, new String[] { owner, tableName });
if (dtView != null && dtView.Rows.Count != 0)
{
foreach (var dr in dtView.Rows?.ToArray())
{
var drNew = dt.NewRow();
drNew["OWNER"] = dr["OWNER"];
drNew["TABLE_NAME"] = dr["VIEW_NAME"];
drNew["TABLE_TYPE"] = "View";
dt.Rows.Add(drNew);
}
}
var data = new NullableDictionary<String, DataTable>(StringComparer.OrdinalIgnoreCase);
// 如果表太多，则只要目标表数据
var mulTable = "";
if (dt.Rows.Count > 10 && names != null && names.Length > 0)
{
mulTable = $" And TABLE_NAME in ({names.Select(e => $"'{e}'").Join(",")})";
}
// 列和索引
data["Columns"] = Get("all_tab_columns", owner, tableName, mulTable);
data["Indexes"] = Get("all_indexes", owner, tableName, mulTable);
data["IndexColumns"] = Get("all_ind_columns", owner, tableName, mulTable, "Table_Owner");
// 主键
if (MetaDataCollections.Contains(_.PrimaryKeys)) data["PrimaryKeys"] = GetSchema(_.PrimaryKeys, new String[] { owner, tableName, null });
// 序列
data["Sequences"] = Get("all_sequences", owner, null, null, "Sequence_Owner");
// 表注释
data["TableComment"] = Get("all_tab_comments", owner, tableName, mulTable);
// 列注释
data["ColumnComment"] = Get("all_col_comments", owner, tableName, mulTable);
var list = GetTables(dt.Rows.ToArray(), names, data);
return list;
}
/// <summary>
/// 快速取得所有表名
/// </summary>
/// <returns></returns>
public override IList<String> GetTableNames()
{
var list = new List<String>();
var dt = GetSchema(_.Tables, new String[] { Owner, null });
if (dt?.Rows == null || dt.Rows.Count < 1) return list;
foreach (DataRow dr in dt.Rows)
{
list.Add(GetDataRowValue<String>(dr, _.TalbeName));
}
return list;
}
private DataTable Get(String name, String owner, String tableName, String mulTable = null, String ownerName = null)
{
if (ownerName.IsNullOrEmpty()) ownerName = "Owner";
var sql = $"Select * From {name} Where {ownerName}='{owner}'";
if (!tableName.IsNullOrEmpty())
sql += $" And TABLE_NAME='{tableName}'";
else if (!mulTable.IsNullOrEmpty())
sql += mulTable;
return Database.CreateSession().Query(sql).Tables[0];
}
protected override void FixTable(IDataTable table, DataRow dr, IDictionary<String, DataTable> data)
{
base.FixTable(table, dr, data);
// 主键
var dt = data?["PrimaryKeys"];
if (dt != null && dt.Rows.Count > 0)
{
var drs = dt.Select($"{_.TalbeName}='{table.TableName}'");
if (drs != null && drs.Length > 0)
{
// 找到主键所在索引，这个索引的列才是主键
if (TryGetDataRowValue(drs[0], _.IndexName, out String name) && !String.IsNullOrEmpty(name))
{
var di = table.Indexes.FirstOrDefault(i => i.Name == name);
if (di != null)
{
di.PrimaryKey = true;
foreach (var dc in table.Columns)
{
dc.PrimaryKey = di.Columns.Contains(dc.ColumnName);
}
}
}
}
}
// 表注释 USER_TAB_COMMENTS
table.Description = GetTableComment(table.TableName, data);
if (table?.Columns == null || table.Columns.Count == 0) return;
}
String GetTableComment(String name, IDictionary<String, DataTable> data)
{
var dt = data?["TableComment"];
if (dt?.Rows == null || dt.Rows.Count < 1) return null;
var where = $"TABLE_NAME='{name}'";
var drs = dt.Select(where);
if (drs != null && drs.Length > 0) return Convert.ToString(drs[0]["COMMENTS"]);
return null;
}
/// <summary>取得指定表的所有列构架</summary>
/// <param name="table"></param>
/// <param name="columns">列</param>
/// <param name="data"></param>
/// <returns></returns>
protected override List<IDataColumn> GetFields(IDataTable table, DataTable columns, IDictionary<String, DataTable> data)
{
var list = base.GetFields(table, columns, data);
if (list == null || list.Count < 1) return null;
// 字段注释
if (list != null && list.Count > 0)
{
foreach (var field in list)
{
field.Description = GetColumnComment(table.TableName, field.ColumnName, data);
}
}
return list;
}
const String KEY_OWNER = "OWNER";
protected override List<IDataColumn> GetFields(IDataTable table, DataRow[] rows)
{
if (rows == null || rows.Length < 1) return null;
var owner = Owner;
if (owner.IsNullOrEmpty() || !rows[0].Table.Columns.Contains(KEY_OWNER)) return base.GetFields(table, rows);
var list = new List<DataRow>();
foreach (var dr in rows)
{
if (TryGetDataRowValue(dr, KEY_OWNER, out String str) && owner.EqualIgnoreCase(str)) list.Add(dr);
}
return base.GetFields(table, list.ToArray());
}
String GetColumnComment(String tableName, String columnName, IDictionary<String, DataTable> data)
{
var dt = data?["ColumnComment"];
if (dt?.Rows == null || dt.Rows.Count < 1) return null;
var where = $"{_.TalbeName}='{tableName}' AND {_.ColumnName}='{columnName}'";
var drs = dt.Select(where);
if (drs != null && drs.Length > 0) return Convert.ToString(drs[0]["COMMENTS"]);
return null;
}
protected override void FixField(IDataColumn field, DataRow drColumn)
{
var dr = drColumn;
// 长度
//field.Length = GetDataRowValue<Int32>(dr, "CHAR_LENGTH", "DATA_LENGTH");
field.Length = GetDataRowValue<Int32>(dr, "DATA_LENGTH");
if (field is XField fi)
{
// 精度 与 位数
fi.Precision = GetDataRowValue<Int32>(dr, "DATA_PRECISION");
fi.Scale = GetDataRowValue<Int32>(dr, "DATA_SCALE");
if (field.Length == 0) field.Length = fi.Precision;
// 处理数字类型
if (field.RawType.StartsWithIgnoreCase("NUMBER"))
{
var prec = fi.Precision;
Type type = null;
if (fi.Scale == 0)
{
// 0表示长度不限制，为了方便使用，转为最常见的Int32
if (prec == 0)
type = typeof(Int32);
else if (prec == 1)
type = typeof(Boolean);
else if (prec <= 5)
type = typeof(Int16);
else if (prec <= 10)
type = typeof(Int32);
else
type = typeof(Int64);
}
else
{
if (prec == 0)
type = typeof(Decimal);
else if (prec <= 5)
type = typeof(Single);
else if (prec <= 10)
type = typeof(Double);
else
type = typeof(Decimal);
}
field.DataType = type;
if (prec > 0 && field.RawType.EqualIgnoreCase("NUMBER")) field.RawType += $"({prec},{fi.Scale})";
}
}
// 长度
if (TryGetDataRowValue(drColumn, "LENGTHINCHARS", out Int32 len) && len > 0) field.Length = len;
base.FixField(field, drColumn);
}
protected override String GetFieldType(IDataColumn field)
{
var precision = 0;
var scale = 0;
if (field is XField fi)
{
precision = fi.Precision;
scale = fi.Scale;
}
switch (Type.GetTypeCode(field.DataType))
{
case TypeCode.Boolean:
return "NUMBER(1, 0)";
case TypeCode.Int16:
case TypeCode.UInt16:
if (precision <= 0) precision = 5;
return $"NUMBER({precision}, 0)";
case TypeCode.Int32:
case TypeCode.UInt32:
//if (precision <= 0) precision = 10;
if (precision <= 0) return "NUMBER";
return $"NUMBER({precision}, 0)";
case TypeCode.Int64:
case TypeCode.UInt64:
if (precision <= 0) precision = 20;
return $"NUMBER({precision}, 0)";
case TypeCode.Single:
if (precision <= 0) precision = 5;
if (scale <= 0) scale = 1;
return $"NUMBER({precision}, {scale})";
case TypeCode.Double:
if (precision <= 0) precision = 10;
if (scale <= 0) scale = 2;
return $"NUMBER({precision}, {scale})";
case TypeCode.Decimal:
if (precision <= 0) precision = 20;
if (scale <= 0) scale = 4;
return $"NUMBER({precision}, {scale})";
default:
break;
}
return base.GetFieldType(field);
}
protected override void FixIndex(IDataIndex index, DataRow dr)
{
if (TryGetDataRowValue(dr, "UNIQUENESS", out String str))
index.Unique = str == "UNIQUE";
base.FixIndex(index, dr);
}
/// <summary>数据类型映射</summary>
private static readonly Dictionary<Type, String[]> _DataTypes = new()
{
{ typeof(Byte[]), new String[] { "RAW({0})", "BFILE", "BLOB", "LONG RAW" } },
{ typeof(Boolean), new String[] { "NUMBER(1,0)" } },
{ typeof(Byte), new String[] { "NUMBER(1,0)" } },
{ typeof(Int16), new String[] { "NUMBER(5,0)" } },
{ typeof(Int32), new String[] { "NUMBER(10,0)" } },
{ typeof(Int64), new String[] { "NUMBER(20,0)" } },
{ typeof(Single), new String[] { "BINARY_FLOAT" } },
{ typeof(Double), new String[] { "BINARY_DOUBLE" } },
{ typeof(Decimal), new String[] { "NUMBER({0}, {1})", "FLOAT({0})" } },
{ typeof(DateTime), new String[] { "DATE", "TIMESTAMP({0})", "TIMESTAMP({0} WITH LOCAL TIME ZONE)", "TIMESTAMP({0} WITH TIME ZONE)" } },
{ typeof(String), new String[] { "VARCHAR2({0})", "NVARCHAR2({0})", "LONG", "CHAR({0})", "CLOB", "NCHAR({0})", "NCLOB", "XMLTYPE", "ROWID" } }
};
#region 架构定义
public override Object SetSchema(DDLSchema schema, params Object[] values)
{
var session = Database.CreateSession();
var dbname = String.Empty;
var databaseName = String.Empty;
switch (schema)
{
case DDLSchema.DatabaseExist:
// DB2不支持判断数据库是否存在
return true;
default:
break;
}
return base.SetSchema(schema, values);
}
public override String DatabaseExistSQL(String dbname) => String.Empty;
protected override String GetFieldConstraints(IDataColumn field, Boolean onlyDefine)
{
var str = field.Nullable ? " NULL" : " NOT NULL";
// 默认值
if (!field.Nullable && !field.Identity)
{
str = GetDefault(field, onlyDefine) + str;
}
return str;
}
/// <summary>默认值</summary>
/// <param name="field"></param>
/// <param name="onlyDefine"></param>
/// <returns></returns>
protected override String GetDefault(IDataColumn field, Boolean onlyDefine)
{
if (field.DataType == typeof(DateTime)) return " DEFAULT To_Date('0001-01-01','yyyy-mm-dd')";
return base.GetDefault(field, onlyDefine);
}
public override String CreateTableSQL(IDataTable table)
{
var fs = new List<IDataColumn>(table.Columns);
var sb = new StringBuilder(32 + fs.Count * 20);
sb.AppendFormat("Create Table {0}(", FormatName(table));
for (var i = 0; i < fs.Count; i++)
{
sb.AppendLine();
sb.Append('\t');
sb.Append(FieldClause(fs[i], true));
if (i < fs.Count - 1) sb.Append(',');
}
// 主键
var pks = table.PrimaryKeys;
if (pks != null && pks.Length > 0)
{
sb.AppendFormat(",\r\n\tconstraint pk_{0} primary key ({1})", table.TableName, pks.Join(",", FormatName));
}
sb.AppendLine();
sb.Append(')');
var sql = sb.ToString();
if (sql.IsNullOrEmpty()) return sql;
// 去掉分号后的空格，DB2不支持同时执行多个语句
return sb.ToString();
}
public override String AddColumnSQL(IDataColumn field) => $"Alter Table {FormatName(field.Table)} Add {FieldClause(field, true)}";
public override String AlterColumnSQL(IDataColumn field, IDataColumn oldfield) => $"Alter Table {FormatName(field.Table)} Modify {FieldClause(field, false)}";
public override String DropColumnSQL(IDataColumn field) => $"Alter Table {FormatName(field.Table)} Drop Column {field}";
public override String AddTableDescriptionSQL(IDataTable table) => $"Comment On Table {FormatName(table)} is '{table.Description}'";
public override String DropTableDescriptionSQL(IDataTable table) => $"Comment On Table {FormatName(table)} is ''";
public override String AddColumnDescriptionSQL(IDataColumn field) => $"Comment On Column {FormatName(field.Table)}.{FormatName(field)} is '{field.Description}'";
public override String DropColumnDescriptionSQL(IDataColumn field) => $"Comment On Column {FormatName(field.Table)}.{FormatName(field)} is ''";
#endregion
}
}
|