访问数据库的泛型封装类

  /// <summary>
        /// 新增
        /// </summary>
        /// <typeparam name="T">实体类</typeparam>
        /// <param name="t">对象</param>
        /// <param name="flag">返回主键还是都影响行数  false  受影响行数  true 主键值</param>
        /// <returns></returns>
        public int Insert<T>(T t, bool flag = false) 
        {
            List<PropertyInfo> propertyInfos = new List<PropertyInfo>(typeof(T).GetProperties());

            //获取主键
            PropertyInfo pkey = propertyInfos.Where(p => p.GetCustomAttributes(typeof(KeyAttribute), false).Length > 0).FirstOrDefault();

            propertyInfos.Remove(pkey);

            StringBuilder builder = new StringBuilder();
            builder.Append("insert into ").Append(typeof(T).Name).Append("(");

            StringBuilder valueBuilder = new StringBuilder();
            valueBuilder.Append(" values( ");

            for (int i = 0; i < propertyInfos.Count; i++)
            {
                PropertyInfo item = propertyInfos[i];
                if (item.GetValue(t) != null)
                {
                    builder.Append(item.Name);
                    valueBuilder.Append("'");
                    valueBuilder.Append(item.GetValue(t));
                    valueBuilder.Append("'");

                    if (i < propertyInfos.Count - 1)
                    {
                        builder.Append(",");
                        valueBuilder.Append(",");
                    }
                }
            }

            builder.Append(" ) ");
            valueBuilder.Append(" )");

            if (flag)
                builder.Append(" output inserted.").Append(pkey.Name);

            builder.Append(valueBuilder);

            using (SqlConnection conn = new SqlConnection(_connstr))
            {
                conn.Open();
                using (SqlCommand cmd = new SqlCommand(builder.ToString(), conn))
                {
                    if (flag)
                        return Convert.ToInt32(cmd.ExecuteScalar());//返回最新的主键
                    else
                        return cmd.ExecuteNonQuery();//返回受影响行数
                }
            }

        }

        /// <summary>
        /// 修改
        /// </summary>
        /// <typeparam name="T"></typeparam>
        /// <param name="t"></param>
        /// <returns></returns>
        public int Update<T>(T t)
        {
            List<PropertyInfo> propertyInfos = new List<PropertyInfo>(typeof(T).GetProperties());

            //获取主键
            PropertyInfo pkey = propertyInfos.Where(p => p.GetCustomAttributes(typeof(KeyAttribute), false).Length > 0).FirstOrDefault();

            propertyInfos.Remove(pkey);

            StringBuilder builder = new StringBuilder();
            builder.Append("update ").Append(typeof(T).Name).Append(" set ");

            for (int i = 0; i < propertyInfos.Count; i++)
            {
                PropertyInfo item = propertyInfos[i];
                if (item.GetValue(t) != null)
                {
                    builder.Append(item.Name);
                    builder.Append("='");
                    builder.Append(item.GetValue(t));
                    builder.Append("'");

                    if (i < propertyInfos.Count - 1)
                        builder.Append(",");
                }
            }

            builder.Append(" where ");
            builder.Append(pkey.Name);
            builder.Append("='");
            builder.Append(pkey.GetValue(t));
            builder.Append("'");

            using (SqlConnection conn = new SqlConnection(_connstr))
            {
                conn.Open();
                using (SqlCommand cmd = new SqlCommand(builder.ToString(), conn))
                {
                    return cmd.ExecuteNonQuery();
                }
            }
        }

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值