1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117
| using System;
using System.Collections.Generic;
using System.Text;
using System.Data.SqlClient;
using System.Data;
namespace IronViper
{
class DatabaseHandler
{
private SqlConnection connection;
public DatabaseHandler(string connectionString)
{
this.connection = new SqlConnection(connectionString);
this.connection.Open();
Console.WriteLine("Connected to database !");
}
public bool UpdateChannel(string channel)
{
// Let's check if the channel is already in the database.
SqlCommand checkCommand = new SqlCommand("SELECT Count(*) AS Count FROM Channels WHERE Name = @ChannelName", this.connection);
checkCommand.Parameters.Add(new SqlParameter("@ChannelName", SqlDbType.VarChar));
checkCommand.Parameters["@ChannelName"].Value = channel;
SqlDataReader result = checkCommand.ExecuteReader();
result.Read();
int doesItExists = (int)result["Count"];
result.Close();
if (doesItExists == 0)
{
// If not, we register it.
SqlCommand registerCommand = new SqlCommand("INSERT INTO Channels(Name) VALUES (@ChannelName)", this.connection);
SqlTransaction transaction = this.connection.BeginTransaction(IsolationLevel.ReadUncommitted);
registerCommand.Transaction = transaction;
registerCommand.Parameters.Add(new SqlParameter("@ChannelName", SqlDbType.VarChar));
registerCommand.Parameters["@ChannelName"].Value = channel;
registerCommand.ExecuteNonQuery();
transaction.Commit();
return false;
}
else
{
// If yes, we update it
SqlCommand registerCommand = new SqlCommand("UPDATE Channels SET LastActivity=GetDate() WHERE Name = @ChannelName", this.connection);
SqlTransaction transaction = this.connection.BeginTransaction(IsolationLevel.ReadUncommitted);
registerCommand.Transaction = transaction;
registerCommand.Parameters.Add(new SqlParameter("@ChannelName", SqlDbType.VarChar));
registerCommand.Parameters["@ChannelName"].Value = channel;
registerCommand.ExecuteNonQuery();
transaction.Commit();
return true;
}
}
public bool UpdateUser(string user, string host)
{
// Let's check if the user is already in the database.
SqlCommand checkCommand = new SqlCommand("SELECT Count(*) AS Count FROM Users WHERE Name=@UserName AND Host = @HostName", this.connection);
checkCommand.Parameters.Add(new SqlParameter("@UserName", SqlDbType.VarChar));
checkCommand.Parameters.Add(new SqlParameter("@HostName", SqlDbType.VarChar));
checkCommand.Parameters["@UserName"].Value = user;
checkCommand.Parameters["@HostName"].Value = host;
SqlDataReader result = checkCommand.ExecuteReader();
result.Read();
int doesItExists = (int)result["Count"];
result.Close();
if (doesItExists == 0)
{
// If not, we register it.
SqlCommand registerCommand = new SqlCommand("INSERT INTO Users(Name,Host) VALUES (@UserName, @HostName)", this.connection);
SqlTransaction transaction = this.connection.BeginTransaction(IsolationLevel.ReadUncommitted);
registerCommand.Transaction = transaction;
registerCommand.Parameters.Add(new SqlParameter("@UserName", SqlDbType.VarChar));
registerCommand.Parameters.Add(new SqlParameter("@HostName", SqlDbType.VarChar));
registerCommand.Parameters["@UserName"].Value = user;
registerCommand.Parameters["@HostName"].Value = host;
registerCommand.ExecuteNonQuery();
transaction.Commit();
return false;
}
else
{
// If yes, we update it
SqlCommand registerCommand = new SqlCommand("UPDATE Users SET LastActivity=GetDate() WHERE Name=@UserName AND Host = @HostName", this.connection);
SqlTransaction transaction = this.connection.BeginTransaction(IsolationLevel.ReadUncommitted);
registerCommand.Transaction = transaction;
registerCommand.Parameters.Add(new SqlParameter("@UserName", SqlDbType.VarChar));
registerCommand.Parameters.Add(new SqlParameter("@HostName", SqlDbType.VarChar));
registerCommand.Parameters["@UserName"].Value = user;
registerCommand.Parameters["@HostName"].Value = host;
registerCommand.ExecuteNonQuery();
transaction.Commit();
return true;
}
}
public void RegisterMessage(string message, string channel, string user, string host)
{
// We register the message
SqlCommand registerCommand = new SqlCommand("INSERT INTO Messages(Data,UserId,ChannelId) VALUES (@Message, (SELECT Id FROM Users WHERE Name=@UserName AND Host=@HostName), (SELECT Id FROM Channels WHERE Name=@ChannelName)) ", this.connection);
SqlTransaction transaction = this.connection.BeginTransaction(IsolationLevel.ReadUncommitted);
registerCommand.Transaction = transaction;
registerCommand.Parameters.Add(new SqlParameter("@Message", SqlDbType.VarChar));
registerCommand.Parameters.Add(new SqlParameter("@UserName", SqlDbType.VarChar));
registerCommand.Parameters.Add(new SqlParameter("@ChannelName", SqlDbType.VarChar));
registerCommand.Parameters.Add(new SqlParameter("@HostName", SqlDbType.VarChar));
registerCommand.Parameters["@Message"].Value = message;
registerCommand.Parameters["@UserName"].Value = user;
registerCommand.Parameters["@ChannelName"].Value = channel;
registerCommand.Parameters["@HostName"].Value = host;
registerCommand.ExecuteNonQuery();
transaction.Commit();
}
}
} |
Partager