wvdbi.cs

   1 using System;
   2 using System.Data;
   3 using System.Data.Odbc;
   4 using System.Collections.Specialized;
   5 using Wv.Utils;
   6
   7 namespace Wv.Dbi
   8 {
   9     public class Db
  10     {
  11         IDbConnection db;
  12         static Ini settings = new Ini("wvodbc.ini");
  13
  14         public Db(string odbcstr)
  15         {
  16             string real;
  17
  18             if (settings[odbcstr].Count > 0)
  19             {
  20                 StringDictionary sect = settings[odbcstr];
  21
  22                 string s = wv.fmt("driver={0};server={1};database={2};"
  23                                   + "uid={3};pwd={4};",
  24                                   sect["driver"], sect["server"],
  25                                   sect["database"],
  26                                   sect["user"], sect["password"]);
  27                 real = s;
  28             }
  29             else if (String.Compare(odbcstr, 0, "dsn=", 0, 4, true) == 0)
  30                 real = odbcstr;
  31             else if (String.Compare(odbcstr, 0, "driver=", 0, 7, true) == 0)
  32                 real = odbcstr;
  33             else
  34                 throw new ArgumentException
  35                    ("unrecognized odbc string '" + odbcstr + "'");
  36             db = new OdbcConnection(real);
  37             db.Open();
  38         }
  39
  40         public IDbCommand prepare(string sql, int nparams)
  41         {
  42             IDbCommand cmd = db.CreateCommand();
  43             cmd.CommandText = sql;
  44             for (int i = 0; i < nparams; i++)
  45                 cmd.Parameters.Add(cmd.CreateParameter());
  46             cmd.Prepare();
  47             return cmd;
  48         }
  49
  50         void bind(IDbCommand cmd, params object[] args)
  51         {
  52             int i = 0;
  53             foreach (IDataParameter param in cmd.Parameters)
  54             {
  55                 object o = args[i++];
  56                 if (o is DateTime)
  57                     param.DbType = DbType.DateTime;
  58                 else
  59                     param.DbType = DbType.String; // I sure hope so...
  60                 param.Value = o;
  61             }
  62         }
  63
  64         public IDataReader select(string sql, params object[] args)
  65         {
  66             return select(prepare(sql, args.Length), args);
  67         }
  68
  69         public IDataReader select(IDbCommand cmd, params object[] args)
  70         {
  71             bind(cmd, args);
  72             return cmd.ExecuteReader();
  73         }
  74
  75         public int execute(string sql, params object[] args)
  76         {
  77             return execute(prepare(sql, args.Length), args);
  78         }
  79
  80         public int execute(IDbCommand cmd, params object[] args)
  81         {
  82             bind(cmd, args);
  83             return cmd.ExecuteNonQuery();
  84         }
  85
  86         public int try_execute(string sql, params object[] args)
  87         {
  88             try
  89             {
  90                 return execute(sql, args);
  91             }
  92             catch (OdbcException)
  93             {
  94                 // well, I guess no rows were affected...
  95                 return 0;
  96             }
  97         }
  98     }
  99 }
 100