mirror of
https://github.com/ckaczor/HomeMonitor.git
synced 2026-02-12 10:58:30 -05:00
Add power database support
This commit is contained in:
91
Power/Service/Data/Database.cs
Normal file
91
Power/Service/Data/Database.cs
Normal file
@@ -0,0 +1,91 @@
|
||||
using ChrisKaczor.HomeMonitor.Power.Service.Models;
|
||||
using Dapper;
|
||||
using Microsoft.Data.SqlClient;
|
||||
using Microsoft.Extensions.Configuration;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace ChrisKaczor.HomeMonitor.Power.Service.Data
|
||||
{
|
||||
public class Database
|
||||
{
|
||||
private readonly IConfiguration _configuration;
|
||||
|
||||
public Database(IConfiguration configuration)
|
||||
{
|
||||
_configuration = configuration;
|
||||
}
|
||||
|
||||
public void EnsureDatabase()
|
||||
{
|
||||
var connectionStringBuilder = new SqlConnectionStringBuilder
|
||||
{
|
||||
DataSource = $"{_configuration["Power:Database:Host"]},{_configuration["Power:Database:Port"]}",
|
||||
UserID = _configuration["Power:Database:User"],
|
||||
Password = _configuration["Power:Database:Password"],
|
||||
InitialCatalog = "master"
|
||||
};
|
||||
|
||||
using var connection = new SqlConnection(connectionStringBuilder.ConnectionString);
|
||||
|
||||
var command = new SqlCommand { Connection = connection };
|
||||
|
||||
connection.Open();
|
||||
|
||||
// Check to see if the database exists
|
||||
command.CommandText = $"SELECT CAST(1 as bit) from sys.databases WHERE name='{_configuration["Power:Database:Name"]}'";
|
||||
var databaseExists = (bool?)command.ExecuteScalar();
|
||||
|
||||
// Create database if needed
|
||||
if (!databaseExists.GetValueOrDefault(false))
|
||||
{
|
||||
command.CommandText = $"CREATE DATABASE {_configuration["Power:Database:Name"]}";
|
||||
command.ExecuteNonQuery();
|
||||
}
|
||||
|
||||
// Switch to the database now that we're sure it exists
|
||||
connection.ChangeDatabase(_configuration["Power:Database:Name"]);
|
||||
|
||||
var schema = ResourceReader.GetString("ChrisKaczor.HomeMonitor.Power.Service.Data.Resources.Schema.sql");
|
||||
|
||||
// Make sure the database is up to date
|
||||
command.CommandText = schema;
|
||||
command.ExecuteNonQuery();
|
||||
}
|
||||
|
||||
private SqlConnection CreateConnection()
|
||||
{
|
||||
var connectionStringBuilder = new SqlConnectionStringBuilder
|
||||
{
|
||||
DataSource = $"{_configuration["Power:Database:Host"]},{_configuration["Power:Database:Port"]}",
|
||||
UserID = _configuration["Power:Database:User"],
|
||||
Password = _configuration["Power:Database:Password"],
|
||||
InitialCatalog = _configuration["Power:Database:Name"]
|
||||
};
|
||||
|
||||
var connection = new SqlConnection(connectionStringBuilder.ConnectionString);
|
||||
connection.Open();
|
||||
|
||||
return connection;
|
||||
}
|
||||
|
||||
public void StorePowerData(PowerStatus powerStatus)
|
||||
{
|
||||
using var connection = CreateConnection();
|
||||
|
||||
var query = ResourceReader.GetString("ChrisKaczor.HomeMonitor.Power.Service.Data.Resources.CreateStatus.sql");
|
||||
|
||||
connection.Query(query, powerStatus);
|
||||
}
|
||||
|
||||
public async Task<IEnumerable<PowerStatusGrouped>> GetStatusHistoryGrouped(DateTimeOffset start, DateTimeOffset end, int bucketMinutes)
|
||||
{
|
||||
await using var connection = CreateConnection();
|
||||
|
||||
var query = ResourceReader.GetString("ChrisKaczor.HomeMonitor.Power.Service.Data.Resources.GetStatusHistoryGrouped.sql");
|
||||
|
||||
return await connection.QueryAsync<PowerStatusGrouped>(query, new { Start = start, End = end, BucketMinutes = bucketMinutes });
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user