45 lines
1.5 KiB
C#
45 lines
1.5 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Data;
|
|
using System.Linq;
|
|
using System.Text;
|
|
|
|
namespace TShockAPI.DB
|
|
{
|
|
public static class DbExt
|
|
{
|
|
public static IDbDataParameter AddParameter(this IDbCommand command, string name, object data)
|
|
{
|
|
var parm = command.CreateParameter();
|
|
parm.ParameterName = name;
|
|
parm.Value = data;
|
|
command.Parameters.Add(parm);
|
|
return parm;
|
|
}
|
|
|
|
static Dictionary<Type, Func<IDataReader, int, object>> ReadFuncs = new Dictionary<Type, Func<IDataReader, int, object>>()
|
|
{
|
|
{typeof(bool), (s, i) => s.GetBoolean(i)},
|
|
{typeof(byte), (s, i) => s.GetByte(i)},
|
|
{typeof(Int16), (s, i) => s.GetInt16(i)},
|
|
{typeof(Int32), (s, i) => s.GetInt32(i)},
|
|
{typeof(Int64), (s, i) => s.GetInt64(i)},
|
|
{typeof(string), (s, i) => s.GetString(i)},
|
|
{typeof(decimal), (s, i) => s.GetDecimal(i)},
|
|
{typeof(decimal), (s, i) => s.GetFloat(i)},
|
|
{typeof(double), (s, i) => s.GetDouble(i)},
|
|
};
|
|
|
|
public static T Get<T>(this IDataReader reader, string column)
|
|
{
|
|
return reader.Get<T>(column);
|
|
}
|
|
public static T Get<T>(this IDataReader reader, int column)
|
|
{
|
|
if (ReadFuncs.ContainsKey(typeof(T)))
|
|
return (T)ReadFuncs[typeof(T)](reader, column);
|
|
|
|
throw new NotImplementedException();
|
|
}
|
|
}
|
|
}
|