问题描述:
假设在数据库中存在以下两张数据表:
User表,存放用户的基本信息,基本结构如下所示:
类型 | 说明 | |
ID_User | int | 自动增长字段,用作该表的主键 |
UserName | varchar |
UserDepart表,存放用户所拥有的部门(我们假设一个用户拥有多个部门,虽然听起来有点别扭,此处仅作示例,可以理解为一个用户拥有多个职位等等),该表的基本结构如下所示:
类型 | 说明 | |
ID_UserDepart | int | 自动增长字段,用作该表的主键 |
ID_User | int | 用户编号 |
ID_Depart | int | 部门编号 |
向数据库中插入一条用户信息的时候,为了保证数据的一致性,必须使用事务的方式“同时”操作User表和UserDepart表。先将用户姓名写入User表中,再将其所拥有的部门写入UserDepart表中,使用事务机制保证这两步操作要么同时成功,要么同时失败。问题就出在:第一步操作完成后,我们并不知道该向第二步操作写入的ID_User的值是多少,因为这个值是SQL Server自动生成的。
解决思路:
可以借助 SELECT IDENT_CURRENT('User') AS ‘NewInsertID’ 来查询最近一次插入User表的数据的自动编号的值。
程序实现:
1 public struct Chaos_TranSQLCmd
2 {
3 /// <summary>
4 /// 一条SQL语句
5 /// </summary>
6 public string strSQL;
7
8 /// <summary>
9 /// 标记该条SQL语句是否需要连接ID_User
10 /// </summary>
11 public bool bNeedID;
12 }
13
14
15 public void Chaos_ExecuteSqlTran(List<Chaos_TranSQLCmd> listTranSQLCmd,string strInsertID_SQL)
16 {
17 using (SqlConnection ChaosSqlConn = new SqlConnection(strSqlConnString))
18 {
19 SqlCommand ChaosSqlCmd = new SqlCommand();
20 ChaosSqlCmd.Connection = ChaosSqlConn;
21 ChaosSqlConn.Open();
22 SqlTransaction ChaosSqlTran = ChaosSqlConn.BeginTransaction();
23 ChaosSqlCmd.Transaction = ChaosSqlTran;
24
25 try
26 {
27 string mID_User = "";
28
29 //先将数据插入User
30 ChaosSqlCmd.CommandText = strInsertID_SQL;
31 ChaosSqlCmd.ExecuteNonQuery();
32
33 //再获取ID_User
34 DataSet ds = this.ExecAdapter("select IDENT_CURRENT('PT_User') as 'ID'", "T");
35 DataTable dt = ds.Tables["T"];
36 if (dt.Rows.Count>0)
37 {
38 mID_User = dt.Rows[0]["ID"].ToString();
39 }
40
41 for (int i = 0; i < listTranSQLCmd.Count; i++)
42 {
43 //如果队列中的语句需要连接ID,则处理SQL语句后再执行
44 string strSQL = "";
45 if (listTranSQLCmd[i].bNeedID==true)
46 {
47 strSQL = string.Format(listTranSQLCmd[i].strSQL, mID_User);
48 }
49 else
50 {
51 strSQL = listTranSQLCmd[i].strSQL;
52 }
53
54 ChaosSqlCmd.CommandText = strSQL;
55 ChaosSqlCmd.ExecuteNonQuery();
56 }
57
58 //全部成功执行则提交
59 ChaosSqlTran.Commit();
60 }
61 catch (System.Data.SqlClient.SqlException Ex)
62 {
63 //发生问题则回滚
64 ChaosSqlTran.Rollback();
65 throw new Exception(Ex.Message);
66 }
67 }
68 }
测试代码如下:
1 static void Main(string[] args)
2 {
3 try
4 {
5 List<DB_Operation.ChaosDbOprt.Chaos_TranSQLCmd> Chaos_SQLCmdList = new List<DB_Operation.ChaosDbOprt.Chaos_TranSQLCmd>();
6
7 //构造SQL语句向User表中写入数据
8 string strSQL = "insert into PT_User (UserName) values ('Lee')";
9
10 //构造SQL语句向UserDepart表写入数据
11 for (int i = 0; i < 10; i++)
12 {
13 DB_Operation.ChaosDbOprt.Chaos_TranSQLCmd nCmd = new DB_Operation.ChaosDbOprt.Chaos_TranSQLCmd();
14 if (i==6)
15 {
16 //构造错误SQL语句,使写入数据库的操作不能成功执行
17 nCmd.strSQL = "insert into PT_UserDepart (ID_User,ID_Depart) values ({0}," + "A String which can't be inserted as ID_Depart)";
18 }
19 else
20 {
21 //正常SQL语句
22 nCmd.strSQL = "insert into PT_UserDepart (ID_User,ID_Depart) values ({0}," + i.ToString() + ")";
23 }
24 nCmd.bNeedID = true;
25 Chaos_SQLCmdList.Add(nCmd);
26 }
27
28 DB_Operation.ChaosDbOprt CDO = new DB_Operation.ChaosDbOprt();
29 CDO.Chaos_ExecuteSqlTran(Chaos_SQLCmdList, strSQL);
30
31 Console.WriteLine("数据写入成功!");
32 Console.ReadLine();
33 }
34 catch (Exception ex)
35 {
36 Console.WriteLine("Error:\r\n"+ex.Message);
37 Console.ReadLine();
38 }
39 }
规范化代码如下:
1 #region 实现数据库事务的方法,实现以事务的方式将多条SQL语句同时写入数据库(其中某些语句依赖于第一条语句插入数据库后自动生成的ID)
2
3 public struct Chaos_TranSQLCmd
4 {
5 /// <summary>
6 /// 一条SQL语句,在需要添加ID的地方用"{0}"来代替
7 /// 如:INSERT INTO PT_FeeItemDetails(ID_FeeItem,ID_ExamItem) VALUES ({0},005)等
8 /// </summary>
9 public string strSQL;
10
11 /// <summary>
12 /// 标记该条SQL语句是否需要连接ID
13 /// </summary>
14 public bool bNeedID;
15 }
16 /// <summary>
17 /// 该函数用于实现以事务的方式将多条SQL语句同时写入数据库(其中某些语句依赖于第一条语句插入数据库后自动生成的ID)
18 /// </summary>
19 /// <param name="strInsertID_SQL">需要先插入数据库中以产生ID的SQL语句</param>
20 /// <param name="strTableName">需要首先插入数据库中以产生ID的数据表的名称,如"PT_FeeItem"等</param>
21 /// <param name="listTranSQLCmd">需要连接ID的SQL语句的列表</param>
22 public void Chaos_ExecuteSqlTran_InsertID(string strInsertID_SQL,string strTableName, List<Chaos_TranSQLCmd> listTranSQLCmd)
23 {
24 using (SqlConnection ChaosSqlConn = new SqlConnection(strSqlConnString))
25 {
26 SqlCommand ChaosSqlCmd = new SqlCommand();
27 ChaosSqlCmd.Connection = ChaosSqlConn;
28 ChaosSqlConn.Open();
29 SqlTransaction ChaosSqlTran = ChaosSqlConn.BeginTransaction();
30 ChaosSqlCmd.Transaction = ChaosSqlTran;
31
32 try
33 {
34 string m_strID = "";
35
36 //先将数据插入User
37 ChaosSqlCmd.CommandText = strInsertID_SQL;
38 ChaosSqlCmd.ExecuteNonQuery();
39
40 string strSQL_Tmp = string.Format("SELECT IDENT_CURRENT('{0}') as 'ID'",strTableName);
41 //再获取ID
42 DataSet ds = this.ExecAdapter(strSQL_Tmp, "T");
43 DataTable dt = ds.Tables["T"];
44 if (dt.Rows.Count>0)
45 {
46 m_strID = dt.Rows[0]["ID"].ToString();
47
48 for (int i = 0; i < listTranSQLCmd.Count; i++)
49 {
50 //如果队列中的语句需要连接ID,则处理SQL语句后再执行
51 string strSQL = "";
52 if (listTranSQLCmd[i].bNeedID == true)
53 {
54 strSQL = string.Format(listTranSQLCmd[i].strSQL, m_strID);
55 }
56 else
57 {
58 strSQL = listTranSQLCmd[i].strSQL;
59 }
60
61 ChaosSqlCmd.CommandText = strSQL;
62 ChaosSqlCmd.ExecuteNonQuery();
63 }
64 }
65 else
66 {
67 //如果没有正确获取首先插入语句的ID,则回滚
68 ChaosSqlTran.Rollback();
69 throw new Exception("产生ID语句没有成功执行,后续语句无法继续执行,已回滚!\r\n");
70 }
71
72
73 //全部成功执行则提交
74 ChaosSqlTran.Commit();
75 }
76 catch (System.Data.SqlClient.SqlException Ex)
77 {
78 //发生问题则回滚
79 ChaosSqlTran.Rollback();
80 throw new Exception(Ex.Message);
81 }
82 }
83 }
84 #endregion