72 lines
3.5 KiB
C#
72 lines
3.5 KiB
C#
|
using System.Data;
|
|||
|
|
using Dapper;
|
|||
|
|
using Microsoft.Extensions.Configuration;
|
|||
|
|
using Npgsql;
|
|||
|
|
using UserOfTheDayBot.Model;
|
|||
|
|
|
|||
|
|
namespace UserOfTheDayBot.Data;
|
|||
|
|
|
|||
|
|
public interface IUserOfTheDayRepository
|
|||
|
|
{
|
|||
|
|
Task RecordUserOfTheDayAsync(long chatId, long userId, DateTime date, UserOfTheDayType type);
|
|||
|
|
Task<bool> IsUserOfTheDaySelectedAsync(long chatId, DateTime date, UserOfTheDayType type);
|
|||
|
|
Task<bool> IsUserAlreadySelectedAsync(long chatId, DateTime date, long userId);
|
|||
|
|
Task<Dictionary<(long userId, string userName), (int userOfTheDayCount, int loserOfTheDayCount)>> GetUserStatisticsAsync(long chatId);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public class UserOfTheDayRepository : IUserOfTheDayRepository
|
|||
|
|
{
|
|||
|
|
private readonly string _connectionString;
|
|||
|
|
|
|||
|
|
public UserOfTheDayRepository(IConfiguration configuration)
|
|||
|
|
{
|
|||
|
|
_connectionString = configuration["DatabaseSettings:ConnectionString"];
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public async Task RecordUserOfTheDayAsync(long chatId, long userId, DateTime date, UserOfTheDayType type)
|
|||
|
|
{
|
|||
|
|
using (IDbConnection db = new NpgsqlConnection(_connectionString))
|
|||
|
|
{
|
|||
|
|
string query = "INSERT INTO user_of_the_day (chat_id, user_id, date, type) VALUES (@ChatId, @UserId, @Date, @Type)";
|
|||
|
|
await db.ExecuteAsync(query, new { ChatId = chatId, UserId = userId, Date = date, Type = type });
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public async Task<bool> IsUserOfTheDaySelectedAsync(long chatId, DateTime date, UserOfTheDayType type)
|
|||
|
|
{
|
|||
|
|
using (IDbConnection db = new NpgsqlConnection(_connectionString))
|
|||
|
|
{
|
|||
|
|
string query = "SELECT COUNT(1) FROM user_of_the_day WHERE chat_id = @ChatId AND date = @Date AND type = @Type";
|
|||
|
|
return await db.ExecuteScalarAsync<bool>(query, new { ChatId = chatId, Date = date, Type = type });
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public async Task<bool> IsUserAlreadySelectedAsync(long chatId, DateTime date, long userId)
|
|||
|
|
{
|
|||
|
|
using (IDbConnection db = new NpgsqlConnection(_connectionString))
|
|||
|
|
{
|
|||
|
|
string query = "SELECT COUNT(1) FROM user_of_the_day WHERE chat_id = @ChatId AND date = @Date AND user_id = @UserId";
|
|||
|
|
return await db.ExecuteScalarAsync<bool>(query, new { ChatId = chatId, Date = date, UserId = userId });
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public async Task<Dictionary<(long userId, string userName), (int userOfTheDayCount, int loserOfTheDayCount)>> GetUserStatisticsAsync(long chatId)
|
|||
|
|
{
|
|||
|
|
using (IDbConnection db = new NpgsqlConnection(_connectionString))
|
|||
|
|
{
|
|||
|
|
string query = @"SELECT u.user_id, u.user_name,
|
|||
|
|
SUM(CASE WHEN ud.type = 0 THEN 1 ELSE 0 END) AS user_of_the_day_count,
|
|||
|
|
SUM(CASE WHEN ud.type = 1 THEN 1 ELSE 0 END) AS loser_of_the_day_count
|
|||
|
|
FROM user_of_the_day ud
|
|||
|
|
JOIN users u ON ud.user_id = u.user_id AND ud.chat_id = u.chat_id
|
|||
|
|
WHERE ud.chat_id = @ChatId
|
|||
|
|
GROUP BY u.user_id, u.user_name";
|
|||
|
|
|
|||
|
|
var results = await db.QueryAsync<(long userId, string userName, int userOfTheDayCount, int loserOfTheDayCount)>(query, new { ChatId = chatId });
|
|||
|
|
return results.ToDictionary(
|
|||
|
|
row => (row.userId, row.userName),
|
|||
|
|
row => (row.userOfTheDayCount, row.loserOfTheDayCount)
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|