using FreeSql.Internal; using FreeSql.Internal.Model; using System; using System.Collections.Generic; using System.Data; using System.Linq; using System.Text; using System.Threading.Tasks; namespace FreeSql.PostgreSQL.Curd { class PostgreSQLUpdate : Internal.CommonProvider.UpdateProvider where T1 : class { public PostgreSQLUpdate(IFreeSql orm, CommonUtils commonUtils, CommonExpression commonExpression, object dywhere) : base(orm, commonUtils, commonExpression, dywhere) { } public override int ExecuteAffrows() => base.SplitExecuteAffrows(500, 3000); public override Task ExecuteAffrowsAsync() => base.SplitExecuteAffrowsAsync(500, 3000); public override List ExecuteUpdated() => base.SplitExecuteUpdated(500, 3000); public override Task> ExecuteUpdatedAsync() => base.SplitExecuteUpdatedAsync(500, 3000); protected override List RawExecuteUpdated() { var sql = this.ToSql(); if (string.IsNullOrEmpty(sql)) return new List(); var sb = new StringBuilder(); sb.Append(sql).Append(" RETURNING "); var colidx = 0; foreach (var col in _table.Columns.Values) { if (colidx > 0) sb.Append(", "); sb.Append(_commonUtils.QuoteReadColumn(col.Attribute.MapType, _commonUtils.QuoteSqlName(col.Attribute.Name))).Append(" as ").Append(_commonUtils.QuoteSqlName(col.CsName)); ++colidx; } sql = sb.ToString(); var dbParms = _params.Concat(_paramsSource).ToArray(); var before = new Aop.CurdBeforeEventArgs(_table.Type, Aop.CurdType.Update, sql, dbParms); _orm.Aop.CurdBefore?.Invoke(this, before); var ret = new List(); Exception exception = null; try { ret = _orm.Ado.Query(_connection, _transaction, CommandType.Text, sql, dbParms); ValidateVersionAndThrow(ret.Count); } catch (Exception ex) { exception = ex; throw ex; } finally { var after = new Aop.CurdAfterEventArgs(before, exception, ret); _orm.Aop.CurdAfter?.Invoke(this, after); } this.ClearData(); return ret; } async protected override Task> RawExecuteUpdatedAsync() { var sql = this.ToSql(); if (string.IsNullOrEmpty(sql)) return new List(); var sb = new StringBuilder(); sb.Append(sql).Append(" RETURNING "); var colidx = 0; foreach (var col in _table.Columns.Values) { if (colidx > 0) sb.Append(", "); sb.Append(_commonUtils.QuoteReadColumn(col.Attribute.MapType, _commonUtils.QuoteSqlName(col.Attribute.Name))).Append(" as ").Append(_commonUtils.QuoteSqlName(col.CsName)); ++colidx; } sql = sb.ToString(); var dbParms = _params.Concat(_paramsSource).ToArray(); var before = new Aop.CurdBeforeEventArgs(_table.Type, Aop.CurdType.Update, sql, dbParms); _orm.Aop.CurdBefore?.Invoke(this, before); var ret = new List(); Exception exception = null; try { ret = await _orm.Ado.QueryAsync(_connection, _transaction, CommandType.Text, sql, dbParms); ValidateVersionAndThrow(ret.Count); } catch (Exception ex) { exception = ex; throw ex; } finally { var after = new Aop.CurdAfterEventArgs(before, exception, ret); _orm.Aop.CurdAfter?.Invoke(this, after); } this.ClearData(); return ret; } protected override void ToSqlCase(StringBuilder caseWhen, ColumnInfo[] primarys) { if (_table.Primarys.Length == 1) { caseWhen.Append(_commonUtils.QuoteReadColumn(_table.Primarys.First().Attribute.MapType, _commonUtils.QuoteSqlName(_table.Primarys.First().Attribute.Name))); return; } caseWhen.Append("("); var pkidx = 0; foreach (var pk in _table.Primarys) { if (pkidx > 0) caseWhen.Append(" || "); caseWhen.Append(_commonUtils.QuoteReadColumn(pk.Attribute.MapType, _commonUtils.QuoteSqlName(pk.Attribute.Name))).Append("::varchar"); ++pkidx; } caseWhen.Append(")"); } protected override void ToSqlWhen(StringBuilder sb, ColumnInfo[] primarys, object d) { if (_table.Primarys.Length == 1) { sb.Append(_commonUtils.FormatSql("{0}", _table.Primarys.First().GetMapValue(d))); return; } sb.Append("("); var pkidx = 0; foreach (var pk in _table.Primarys) { if (pkidx > 0) sb.Append(" || "); sb.Append(_commonUtils.FormatSql("{0}", pk.GetMapValue(d))).Append("::varchar"); ++pkidx; } sb.Append(")"); } } }