blob: 660cd4dcbc4133ccee30f115ac935ef7bb6c24ac (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
using Dapper;
using Microsoft.Data.Sqlite;
namespace CrupestApi.Commons.Crud;
public class TableInfo
{
// For custom name.
public TableInfo(Type entityType)
: this(entityType.Name, entityType)
{
}
public TableInfo(string tableName, Type entityType)
{
TableName = tableName;
EntityType = entityType;
var properties = entityType.GetProperties();
var columnInfos = new List<ColumnInfo>();
bool hasPrimaryKey = false;
bool hasId = false;
foreach (var property in properties)
{
var columnInfo = new ColumnInfo(entityType, property.Name);
columnInfos.Add(columnInfo);
if (columnInfo.IsPrimaryKey)
hasPrimaryKey = true;
if (columnInfo.SqlColumnName.Equals("id", StringComparison.OrdinalIgnoreCase))
{
hasId = true;
}
}
if (!hasPrimaryKey)
{
if (hasId) throw new Exception("A column named id already exists but is not primary key.");
var columnInfo = new ColumnInfo(entityType, "id", true, true, ColumnTypeInfoRegistry.Singleton.GetRequiredByDataType(typeof(int)));
columnInfos.Add(columnInfo);
}
ColumnInfos = columnInfos;
CheckValidity();
}
public Type EntityType { get; }
public string TableName { get; }
public IReadOnlyList<ColumnInfo> ColumnInfos { get; }
public void CheckValidity()
{
// Check if there is only one primary key.
bool hasPrimaryKey = false;
foreach (var column in ColumnInfos)
{
if (column.IsPrimaryKey)
{
if (hasPrimaryKey) throw new Exception("Two columns are primary key.");
hasPrimaryKey = true;
}
}
if (!hasPrimaryKey) throw new Exception("No column is primary key.");
// Check two columns have the same sql name.
HashSet<string> sqlNameSet = new HashSet<string>();
foreach (var column in ColumnInfos)
{
if (sqlNameSet.Contains(column.SqlColumnName))
throw new Exception($"Two columns have the same sql name '{column.SqlColumnName}'.");
sqlNameSet.Add(column.SqlColumnName);
}
}
public string GenerateCreateTableSql()
{
var tableName = TableName;
var columnSql = string.Join(",\n", ColumnInfos.Select(c => c.GenerateCreateTableColumnString()));
var sql = $@"
CREATE TABLE {tableName}(
{columnSql}
);
";
return sql;
}
public async Task<bool> CheckExistence(SqliteConnection connection)
{
var tableName = TableName;
var count = (await connection.QueryAsync<int>(
@"SELECT count(*) FROM sqlite_schema WHERE type = 'table' AND tbl_name = @TableName;",
new { TableName = tableName })).Single();
if (count == 0)
{
return false;
}
else if (count > 1)
{
throw new DatabaseInternalException($"More than 1 table has name {tableName}. What happened?");
}
else
{
return true;
}
}
}
|
