HYTOS

using System;

using System.Collections.Generic;

using System.Linq;

using System.Text;

using System.Threading.Tasks;

using System.Data;

using System.Data.SqlClient;

using System.ComponentModel.DataAnnotations.Schema;

using Dapper;

namespace ID2.Manager.Dapper.Repository

{

    public abstract class BaseRepository : IBaseRepository

    {

        protected readonly IDbConnection _DbConnection;//readonly 삭제해야하는지 확인

        protected IDbTransaction _DbTransaction = null;

        public SqlConnection DBConnection(string connectionStr)

        {

            return new SqlConnection(connectionStr);

        }

        public IDbTransaction BeginTransaction()

        {

            if (this._DbConnection.State != ConnectionState.Open)

            {

                this._DbConnection.Open();

            }

            return this._DbConnection.BeginTransaction();

        }

        public void Dispose()

        {

            if (this._DbTransaction != null)

            {

                this._DbTransaction.Commit();

            }

            if (this._DbConnection != null)

            {

                this._DbConnection.Close();

            }

        }

        protected BaseRepository(string connectionStr)

        {

            this._DbConnection = this.DBConnection(connectionStr);

            SqlMapper.TypeMapProvider = type =>

            {

                // create fallback default type map

                var fallback = new DefaultTypeMap(type);

                return new CustomPropertyTypeMap(type, (t, column) =>

                {

                    var property = t.GetProperties().FirstOrDefault(prop =>

                        prop.GetCustomAttributes(typeof(ColumnAttribute), true)

                            .Cast<ColumnAttribute>()

                            .Any(attr => attr.Name == column));

                    // if no property matched - fall back to default type map

                    if (property == null)

                    {

                        property = fallback.GetMember(column)?.Property;

                    }

                    return property;

                });

            };

        }

        public IEnumerable<T> Query<T>(string sql)

        {

            if (this._DbConnection.State != ConnectionState.Open)

            {

                this._DbConnection.Open();

            }

            return this._DbConnection.Query<T>(sql);

        }

        public IEnumerable<T> Query<T>(string sql, object param)

        {

            if (this._DbConnection.State != ConnectionState.Open)

            {

                this._DbConnection.Open();

            }

            return this._DbConnection.Query<T>(sql, param);

        }

        public async Task<IEnumerable<T>> QueryAsync<T>(string sql)

        {

            if (this._DbConnection.State != ConnectionState.Open)

            {

                this._DbConnection.Open();

            }

            return await this._DbConnection.QueryAsync<T>(sql);

        }

        public async Task<IEnumerable<T>> QueryAsync<T>(string sql, object param)

        {

            if (this._DbConnection.State != ConnectionState.Open)

            {

                this._DbConnection.Open();

            }

            return await this._DbConnection.QueryAsync<T>(sql, param);

        }

        public int Execute(string query, IDbTransaction dbTran)

        {

            return this.Execute(query, null, dbTran);

        }

        public int Execute(string query, object param, IDbTransaction dbTran)

        {

            if (this._DbConnection.State != ConnectionState.Open)

            {

                this._DbConnection.Open();

            }

            return this._DbConnection.Execute(query, param, dbTran);

        }

        public async Task<int> ExecuteAsync(string query, object param, IDbTransaction dbTran)

        {

            if (this._DbConnection.State != ConnectionState.Open)

            {

                this._DbConnection.Open();

            }

            return await this._DbConnection.ExecuteAsync(query, param, dbTran);

        }

    }

}