SqlServerDbMaintenance.cs 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using System.Text;
  5. using System.Text.RegularExpressions;
  6. namespace SqlSugar
  7. {
  8. public class SqlServerDbMaintenance : DbMaintenanceProvider
  9. {
  10. #region DML
  11. protected override string GetDataBaseSql
  12. {
  13. get
  14. {
  15. return "SELECT NAME FROM master.dbo.sysdatabases ORDER BY NAME";
  16. }
  17. }
  18. protected override string GetColumnInfosByTableNameSql
  19. {
  20. get
  21. {
  22. string sql = @"SELECT sysobjects.name AS TableName,
  23. syscolumns.Id AS TableId,
  24. syscolumns.name AS DbColumnName,
  25. systypes.name AS DataType,
  26. COLUMNPROPERTY(syscolumns.id,syscolumns.name,'PRECISION') as [length],
  27. isnull(COLUMNPROPERTY(syscolumns.id,syscolumns.name,'Scale'),0) as Scale,
  28. isnull(COLUMNPROPERTY(syscolumns.id,syscolumns.name,'Scale'),0) as DecimalDigits,
  29. Cast( sys.extended_properties.[value] as nvarchar(2000)) AS [ColumnDescription],
  30. syscomments.text AS DefaultValue,
  31. syscolumns.isnullable AS IsNullable,
  32. columnproperty(syscolumns.id,syscolumns.name,'IsIdentity')as IsIdentity,
  33. (CASE
  34. WHEN EXISTS
  35. (
  36. select 1
  37. from sysindexes i
  38. join sysindexkeys k on i.id = k.id and i.indid = k.indid
  39. join sysobjects o on i.id = o.id
  40. join syscolumns c on i.id=c.id and k.colid = c.colid
  41. where o.xtype = 'U'
  42. and exists(select 1 from sysobjects where xtype = 'PK' and name = i.name)
  43. and o.name=sysobjects.name and c.name=syscolumns.name
  44. ) THEN 1
  45. ELSE 0
  46. END) AS IsPrimaryKey
  47. FROM syscolumns
  48. INNER JOIN systypes ON syscolumns.xtype = systypes.xtype
  49. LEFT JOIN sysobjects ON syscolumns.id = sysobjects.id
  50. LEFT OUTER JOIN sys.extended_properties ON (sys.extended_properties.minor_id = syscolumns.colid
  51. AND sys.extended_properties.major_id = syscolumns.id)
  52. LEFT OUTER JOIN syscomments ON syscolumns.cdefault = syscomments.id
  53. WHERE syscolumns.id IN
  54. (SELECT id
  55. FROM sysobjects
  56. WHERE upper(xtype) IN('U',
  57. 'V') )
  58. AND (systypes.name <> 'sysname')
  59. AND sysobjects.name=N'{0}'
  60. AND systypes.name<>'geometry'
  61. AND systypes.name<>'geography'
  62. ORDER BY syscolumns.colid";
  63. return sql;
  64. }
  65. }
  66. protected override string GetTableInfoListSql
  67. {
  68. get
  69. {
  70. return @"SELECT s.Name,Convert(nvarchar(max),tbp.value) as Description
  71. FROM sysobjects s
  72. LEFT JOIN sys.extended_properties as tbp ON s.id=tbp.major_id and tbp.minor_id=0 AND (tbp.Name='MS_Description' OR tbp.Name is null) WHERE s.xtype IN('U') ";
  73. }
  74. }
  75. protected override string GetViewInfoListSql
  76. {
  77. get
  78. {
  79. return @"SELECT s.Name,Convert(varchar(max),tbp.value) as Description
  80. FROM sysobjects s
  81. LEFT JOIN sys.extended_properties as tbp ON s.id=tbp.major_id and tbp.minor_id=0 AND (tbp.Name='MS_Description' OR tbp.Name is null) WHERE s.xtype IN('V') ";
  82. }
  83. }
  84. #endregion
  85. #region DDL
  86. protected override string CreateDataBaseSql
  87. {
  88. get
  89. {
  90. return @"create database {0} ";
  91. }
  92. }
  93. protected override string AddPrimaryKeySql
  94. {
  95. get
  96. {
  97. return "ALTER TABLE {0} ADD CONSTRAINT {1} PRIMARY KEY({2})";
  98. }
  99. }
  100. protected override string AddColumnToTableSql
  101. {
  102. get
  103. {
  104. return "ALTER TABLE {0} ADD {1} {2}{3} {4} {5} {6}";
  105. }
  106. }
  107. protected override string AlterColumnToTableSql
  108. {
  109. get
  110. {
  111. return "ALTER TABLE {0} ALTER COLUMN {1} {2}{3} {4} {5} {6}";
  112. }
  113. }
  114. protected override string BackupDataBaseSql
  115. {
  116. get
  117. {
  118. return @"USE master;BACKUP DATABASE {0} TO disk = '{1}'";
  119. }
  120. }
  121. protected override string CreateTableSql
  122. {
  123. get
  124. {
  125. return "CREATE TABLE {0}(\r\n{1})";
  126. }
  127. }
  128. protected override string CreateTableColumn
  129. {
  130. get
  131. {
  132. return "{0} {1}{2} {3} {4} {5}";
  133. }
  134. }
  135. protected override string TruncateTableSql
  136. {
  137. get
  138. {
  139. return "TRUNCATE TABLE {0}";
  140. }
  141. }
  142. protected override string BackupTableSql
  143. {
  144. get
  145. {
  146. return "SELECT TOP {0} * INTO {1} FROM {2}";
  147. }
  148. }
  149. protected override string DropTableSql
  150. {
  151. get
  152. {
  153. return "DROP TABLE {0}";
  154. }
  155. }
  156. protected override string DropColumnToTableSql
  157. {
  158. get
  159. {
  160. return "ALTER TABLE {0} DROP COLUMN {1}";
  161. }
  162. }
  163. protected override string DropConstraintSql
  164. {
  165. get
  166. {
  167. return "ALTER TABLE {0} DROP CONSTRAINT {1}";
  168. }
  169. }
  170. protected override string RenameColumnSql
  171. {
  172. get
  173. {
  174. return "exec sp_rename '{0}.{1}','{2}','column';";
  175. }
  176. }
  177. protected override string AddColumnRemarkSql
  178. {
  179. get
  180. {
  181. return "EXECUTE sp_addextendedproperty N'MS_Description', N'{2}', N'user', N'dbo', N'table', N'{1}', N'column', N'{0}'"; ;
  182. }
  183. }
  184. protected override string DeleteColumnRemarkSql
  185. {
  186. get
  187. {
  188. return "EXEC sp_dropextendedproperty 'MS_Description','user',dbo,'table','{1}','column','{0}'";
  189. }
  190. }
  191. protected override string IsAnyColumnRemarkSql
  192. {
  193. get
  194. {
  195. return @"SELECT" +
  196. " A.name AS table_name," +
  197. " B.name AS column_name," +
  198. " C.value AS column_description" +
  199. " FROM sys.tables A" +
  200. " LEFT JOIN sys.extended_properties C ON C.major_id = A.object_id" +
  201. " LEFT JOIN sys.columns B ON B.object_id = A.object_id AND C.minor_id = B.column_id" +
  202. " INNER JOIN sys.schemas SC ON SC.schema_id = A.schema_id AND SC.name = 'dbo'" +
  203. " WHERE A.name = '{1}' and B.name = '{0}'";
  204. }
  205. }
  206. protected override string AddTableRemarkSql
  207. {
  208. get
  209. {
  210. return "EXECUTE sp_addextendedproperty N'MS_Description', N'{1}', N'user', N'dbo', N'table', N'{0}', NULL, NULL";
  211. }
  212. }
  213. protected override string DeleteTableRemarkSql
  214. {
  215. get
  216. {
  217. return "EXEC sp_dropextendedproperty 'MS_Description','user',dbo,'table','{0}' ";
  218. }
  219. }
  220. protected override string IsAnyTableRemarkSql
  221. {
  222. get
  223. {
  224. return @"SELECT C.class_desc
  225. FROM sys.tables A
  226. LEFT JOIN sys.extended_properties C ON C.major_id = A.object_id
  227. INNER JOIN sys.schemas SC ON SC.schema_id=A.schema_id AND SC.name='dbo'
  228. WHERE A.name = '{0}' AND minor_id=0";
  229. }
  230. }
  231. protected override string RenameTableSql
  232. {
  233. get
  234. {
  235. return "EXEC sp_rename '{0}','{1}'";
  236. }
  237. }
  238. protected override string CreateIndexSql
  239. {
  240. get
  241. {
  242. return "CREATE {3} NONCLUSTERED INDEX Index_{0}_{2} ON {0}({1})";
  243. }
  244. }
  245. protected override string AddDefaultValueSql
  246. {
  247. get
  248. {
  249. return "alter table {0} ADD DEFAULT '{2}' FOR {1}";
  250. }
  251. }
  252. protected override string IsAnyIndexSql
  253. {
  254. get
  255. {
  256. return "select count(*) from sys.indexes where name='{0}'";
  257. }
  258. }
  259. protected override string IsAnyProcedureSql
  260. {
  261. get
  262. {
  263. return "select count(*) from sys.objects where [object_id] = OBJECT_ID(N'sp_GetSubLedgerJoinWithdrawalApplicationRecords') and [type] in (N'P')";
  264. }
  265. }
  266. #endregion
  267. #region Check
  268. protected override string CheckSystemTablePermissionsSql
  269. {
  270. get
  271. {
  272. return "select top 1 id from sysobjects";
  273. }
  274. }
  275. #endregion
  276. #region Scattered
  277. protected override string CreateTableNull
  278. {
  279. get
  280. {
  281. return "NULL";
  282. }
  283. }
  284. protected override string CreateTableNotNull
  285. {
  286. get
  287. {
  288. return "NOT NULL";
  289. }
  290. }
  291. protected override string CreateTablePirmaryKey
  292. {
  293. get
  294. {
  295. return "PRIMARY KEY";
  296. }
  297. }
  298. protected override string CreateTableIdentity
  299. {
  300. get
  301. {
  302. return "IDENTITY(1,1)";
  303. }
  304. }
  305. #endregion
  306. #region Methods
  307. public override List<DbTableInfo> GetSchemaTables(EntityInfo entityInfo)
  308. {
  309. if (entityInfo.DbTableName.Contains(".") && this.Context.CurrentConnectionConfig.DbType == DbType.SqlServer)
  310. {
  311. var schema = entityInfo.DbTableName.Split('.').First();
  312. var isAny = GetSchemas().Any(it => it.EqualCase(schema))||schema.EqualCase("dbo");
  313. if (isAny)
  314. {
  315. var tableInfos = this.Context.Ado.SqlQuery<DbTableInfo>(@"SELECT schem.name+'.'+tb.name Name,tb.Description from
  316. ( SELECT obj.name,Convert(nvarchar(max),prop.value)as Description,obj.schema_id FROM sys.objects obj
  317. LEFT JOIN sys.extended_properties prop
  318. ON obj.object_id=prop.major_id
  319. and prop.minor_id=0
  320. AND (prop.Name='MS_Description' OR prop.Name is null)
  321. WHERE obj.type IN('U')) tb
  322. inner join sys.schemas as schem
  323. on tb.schema_id=schem.schema_id ");
  324. return tableInfos;
  325. }
  326. }
  327. return null;
  328. }
  329. public override bool DropColumn(string tableName, string columnName)
  330. {
  331. if (Regex.IsMatch(tableName, @"^\w+$") && Regex.IsMatch(columnName, @"^\w+$"))
  332. {
  333. var sql = $"SELECT distinct dc.name AS ConstraintName \r\nFROM sys.default_constraints dc\r\nJOIN sys.columns c ON dc.parent_column_id = c.column_id\r\nWHERE dc.parent_object_id = OBJECT_ID('{tableName}')\r\nAND c.name = '{columnName}';";
  334. var checks = this.Context.Ado.SqlQuery<string>(sql);
  335. foreach (var checkName in checks)
  336. {
  337. if (checkName?.ToUpper()?.StartsWith("DF__") == true)
  338. {
  339. this.Context.Ado.ExecuteCommand($"ALTER TABLE {SqlBuilder.GetTranslationColumnName(tableName)} DROP CONSTRAINT {checkName}");
  340. }
  341. }
  342. }
  343. return base.DropColumn(tableName, columnName);
  344. }
  345. public override List<string> GetDbTypes()
  346. {
  347. return this.Context.Ado.SqlQuery<string>(@"SELECT name
  348. FROM sys.types
  349. WHERE is_user_defined = 0;");
  350. }
  351. public override List<string> GetTriggerNames(string tableName)
  352. {
  353. return this.Context.Ado.SqlQuery<string>(@"SELECT DISTINCT sysobjects.name AS TriggerName
  354. FROM sysobjects
  355. JOIN syscomments ON sysobjects.id = syscomments.id
  356. WHERE sysobjects.xtype = 'TR'
  357. AND syscomments.text LIKE '%"+tableName+"%'");
  358. }
  359. public override List<string> GetFuncList()
  360. {
  361. return this.Context.Ado.SqlQuery<string>("SELECT name\r\nFROM sys.objects\r\nWHERE type_desc IN( 'SQL_SCALAR_FUNCTION','SQL_INLINE_TABLE_VALUED_FUNCTION') ");
  362. }
  363. private bool IsAnySchemaTable(string tableName)
  364. {
  365. if (tableName == null||!tableName.Contains(".") )
  366. {
  367. return false;
  368. }
  369. var list = GetSchemas() ?? new List<string>();
  370. list.Add("dbo");
  371. var isAnySchemas = list.Any(it => it.EqualCase(tableName?.Split('.').FirstOrDefault()));
  372. return isAnySchemas;
  373. }
  374. public override bool IsAnyColumnRemark(string columnName, string tableName)
  375. {
  376. if (tableName!=null&&tableName.Contains(".") && tableName.Contains(SqlBuilder.SqlTranslationLeft))
  377. {
  378. tableName =string.Join(".", tableName.Split('.').Select(it => SqlBuilder.GetNoTranslationColumnName(it)));
  379. }
  380. if (IsAnySchemaTable(tableName))
  381. {
  382. var schema = tableName.Split('.').First();
  383. tableName = tableName.Split('.').Last();
  384. var temp = this.IsAnyColumnRemarkSql.Replace("'dbo'", $"'{schema}'");
  385. string sql = string.Format(temp, columnName, tableName);
  386. var dt = this.Context.Ado.GetDataTable(sql);
  387. return dt.Rows != null && dt.Rows.Count > 0;
  388. }
  389. return base.IsAnyColumnRemark(columnName, tableName);
  390. }
  391. public override bool DeleteColumnRemark(string columnName, string tableName)
  392. {
  393. if (tableName != null&&tableName.Contains(".") && tableName.Contains(SqlBuilder.SqlTranslationLeft))
  394. {
  395. tableName = string.Join(".", tableName.Split('.').Select(it => SqlBuilder.GetNoTranslationColumnName(it)));
  396. }
  397. if (IsAnySchemaTable(tableName))
  398. {
  399. var schema = tableName.Split('.').First();
  400. tableName = tableName.Split('.').Last();
  401. var temp = this.DeleteColumnRemarkSql.Replace(",dbo,", $",{schema},");
  402. if (!schema.EqualCase("dbo"))
  403. {
  404. temp = temp.Replace("N'user'", $"N'schema'");
  405. }
  406. string sql = string.Format(temp, columnName, tableName);
  407. this.Context.Ado.ExecuteCommand(sql);
  408. return true;
  409. }
  410. return base.DeleteColumnRemark(columnName, tableName);
  411. }
  412. public override bool AddColumnRemark(string columnName, string tableName, string description)
  413. {
  414. if (tableName != null&&tableName.Contains(".") && tableName.Contains(SqlBuilder.SqlTranslationLeft))
  415. {
  416. tableName = string.Join(".", tableName.Split('.').Select(it => SqlBuilder.GetNoTranslationColumnName(it)));
  417. }
  418. if (IsAnySchemaTable(tableName))
  419. {
  420. var schema = tableName.Split('.').First();
  421. tableName = tableName.Split('.').Last();
  422. var temp = this.AddColumnRemarkSql.Replace("N'dbo'", $"N'{schema}'");
  423. if (!schema.EqualCase("dbo"))
  424. {
  425. temp= temp.Replace("N'user'", $"N'schema'");
  426. }
  427. string sql = string.Format(temp, columnName, tableName, description);
  428. this.Context.Ado.ExecuteCommand(sql);
  429. return true;
  430. }
  431. return base.AddColumnRemark(columnName, tableName, description);
  432. }
  433. public override void AddDefaultValue(EntityInfo entityInfo)
  434. {
  435. var dbColumns = this.GetColumnInfosByTableName(entityInfo.DbTableName, false);
  436. var db = this.Context;
  437. var columns = entityInfo.Columns.Where(it => it.IsIgnore == false).ToList();
  438. foreach (var item in columns)
  439. {
  440. if (item.DefaultValue.HasValue() || (item.DefaultValue == "" && item.UnderType == UtilConstants.StringType))
  441. {
  442. if (!IsAnyDefaultValue(entityInfo.DbTableName, item.DbColumnName, dbColumns))
  443. {
  444. this.AddDefaultValue(entityInfo.DbTableName, item.DbColumnName, item.DefaultValue);
  445. }
  446. }
  447. }
  448. }
  449. public override List<string> GetIndexList(string tableName)
  450. {
  451. return this.Context.Ado.SqlQuery<string>($"SELECT indexname = i.name FROM sys.indexes i\r\nJOIN sys.index_columns ic ON i.object_id = ic.object_id AND i.index_id = ic.index_id\r\nJOIN sys.columns c ON ic.object_id = c.object_id AND ic.column_id = c.column_id\r\nWHERE i.object_id = OBJECT_ID('{tableName}')");
  452. }
  453. public override List<string> GetProcList(string dbName)
  454. {
  455. var sql = $"SELECT name FROM {dbName}.sys.procedures";
  456. return this.Context.Ado.SqlQuery<string>(sql);
  457. }
  458. public override bool UpdateColumn(string tableName, DbColumnInfo column)
  459. {
  460. ConvertCreateColumnInfo(column);
  461. if (column.DataType != null && this.Context.CurrentConnectionConfig?.MoreSettings?.SqlServerCodeFirstNvarchar == true)
  462. {
  463. if (!column.DataType.ToLower().Contains("nvarchar"))
  464. {
  465. column.DataType = column.DataType.ToLower().Replace("varchar", "nvarchar");
  466. }
  467. }
  468. return base.UpdateColumn(tableName, column);
  469. }
  470. public override bool IsAnyTable(string tableName, bool isCache = true)
  471. {
  472. if (tableName.Contains("."))
  473. {
  474. var schemas = GetSchemas();
  475. var first =this.SqlBuilder.GetNoTranslationColumnName(tableName.Split('.').First());
  476. var schemaInfo= schemas.FirstOrDefault(it => it.EqualCase(first));
  477. if (schemaInfo == null)
  478. {
  479. return base.IsAnyTable(tableName, isCache);
  480. }
  481. else
  482. {
  483. var result= this.Context.Ado.GetInt($"select object_id(N'{tableName}')");
  484. return result > 0;
  485. }
  486. }
  487. else if (isCache)
  488. {
  489. return base.IsAnyTable(tableName, isCache);
  490. }
  491. else
  492. {
  493. if (tableName.Contains(SqlBuilder.SqlTranslationLeft))
  494. {
  495. tableName = SqlBuilder.GetNoTranslationColumnName(tableName);
  496. }
  497. var sql = @"IF EXISTS (SELECT * FROM sys.objects with(nolock)
  498. WHERE type='u' AND name=N'"+tableName.ToSqlFilter()+@"')
  499. SELECT 1 AS res ELSE SELECT 0 AS res;";
  500. return this.Context.Ado.GetInt(sql) > 0;
  501. }
  502. }
  503. public List<string> GetSchemas()
  504. {
  505. return this.Context.Ado.SqlQuery<string>("SELECT name FROM sys.schemas where name <> 'dbo'");
  506. }
  507. public override bool DeleteTableRemark(string tableName)
  508. {
  509. string sql = string.Format(this.DeleteTableRemarkSql, tableName);
  510. if (tableName.Contains("."))
  511. {
  512. var schemas = GetSchemas();
  513. var tableSchemas = this.SqlBuilder.GetNoTranslationColumnName(tableName.Split('.').First());
  514. if (schemas.Any(y => y.EqualCase(tableSchemas)))
  515. {
  516. sql = string.Format(this.DeleteTableRemarkSql, this.SqlBuilder.GetNoTranslationColumnName(tableName.Split('.').Last()));
  517. if (tableSchemas.EqualCase("user"))
  518. {
  519. sql = sql.Replace("'user'", "'SCHEMA'").Replace("dbo", $"'{tableSchemas}'");
  520. }
  521. else
  522. {
  523. sql = sql.Replace(",dbo,", $",{tableSchemas},").Replace("'user'", "'SCHEMA'");
  524. }
  525. }
  526. }
  527. this.Context.Ado.ExecuteCommand(sql);
  528. return true;
  529. }
  530. public override bool IsAnyTableRemark(string tableName)
  531. {
  532. string sql = string.Format(this.IsAnyTableRemarkSql, tableName);
  533. if (tableName.Contains("."))
  534. {
  535. var schemas = GetSchemas();
  536. var tableSchemas = this.SqlBuilder.GetNoTranslationColumnName(tableName.Split('.').First());
  537. if (schemas.Any(y => y.EqualCase(tableSchemas)))
  538. {
  539. sql = string.Format(this.IsAnyTableRemarkSql, this.SqlBuilder.GetNoTranslationColumnName(tableName.Split('.').Last()));
  540. sql = sql.Replace("'dbo'", $"'{tableSchemas}'");
  541. }
  542. }
  543. var dt = this.Context.Ado.GetDataTable(sql);
  544. return dt.Rows != null && dt.Rows.Count > 0;
  545. }
  546. public override bool AddTableRemark(string tableName, string description)
  547. {
  548. string sql = string.Format(this.AddTableRemarkSql, tableName, description);
  549. if (tableName.Contains("."))
  550. {
  551. var schemas = GetSchemas();
  552. var tableSchemas =this.SqlBuilder.GetNoTranslationColumnName(tableName.Split('.').First());
  553. if (schemas.Any(y => y.EqualCase(tableSchemas)))
  554. {
  555. sql = string.Format(this.AddTableRemarkSql, this.SqlBuilder.GetNoTranslationColumnName(tableName.Split('.').Last()), description);
  556. if (tableSchemas.EqualCase("user"))
  557. {
  558. sql = sql.Replace("N'user', N'dbo'", $"N'user', '{tableSchemas}'").Replace("N'user'", "N'SCHEMA'");
  559. }
  560. else
  561. {
  562. sql = sql.Replace("N'dbo'", $"N'{tableSchemas}'").Replace("N'user'", "N'SCHEMA'");
  563. }
  564. }
  565. }
  566. this.Context.Ado.ExecuteCommand(sql);
  567. return true;
  568. }
  569. public override bool AddDefaultValue(string tableName, string columnName, string defaultValue)
  570. {
  571. if (defaultValue == "''")
  572. {
  573. defaultValue = "";
  574. }
  575. var template = AddDefaultValueSql;
  576. if (defaultValue != null && defaultValue.Replace(" ","").Contains("()"))
  577. {
  578. template = template.Replace("'{2}'", "{2}");
  579. }
  580. tableName=SqlBuilder.GetTranslationTableName(tableName);
  581. columnName = SqlBuilder.GetTranslationTableName(columnName);
  582. string sql = string.Format(template, tableName, columnName, defaultValue);
  583. this.Context.Ado.ExecuteCommand(sql);
  584. return true;
  585. }
  586. /// <summary>
  587. ///by current connection string
  588. /// </summary>
  589. /// <param name="databaseDirectory"></param>
  590. /// <returns></returns>
  591. public override bool CreateDatabase(string databaseName, string databaseDirectory = null)
  592. {
  593. if (databaseDirectory != null)
  594. {
  595. try
  596. {
  597. if (!FileHelper.IsExistDirectory(databaseDirectory))
  598. {
  599. FileHelper.CreateDirectory(databaseDirectory);
  600. }
  601. }
  602. catch
  603. {
  604. //Databases and sites are not in the same service
  605. }
  606. }
  607. var oldDatabaseName = this.Context.Ado.Connection.Database;
  608. var connection = this.Context.CurrentConnectionConfig.ConnectionString;
  609. if (Regex.Split(connection, oldDatabaseName).Length > 2)
  610. {
  611. var name=Regex.Match(connection, @"database\=\w+|datasource\=\w+",RegexOptions.IgnoreCase).Value;
  612. if (!string.IsNullOrEmpty(name))
  613. {
  614. connection = connection.Replace(name, "database=master");
  615. }
  616. else
  617. {
  618. Check.ExceptionEasy("Failed to create the database. The database name has a keyword. Please change the name", "建库失败,库名存在关键字,请换一个名字");
  619. }
  620. }
  621. else
  622. {
  623. connection = connection.Replace(oldDatabaseName, "master");
  624. }
  625. var newDb = new SqlSugarClient(new ConnectionConfig()
  626. {
  627. DbType = this.Context.CurrentConnectionConfig.DbType,
  628. IsAutoCloseConnection = true,
  629. ConnectionString = connection
  630. });
  631. if (!GetDataBaseList(newDb).Any(it => it.Equals(databaseName, StringComparison.CurrentCultureIgnoreCase)))
  632. {
  633. var separatorChar = UtilMethods.GetSeparatorChar();
  634. var sql = CreateDataBaseSql;
  635. if (databaseDirectory.HasValue())
  636. {
  637. sql += @"on primary
  638. (
  639. name = N'{0}',
  640. filename = N'{1}\{0}.mdf',
  641. size = 10mb,
  642. maxsize = 5000mb,
  643. filegrowth = 1mb
  644. ),
  645. (
  646. name = N'{0}_ndf',
  647. filename = N'{1}\{0}.ndf',
  648. size = 10mb,
  649. maxsize = 5000mb,
  650. filegrowth =10mb
  651. )
  652. log on
  653. (
  654. name = N'{0}_log',
  655. filename = N'{1}\{0}.ldf',
  656. size = 100mb,
  657. maxsize = 1gb,
  658. filegrowth = 10mb
  659. ); ";
  660. databaseDirectory = databaseDirectory.Replace("\\", separatorChar);
  661. }
  662. if (databaseName.Contains("."))
  663. {
  664. databaseName = $"[{databaseName}]";
  665. }
  666. else if (Regex.IsMatch(databaseName,@"^\d.*"))
  667. {
  668. databaseName = $"[{databaseName}]";
  669. }
  670. newDb.Ado.ExecuteCommand(string.Format(sql, databaseName, databaseDirectory));
  671. }
  672. return true;
  673. }
  674. public override bool CreateTable(string tableName, List<DbColumnInfo> columns, bool isCreatePrimaryKey = true)
  675. {
  676. tableName = this.SqlBuilder.GetTranslationTableName(tableName);
  677. foreach (var item in columns)
  678. {
  679. ConvertCreateColumnInfo(item);
  680. if (item.DataType == "decimal" && item.DecimalDigits == 0 && item.Length == 0)
  681. {
  682. item.DecimalDigits = 4;
  683. item.Length = 18;
  684. }
  685. else if (item.DataType != null && this.Context.CurrentConnectionConfig?.MoreSettings?.SqlServerCodeFirstNvarchar == true)
  686. {
  687. if (!item.DataType.ToLower().Contains("nvarchar"))
  688. {
  689. item.DataType = item.DataType.ToLower().Replace("varchar", "nvarchar");
  690. }
  691. }
  692. }
  693. string sql = GetCreateTableSql(tableName, columns);
  694. this.Context.Ado.ExecuteCommand(sql);
  695. if (isCreatePrimaryKey)
  696. {
  697. var pkColumns = columns.Where(it => it.IsPrimarykey).ToList();
  698. if (pkColumns.Count > 1)
  699. {
  700. this.Context.DbMaintenance.AddPrimaryKeys(tableName, pkColumns.Select(it => it.DbColumnName).ToArray());
  701. }
  702. else
  703. {
  704. foreach (var item in pkColumns)
  705. {
  706. this.Context.DbMaintenance.AddPrimaryKey(tableName, item.DbColumnName);
  707. }
  708. }
  709. }
  710. return true;
  711. }
  712. private static void ConvertCreateColumnInfo(DbColumnInfo x)
  713. {
  714. string[] array = new string[] { "int", "text", "image", "smallint", "bigint", "date", "bit", "ntext", "datetime" };
  715. if (x.DataType.EqualCase( "nvarchar") || x.DataType .EqualCase( "varchar"))
  716. {
  717. if (x.Length < 1)
  718. {
  719. x.DataType = $"{x.DataType}(max)";
  720. }
  721. }
  722. else if (array.Contains(x.DataType?.ToLower()))
  723. {
  724. x.Length = 0;
  725. x.DecimalDigits = 0;
  726. }
  727. }
  728. public override List<DbColumnInfo> GetColumnInfosByTableName(string tableName, bool isCache = true)
  729. {
  730. tableName = SqlBuilder.GetNoTranslationColumnName(tableName);
  731. var result= base.GetColumnInfosByTableName(tableName, isCache);
  732. return result;
  733. }
  734. public override bool RenameColumn(string tableName, string oldColumnName, string newColumnName)
  735. {
  736. tableName = this.SqlBuilder.GetTranslationTableName(tableName);
  737. oldColumnName = this.SqlBuilder.GetTranslationColumnName(oldColumnName);
  738. newColumnName = this.SqlBuilder.GetNoTranslationColumnName(newColumnName);
  739. string sql = string.Format(this.RenameColumnSql, tableName, oldColumnName, newColumnName);
  740. this.Context.Ado.ExecuteCommand(sql);
  741. return true;
  742. }
  743. #endregion
  744. }
  745. }