1
0
mirror of https://github.com/bitwarden/server synced 2026-01-07 19:13:50 +00:00

set up postgresql repository

This commit is contained in:
Kyle Spearrin
2019-01-15 21:55:42 -05:00
parent 8596ba2caa
commit 84800da1fb
5 changed files with 133 additions and 1 deletions

View File

@@ -1,7 +1,7 @@
using System;
using Dapper;
namespace Bit.Core.Repositories.SqlServer
namespace Bit.Core.Repositories
{
public abstract class BaseRepository
{

View File

@@ -0,0 +1,28 @@
using System.Text.RegularExpressions;
using Dapper;
namespace Bit.Core.Repositories.PostgreSql
{
public abstract class BasePostgreSqlRepository : BaseRepository
{
static BasePostgreSqlRepository()
{
// Support snake case property names
DefaultTypeMap.MatchNamesWithUnderscores = true;
}
public BasePostgreSqlRepository(string connectionString, string readOnlyConnectionString)
: base(connectionString, readOnlyConnectionString)
{ }
protected static string SnakeCase(string input)
{
if(string.IsNullOrWhiteSpace(input))
{
return input;
}
var startUnderscores = Regex.Match(input, @"^_+");
return startUnderscores + Regex.Replace(input, @"([a-z0-9])([A-Z])", "$1_$2").ToLowerInvariant();
}
}
}

View File

@@ -0,0 +1,89 @@
using System;
using System.Data;
using System.Linq;
using System.Threading.Tasks;
using Dapper;
using Bit.Core.Models.Table;
using Npgsql;
namespace Bit.Core.Repositories.PostgreSql
{
public abstract class Repository<T, TId> : BasePostgreSqlRepository, IRepository<T, TId>
where TId : IEquatable<TId>
where T : class, ITableObject<TId>
{
public Repository(string connectionString, string readOnlyConnectionString, string table)
: base(connectionString, readOnlyConnectionString)
{
if(!string.IsNullOrWhiteSpace(table))
{
Table = table;
}
else
{
Table = SnakeCase(typeof(T).Name);
}
}
protected string Table { get; private set; }
public virtual async Task<T> GetByIdAsync(TId id)
{
using(var connection = new NpgsqlConnection(ConnectionString))
{
var results = await connection.QueryAsync<T>(
$"{Table}_read_by_id",
new { id = id },
commandType: CommandType.StoredProcedure);
return results.SingleOrDefault();
}
}
public virtual async Task CreateAsync(T obj)
{
obj.SetNewId();
using(var connection = new NpgsqlConnection(ConnectionString))
{
var results = await connection.ExecuteAsync(
$"{Table}_create",
obj,
commandType: CommandType.StoredProcedure);
}
}
public virtual async Task ReplaceAsync(T obj)
{
using(var connection = new NpgsqlConnection(ConnectionString))
{
var results = await connection.ExecuteAsync(
$"{Table}_update",
obj,
commandType: CommandType.StoredProcedure);
}
}
public virtual async Task UpsertAsync(T obj)
{
if(obj.Id.Equals(default(TId)))
{
await CreateAsync(obj);
}
else
{
await ReplaceAsync(obj);
}
}
public virtual async Task DeleteAsync(T obj)
{
using(var connection = new NpgsqlConnection(ConnectionString))
{
await connection.ExecuteAsync(
$"{Table}_delete_by_id",
new { id = obj.Id },
commandType: CommandType.StoredProcedure);
}
}
}
}