You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.2 KiB
88 lines
2.2 KiB
6 years ago
|
using System;
|
||
|
using System.Collections.Concurrent;
|
||
|
using System.Collections.Generic;
|
||
|
using System.Threading;
|
||
|
using SQLitePCL.pretty;
|
||
|
|
||
|
namespace Emby.Server.Implementations.Data
|
||
|
{
|
||
|
public class ManagedConnection : IDisposable
|
||
|
{
|
||
|
private SQLiteDatabaseConnection _db;
|
||
|
private SemaphoreSlim _writeLock;
|
||
|
private BlockingCollection<SQLiteDatabaseConnection> _readConPool;
|
||
|
private bool _disposed = false;
|
||
|
|
||
|
public ManagedConnection(SQLiteDatabaseConnection db, SemaphoreSlim writeLock)
|
||
|
{
|
||
|
_db = db;
|
||
|
_writeLock = writeLock;
|
||
|
}
|
||
|
|
||
|
public ManagedConnection(SQLiteDatabaseConnection db, BlockingCollection<SQLiteDatabaseConnection> queue)
|
||
|
{
|
||
|
_db = db;
|
||
|
_readConPool = queue;
|
||
|
}
|
||
|
|
||
|
public IStatement PrepareStatement(string sql)
|
||
|
{
|
||
|
return _db.PrepareStatement(sql);
|
||
|
}
|
||
|
|
||
|
public IEnumerable<IStatement> PrepareAll(string sql)
|
||
|
{
|
||
|
return _db.PrepareAll(sql);
|
||
|
}
|
||
|
|
||
|
public void ExecuteAll(string sql)
|
||
|
{
|
||
|
_db.ExecuteAll(sql);
|
||
|
}
|
||
|
|
||
|
public void Execute(string sql, params object[] values)
|
||
|
{
|
||
|
_db.Execute(sql, values);
|
||
|
}
|
||
|
|
||
|
public void RunQueries(string[] sql)
|
||
|
{
|
||
|
_db.RunQueries(sql);
|
||
|
}
|
||
|
|
||
|
public void RunInTransaction(Action<IDatabaseConnection> action, TransactionMode mode)
|
||
|
{
|
||
|
_db.RunInTransaction(action, mode);
|
||
|
}
|
||
|
|
||
|
public T RunInTransaction<T>(Func<IDatabaseConnection, T> action, TransactionMode mode)
|
||
|
{
|
||
|
return _db.RunInTransaction(action, mode);
|
||
|
}
|
||
|
|
||
|
public IEnumerable<IReadOnlyList<IResultSetValue>> Query(string sql)
|
||
|
{
|
||
|
return _db.Query(sql);
|
||
|
}
|
||
|
|
||
|
public IEnumerable<IReadOnlyList<IResultSetValue>> Query(string sql, params object[] values)
|
||
|
{
|
||
|
return _db.Query(sql, values);
|
||
|
}
|
||
|
|
||
|
public void Dispose()
|
||
|
{
|
||
|
if (_disposed)
|
||
|
{
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
_writeLock?.Release();
|
||
|
_readConPool?.Add(_db);
|
||
|
|
||
|
_db = null; // Don't dispose it
|
||
|
_disposed = true;
|
||
|
}
|
||
|
}
|
||
|
}
|