using FreeSql.Internal; using FreeSql.Internal.Model; using Newtonsoft.Json.Linq; using Npgsql; using NpgsqlTypes; using System; using System.Collections; using System.Collections.Concurrent; using System.Collections.Generic; using System.Data; using System.Data.Common; using System.Globalization; using System.Linq; using System.Linq.Expressions; using System.Net; using System.Net.Http; using System.Numerics; using System.Text; using System.Threading; using FreeSql.QuestDb; using System.Web; using System.Threading.Tasks; using Newtonsoft.Json; using System.IO; using CsvHelper; namespace FreeSql.QuestDb { class QuestDbUtils : CommonUtils { string now_to_timezone; public QuestDbUtils(IFreeSql orm) : base(orm) { var timeOffset = (int)DateTime.Now.Subtract(DateTime.UtcNow).TotalMinutes; if (timeOffset == 0) now_to_timezone = "systimestamp()"; else { var absTimeOffset = Math.Abs(timeOffset); now_to_timezone = $"to_timezone(systimestamp(),'{(timeOffset > 0 ? '+' : '-')}{(absTimeOffset / 60).ToString().PadLeft(2, '0')}:{(absTimeOffset % 60).ToString().PadLeft(2, '0')}')"; } } static Array getParamterArrayValue(Type arrayType, object value, object defaultValue) { var valueArr = value as Array; var len = valueArr.GetLength(0); var ret = Array.CreateInstance(arrayType, len); for (var a = 0; a < len; a++) { var item = valueArr.GetValue(a); ret.SetValue(item == null ? defaultValue : getParamterValue(item.GetType(), item, 1), a); } return ret; } static Dictionary> dicGetParamterValue = new Dictionary> { { typeof(JToken).FullName, a => string.Concat(a) }, { typeof(JToken[]).FullName, a => getParamterArrayValue(typeof(string), a, null) }, { typeof(JObject).FullName, a => string.Concat(a) }, { typeof(JObject[]).FullName, a => getParamterArrayValue(typeof(string), a, null) }, { typeof(JArray).FullName, a => string.Concat(a) }, { typeof(JArray[]).FullName, a => getParamterArrayValue(typeof(string), a, null) }, { typeof(uint).FullName, a => long.Parse(string.Concat(a)) }, { typeof(uint[]).FullName, a => getParamterArrayValue(typeof(long), a, 0) }, { typeof(uint?[]).FullName, a => getParamterArrayValue(typeof(long?), a, null) }, { typeof(ulong).FullName, a => decimal.Parse(string.Concat(a)) }, { typeof(ulong[]).FullName, a => getParamterArrayValue(typeof(decimal), a, 0) }, { typeof(ulong?[]).FullName, a => getParamterArrayValue(typeof(decimal?), a, null) }, { typeof(ushort).FullName, a => int.Parse(string.Concat(a)) }, { typeof(ushort[]).FullName, a => getParamterArrayValue(typeof(int), a, 0) }, { typeof(ushort?[]).FullName, a => getParamterArrayValue(typeof(int?), a, null) }, { typeof(byte).FullName, a => short.Parse(string.Concat(a)) }, { typeof(byte[]).FullName, a => getParamterArrayValue(typeof(short), a, 0) }, { typeof(byte?[]).FullName, a => getParamterArrayValue(typeof(short?), a, null) }, { typeof(sbyte).FullName, a => short.Parse(string.Concat(a)) }, { typeof(sbyte[]).FullName, a => getParamterArrayValue(typeof(short), a, 0) }, { typeof(sbyte?[]).FullName, a => getParamterArrayValue(typeof(short?), a, null) }, { typeof(char).FullName, a => string.Concat(a).Replace('\0', ' ').ToCharArray().FirstOrDefault() }, { typeof(BigInteger).FullName, a => BigInteger.Parse(string.Concat(a), System.Globalization.NumberStyles.Any) }, { typeof(BigInteger[]).FullName, a => getParamterArrayValue(typeof(BigInteger), a, 0) }, { typeof(BigInteger?[]).FullName, a => getParamterArrayValue(typeof(BigInteger?), a, null) }, { typeof(NpgsqlPath).FullName, a => { var path = (NpgsqlPath)a; try { int count = path.Count; return path; } catch { return new NpgsqlPath(new NpgsqlPoint(0, 0)); } } }, { typeof(NpgsqlPath[]).FullName, a => getParamterArrayValue(typeof(NpgsqlPath), a, new NpgsqlPath(new NpgsqlPoint(0, 0))) }, { typeof(NpgsqlPath?[]).FullName, a => getParamterArrayValue(typeof(NpgsqlPath?), a, null) }, { typeof(NpgsqlPolygon).FullName, a => { var polygon = (NpgsqlPolygon)a; try { int count = polygon.Count; return polygon; } catch { return new NpgsqlPolygon(new NpgsqlPoint(0, 0), new NpgsqlPoint(0, 0)); } } }, { typeof(NpgsqlPolygon[]).FullName, a => getParamterArrayValue(typeof(NpgsqlPolygon), a, new NpgsqlPolygon(new NpgsqlPoint(0, 0), new NpgsqlPoint(0, 0))) }, { typeof(NpgsqlPolygon?[]).FullName, a => getParamterArrayValue(typeof(NpgsqlPolygon?), a, null) }, { typeof((IPAddress Address, int Subnet)).FullName, a => { var inet = ((IPAddress Address, int Subnet))a; if (inet.Address == null) return (IPAddress.Any, inet.Subnet); return inet; } }, { typeof((IPAddress Address, int Subnet)[]).FullName, a => getParamterArrayValue(typeof((IPAddress Address, int Subnet)), a, (IPAddress.Any, 0)) }, { typeof((IPAddress Address, int Subnet)?[]).FullName, a => getParamterArrayValue(typeof((IPAddress Address, int Subnet)?), a, null) }, }; static object getParamterValue(Type type, object value, int level = 0) { if (type.FullName == "System.Byte[]") return value; if (type.FullName == "System.Char[]") return value; if (type.IsArray && level == 0) { var elementType = type.GetElementType(); Type enumType = null; if (elementType.IsEnum) enumType = elementType; else if (elementType.IsNullableType() && elementType.GenericTypeArguments.First().IsEnum) enumType = elementType.GenericTypeArguments.First(); if (enumType != null) return enumType.GetCustomAttributes(typeof(FlagsAttribute), false).Any() ? getParamterArrayValue(typeof(long), value, elementType.IsEnum ? null : enumType.CreateInstanceGetDefaultValue()) : getParamterArrayValue(typeof(int), value, elementType.IsEnum ? null : enumType.CreateInstanceGetDefaultValue()); return dicGetParamterValue.TryGetValue(type.FullName, out var trydicarr) ? trydicarr(value) : value; } if (type.IsNullableType()) type = type.GenericTypeArguments.First(); if (type.IsEnum) return (int)value; if (dicGetParamterValue.TryGetValue(type.FullName, out var trydic)) return trydic(value); return value; } public override DbParameter AppendParamter(List _params, string parameterName, ColumnInfo col, Type type, object value) { if (string.IsNullOrEmpty(parameterName)) parameterName = $"p_{_params?.Count}"; if (value != null) value = getParamterValue(type, value); var ret = new NpgsqlParameter { ParameterName = QuoteParamterName(parameterName), Value = value }; if (value != null && (value.GetType().IsEnum || value.GetType().GenericTypeArguments.FirstOrDefault()?.IsEnum == true)) { ret.DataTypeName = ""; } else { var tp = _orm.CodeFirst.GetDbInfo(type)?.type; if (tp != null) { var ntp = (NpgsqlDbType)tp; switch (ntp) { case NpgsqlDbType.Boolean: ret.NpgsqlDbType = NpgsqlDbType.Text; ret.DbType = DbType.String; ret.Value = string.Concat(value); break; case NpgsqlDbType.Numeric: ret.NpgsqlDbType = NpgsqlDbType.Double; ret.DbType = DbType.Double; ret.Value = Convert.ToDouble(FormatSql("{0}", value)); break; default: ret.NpgsqlDbType = (NpgsqlDbType)tp.Value; break; } } if (col != null) { var dbtype = (NpgsqlDbType)_orm.DbFirst.GetDbType(new DatabaseModel.DbColumnInfo { DbTypeText = col.DbTypeText }); if (dbtype != NpgsqlDbType.Unknown) { ret.NpgsqlDbType = dbtype; //if (col.DbSize != 0) ret.Size = col.DbSize; if (col.DbPrecision != 0) ret.Precision = col.DbPrecision; if (col.DbScale != 0) ret.Scale = col.DbScale; } } } _params?.Add(ret); return ret; } public override DbParameter[] GetDbParamtersByObject(string sql, object obj) => Utils.GetDbParamtersByObject(sql, obj, "@", (name, type, value) => { if (value != null) value = getParamterValue(type, value); var ret = new NpgsqlParameter { ParameterName = $"@{name}", Value = value }; //if (value.GetType().IsEnum || value.GetType().GenericTypeArguments.FirstOrDefault()?.IsEnum == true) { // ret.DataTypeName = ""; //} else { var tp = _orm.CodeFirst.GetDbInfo(type)?.type; if (tp != null) ret.NpgsqlDbType = (NpgsqlDbType)tp.Value; //} return ret; }); public override string FormatSql(string sql, params object[] args) => sql?.FormatQuestDb(args); public override string QuoteSqlNameAdapter(params string[] name) { if (name.Length == 1) { var nametrim = name[0].Trim(); if (nametrim.StartsWith("(") && nametrim.EndsWith(")")) return nametrim; //原生SQL if (nametrim.StartsWith("\"") && nametrim.EndsWith("\"")) return nametrim; return $"\"{nametrim.Replace(".", "\".\"")}\""; } return $"\"{string.Join("\".\"", name)}\""; } public override string TrimQuoteSqlName(string name) { var nametrim = name.Trim(); if (nametrim.StartsWith("(") && nametrim.EndsWith(")")) return nametrim; //原生SQL return $"{nametrim.Trim('"').Replace("\".\"", ".").Replace(".\"", ".")}"; } public override string[] SplitTableName(string name) => GetSplitTableNames(name, '"', '"', 2); public override string QuoteParamterName(string name) => $"@{name}"; public override string IsNull(string sql, object value) => $"coalesce({sql}, {value})"; public override string StringConcat(string[] objs, Type[] types) => $"{string.Join(" || ", objs)}"; public override string Mod(string left, string right, Type leftType, Type rightType) => $"{left} % {right}"; public override string Div(string left, string right, Type leftType, Type rightType) => $"{left} / {right}"; public override string Now => now_to_timezone; public override string NowUtc => "systimestamp()"; public override string QuoteWriteParamterAdapter(Type type, string paramterName) => paramterName; protected override string QuoteReadColumnAdapter(Type type, Type mapType, string columnName) => columnName; public override string GetNoneParamaterSqlValue(List specialParams, string specialParamFlag, ColumnInfo col, Type type, object value) { if (value == null) return "NULL"; if (type.IsNumberType()) return string.Format(CultureInfo.InvariantCulture, "{0}", value); value = getParamterValue(type, value); var type2 = value.GetType(); if (type2 == typeof(byte[])) return $"'\\x{CommonUtils.BytesSqlRaw(value as byte[])}'"; else if (dicGetParamterValue.ContainsKey(type2.FullName)) { value = string.Concat(value); } return FormatSql("{0}", value, 1); } } } namespace FreeSql { public enum SampleUnit { microsecond = 'U', millisecond = 'T', second = 's', minute = 'm', hour = 'h', day = 'd', month = 'M', year = 'y', } }