Skip to content

Commit 478c63a

Browse files
author
jackyzhou
committed
代码优化,更新表名,支持数据结果写入
1 parent a78b554 commit 478c63a

File tree

3 files changed

+17
-3
lines changed

3 files changed

+17
-3
lines changed

SQLToolkit/SQLToolkit/Business/DatabaseVersion.cs

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,15 +9,22 @@ public static class DatabaseVersion
99
public static int Init()
1010
{
1111
return Helper.DapperHelper.Execute(@"
12-
IF NOT EXISTS (SELECT * FROM sysobjects WHERE name='DatabaseVersion')
13-
CREATE TABLE DatabaseVersion(
14-
[Id] [int] PRIMARY KEY,
12+
IF NOT EXISTS (SELECT * FROM sysobjects WHERE name='ST_DatabaseVersion')
13+
CREATE TABLE ST_DatabaseVersion(
14+
ID int IDENTITY(1,1) PRIMARY KEY,
1515
[Filename] [nvarchar](MAX) NULL,
1616
[ExecuteResult] [nvarchar](MAX) NULL,
1717
[ExecuteTime] [nvarchar](MAX) NULL,
1818
)");
1919
}
2020

21+
public static int UpdateRecord(string filename, string result)
22+
{
23+
var sql = string.Format(@"INSERT INTO ST_DatabaseVersion (Filename, ExecuteResult, ExecuteTime)
24+
VALUES ('{0}', '{1}', '{2}');", filename, "success", DateTime.Now.ToString());
25+
return Helper.DapperHelper.Execute(sql);
26+
}
27+
2128

2229
}
2330
}

SQLToolkit/SQLToolkit/Helper/DapperHelper.cs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,5 +16,7 @@ public static int Execute(string sql)
1616
return conn.Execute(sql);
1717
}
1818
}
19+
20+
1921
}
2022
}

SQLToolkit/SQLToolkit/Program.cs

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,14 +57,19 @@ static void Main(string[] args)
5757
foreach (string file in sqlFiles)
5858
{
5959
Helper.LogHelper.Log(string.Format("Ready to exec sql script:{0}", file));
60+
var filename = Path.GetFileName(file);
6061

6162
try
6263
{
6364
server.ConnectionContext.ExecuteNonQuery(File.ReadAllText(file));
65+
Business.DatabaseVersion.UpdateRecord(filename, "success");
66+
6467
}
6568
catch (Exception ex)
6669
{
6770
Helper.LogHelper.Log(string.Format("Error:{0}", ex.ToString()));
71+
Business.DatabaseVersion.UpdateRecord(filename, "fail");
72+
6873
}
6974

7075
Helper.LogHelper.Log(string.Format("Finish to exec sql script:{0}", file));

0 commit comments

Comments
 (0)