自己动手写ORM框架(八):实现修改功能Update方法

    前面一节中已经实现了简单的新增功能，下面来实现修改功能，代码如下1-1：

#region 将实体数据修改到数据库
public int Update<T>(T entity)
{
    TableInfo tableInfo = DbEntityUtils.GetTableInfo(entity,DbOperateType.UPDATE);

    string strSql = DbEntityUtils.GetUpdateSql(tableInfo);

    IDbDataParameter[] parms = DbFactory.CreateDbParameters(tableInfo.Columns.Count);

    DbEntityUtils.SetParameters(tableInfo.Columns, parms);

    object val = AdoHelper.ExecuteNonQuery(transaction, CommandType.Text, strSql, parms);

    return Convert.ToInt32(val);            
}
#endregion

    在1-1代码中，绝大多数代码都和前面一节中的Save方法相同，这里主要看一下string strSql = DbEntityUtils.GetUpdateSql(tableInfo); 这段代码1-2：

public static string GetUpdateSql(TableInfo tableInfo)
{
    StringBuilder sbBody = new StringBuilder();

    foreach (string key in tableInfo.Columns.Keys)
    {
        object value = tableInfo.Columns[key];

        sbBody.Append(key).Append("=").Append(AdoHelper.DbParmChar + key).Append(",");
    }

    sbBody.Remove(sbBody.ToString().Length - 1, 1);

    tableInfo.Columns.Put(tableInfo.Id.Key, tableInfo.Id.Value);

    string strSql = "update {0} set {1} where {2} =" + AdoHelper.DbParmChar + tableInfo.Id.Key;
    strSql = string.Format(strSql, tableInfo.TableName, sbBody.ToString(), tableInfo.Id.Key);

    return strSql;
}

    上面代码就是通过tableInfo.Columns集合中的列名，组合成字符串，如strSql = “update student set studentno = @studentno,name = @name,sex = @sex where studentid = @studentid”; 然后在Update方法中通过参数数组将值传入并作更新操作。

Technorati 标签: ORM