Cosmos/source2/Debug/Cosmos.Debug.Common/SQL.cs

using System;
using System.Collections.Generic;
using System.Data.Objects;
using System.Data.SqlClient;
using System.Linq;
using System.Text;

namespace Cosmos.Debug.Common {
  public class SQL {
    public readonly SqlConnection Connection;

    public SQL(SqlConnection aConnection) {
      Connection = aConnection;
    }

    public int? SelectOneInt(string aSql) {
      var xQry = Connection.CreateCommand();
      xQry.CommandText = aSql;
      return (int)xQry.ExecuteScalar();
    }

    public void Exec(string aSql) {
      var xQry = Connection.CreateCommand();
      xQry.CommandText = aSql;
      xQry.ExecuteNonQuery();
    }

    public void MakeUniqueInt(string aTable, string aCol, bool aAllowNulls) {
      string xIdxName = "Uq_" + aTable + "_" + aCol;
      if (aAllowNulls) {
        //http://sqlservercodebook.blogspot.com/2008/04/multiple-null-values-in-unique-index-in.html
        Exec("ALTER TABLE " + aTable + " ADD " + xIdxName + " AS (CASE WHEN " + aCol + " IS NULL THEN Id * -1 ELSE " + aCol + " END);");
        // We create index seperately instead of using unique keyword on previous SQL
        // so we can control name of the index for better identification
        Exec("CREATE UNIQUE INDEX Idx" + xIdxName + " ON " + aTable + "(" + xIdxName + ");");
      } else {
        Exec("CREATE UNIQUE INDEX Idx" + xIdxName + " ON " + aTable + "(" + aCol + ");");
      }
    }

    public void MakeIndex(string aTable, string aCol, bool aUnique) {
      Exec("CREATE " + (aUnique ? "UNIQUE " : "") + "INDEX Idx_" + aTable + "_" + aCol + " ON " + aTable + "(" + aCol + ");");
    }

    public void MakeUniqueString(string aTable, string aCol, bool aAllowNulls) {
      string xIdxName = "Uq_" + aTable + "_" + aCol;
      if (aAllowNulls) {
        Exec("ALTER TABLE " + aTable + " ADD " + xIdxName + " AS (CASE WHEN " + aCol + " IS NULL THEN 'key-' + CAST([Id] as varchar(10)) ELSE UPPER(RTRIM(LTRIM(" + aCol + "))) END);");
        Exec("CREATE UNIQUE INDEX Idx" + xIdxName + " ON " + aTable + "(" + xIdxName + ");");
      } else {
        Exec("CREATE UNIQUE INDEX Idx" + xIdxName + " ON " + aTable + "(" + aCol + ");");
      }
    }

  }
}