訪問數據庫的泛型封裝類

  /// <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();
                }
            }
        }