Implementing collections support

src/YesSql.Core/Collections/Collection.cs

@@ -0,0 +1,111 @@
+using System;
+using System.Text;
+
+namespace YesSql.Core.Collections
+{
+ public abstract class Collection : IDisposable
+ {
+ IDisposable _scope;
+
+ public Collection()
+ {
+ _scope = CollectionHelper.EnterScope(this);
+ }
+
+ public abstract string GetSafeName();
+ public static string CreateSafeName(string name)
+ {
+ if (name == null)
+ {
+ throw new ArgumentNullException(nameof(name));
+ }
+
+ var builder = new StringBuilder();
+ bool first = true;
+ foreach (var c in name)
+ {
+ if (Char.IsLetter(c) || (!first && Char.IsDigit(c)))
+ {
+ builder.Append(c);
+ }
+
+ first = false;
+ }
+
+ return builder.ToString();
+ }
+
+ public void Dispose()
+ {
+ _scope.Dispose();
+ }
+ }
+
+ public static class CollectionExtensions
+ {
+ public static string GetPrefixedName(this Collection collection, string table)
+ {
+ var name = collection.GetSafeName();
+
+ if (String.IsNullOrEmpty(name))
+ {
+ return table;
+ }
+
+ return String.Concat(name, "_", table);
+ }
+ }
+
+ public class DefaultCollection : Collection
+ {
+ public DefaultCollection()
+ {
+
+ }
+
+ public override string GetSafeName()
+ {
+ return "";
+ }
+ }
+
+ public class TypeCollection<T> : Collection
+ {
+ private readonly string _safeName;
+
+ public TypeCollection()
+ {
+ _safeName = CreateSafeName(typeof(T).Name);
+
+ if(String.IsNullOrEmpty(_safeName))
+ {
+ throw new ArgumentException("Invalid collection name: " + _safeName);
+ }
+ }
+
+ public override string GetSafeName()
+ {
+ return _safeName;
+ }
+ }
+
+ public class NamedCollection : Collection
+ {
+ private readonly string _safeName;
+
+ public NamedCollection(string name)
+ {
+ _safeName = CreateSafeName(name);
+
+ if (String.IsNullOrEmpty(_safeName))
+ {
+ throw new ArgumentException("Invalid collection name: " + _safeName);
+ }
+ }
+
+ public override string GetSafeName()
+ {
+ return _safeName;
+ }
+ }
+}

src/YesSql.Core/Collections/CollectionHelper.cs

@@ -0,0 +1,90 @@
+using System;
+#if NETSTANDARD1_5
+using System.Threading;
+#else
+using System.Runtime.Remoting;
+using System.Runtime.Remoting.Messaging;
+#endif
+
+namespace YesSql.Core.Collections
+{
+ public class CollectionHelper
+ {
+
+#if NETSTANDARD1_5
+ private static readonly AsyncLocal<CollectionStack> _scopes = new AsyncLocal<CollectionStack>();
+
+ internal static CollectionStack Scopes
+ {
+ get { return _scopes.Value; }
+ set { _scopes.Value = value; }
+ }
+#else
+ private static readonly string CollectionDataName = "Collection.Scopes" + AppDomain.CurrentDomain.Id;
+
+ internal static CollectionStack Scopes
+ {
+ get
+ {
+ var handle = CallContext.LogicalGetData(CollectionDataName) as ObjectHandle;
+
+ if (handle == null)
+ {
+ return null;
+ }
+
+ return handle.Unwrap() as CollectionStack;
+ }
+ set
+ {
+ CallContext.LogicalSetData(CollectionDataName, new ObjectHandle(value));
+ }
+ }
+#endif
+ public static Collection Current
+ {
+ get
+ {
+ var scopes = GetOrCreateScopes();
+ return scopes.Peek();
+ }
+ }
+
+ internal static IDisposable EnterScope(Collection collection)
+ {
+ var scopes = GetOrCreateScopes();
+
+ var scopeLease = new ScopeLease(scopes);
+ Scopes = scopes.Push(collection);
+
+ return scopeLease;
+ }
+
+ private static CollectionStack GetOrCreateScopes()
+ {
+ var scopes = Scopes;
+ if (scopes == null)
+ {
+ scopes = CollectionStack.Empty;
+ Scopes = scopes;
+ }
+
+ return scopes;
+ }
+
+ private sealed class ScopeLease : IDisposable
+ {
+ readonly CollectionStack _collectionStack;
+
+ public ScopeLease(CollectionStack collectionStack)
+ {
+ _collectionStack = collectionStack;
+ }
+
+ public void Dispose()
+ {
+ Scopes = _collectionStack;
+ }
+ }
+ }
+}

src/YesSql.Core/Collections/CollectionStack.cs

@@ -0,0 +1,38 @@
+using System;
+
+namespace YesSql.Core.Collections
+{
+ internal class CollectionStack
+ {
+ private readonly CollectionStack _previous;
+ private readonly Collection _collection;
+
+ private CollectionStack()
+ {
+
+ }
+
+ private CollectionStack(CollectionStack previous, Collection collection)
+ {
+ if (previous == null)
+ {
+ throw new ArgumentNullException(nameof(previous));
+ }
+
+ _previous = previous;
+ _collection = collection;
+ }
+
+ public static readonly CollectionStack Empty = new CollectionStack();
+
+ public CollectionStack Push(Collection c)
+ {
+ return new CollectionStack(this, c);
+ }
+
+ public Collection Peek()
+ {
+ return _collection ?? new DefaultCollection();
+ }
+ }
+}

src/YesSql.Core/Commands/CreateDocumentCommand.cs

@@ -1,7 +1,9 @@
 using System.Data.Common;
 using System.Threading.Tasks;
 using Dapper;
+using YesSql.Core.Collections;
 using YesSql.Core.Indexes;
+using YesSql.Core.Services;
 using YesSql.Core.Sql;
 
 namespace YesSql.Core.Commands
@@ -20,7 +22,8 @@ public CreateDocumentCommand(Document document, string tablePrefix) : base(docum
  public override Task ExecuteAsync(DbConnection connection, DbTransaction transaction)
  {
  var dialect = SqlDialectFactory.For(connection);
- var insertCmd = $"insert into [{_tablePrefix}Document] ([Id], [Type]) values (@Id, @Type);";
+ var documentTable = CollectionHelper.Current.GetPrefixedName(Store.DocumentTable);
+ var insertCmd = $"insert into [{_tablePrefix}{documentTable}] ([Id], [Type]) values (@Id, @Type);";
  return connection.ExecuteScalarAsync<int>(insertCmd, Document, transaction);
  }
  }

src/YesSql.Core/Commands/CreateIndexCommand.cs

@@ -5,6 +5,8 @@
 using System.Threading.Tasks;
 using System.Collections.Generic;
 using System.Data.Common;
+using YesSql.Core.Collections;
+using YesSql.Core.Services;
 
 namespace YesSql.Core.Commands
 {
@@ -26,6 +28,7 @@ public override async Task ExecuteAsync(DbConnection connection, DbTransaction t
  {
  var dialect = SqlDialectFactory.For(connection);
  var type = Index.GetType();
+ var documentTable = CollectionHelper.Current.GetPrefixedName(Store.DocumentTable);
 
  if (Index is MapIndex)
  {
@@ -40,7 +43,7 @@ public override async Task ExecuteAsync(DbConnection connection, DbTransaction t
  var sql = Inserts(type) + $"; {dialect.IdentitySelectString} id";
  Index.Id = await connection.ExecuteScalarAsync<int>(sql, Index, transaction);
 
- var bridgeTableName = type.Name + "_Document";
+ var bridgeTableName = type.Name + "_" + documentTable;
  var columnList = $"[{type.Name}Id], [DocumentId]";
  var parameterList = $"@Id, @DocumentId";
  var bridgeSql = $"insert into [{_tablePrefix}{bridgeTableName}] ({columnList}) values ({parameterList});";

src/YesSql.Core/Commands/DeleteDocumentCommand.cs

@@ -1,7 +1,9 @@
 using System.Data.Common;
 using System.Threading.Tasks;
 using Dapper;
+using YesSql.Core.Collections;
 using YesSql.Core.Indexes;
+using YesSql.Core.Services;
 using YesSql.Core.Sql;
 
 namespace YesSql.Core.Commands
@@ -20,7 +22,8 @@ public DeleteDocumentCommand(Document document, string tablePrefix) : base(docum
  public override Task ExecuteAsync(DbConnection connection, DbTransaction transaction)
  {
  var dialect = SqlDialectFactory.For(connection);
- var deleteCmd = $"delete from [{_tablePrefix}Document] where [Id] = @Id;";
+ var documentTable = CollectionHelper.Current.GetPrefixedName(Store.DocumentTable);
+ var deleteCmd = $"delete from [{_tablePrefix}{documentTable}] where [Id] = @Id;";
  return connection.ExecuteAsync(deleteCmd, Document, transaction);
  }
  }

src/YesSql.Core/Commands/DeleteReduceIndexCommand.cs

@@ -2,6 +2,8 @@
 using System.Data.Common;
 using System.Threading.Tasks;
 using YesSql.Core.Indexes;
+using YesSql.Core.Collections;
+using YesSql.Core.Services;
 
 namespace YesSql.Core.Commands
 {
@@ -17,7 +19,8 @@ public override async Task ExecuteAsync(DbConnection connection, DbTransaction t
  {
  var name = Index.GetType().Name;
 
- var bridgeTableName = name + "_Document";
+ var documentTable = CollectionHelper.Current.GetPrefixedName(Store.DocumentTable);
+ var bridgeTableName = name + "_" + documentTable;
  var bridgeSql = $"delete from [{_tablePrefix}{bridgeTableName}] where {name}Id = @Id";
 
  await connection.ExecuteAsync(bridgeSql, new { Id = Index.Id }, transaction);

src/YesSql.Core/Commands/UpdateIndexCommand.cs

@@ -5,6 +5,8 @@
 using System.Threading.Tasks;
 using YesSql.Core.Indexes;
 using YesSql.Core.Sql;
+using YesSql.Core.Collections;
+using YesSql.Core.Services;
 
 namespace YesSql.Core.Commands
 {
@@ -37,7 +39,8 @@ public override async Task ExecuteAsync(DbConnection connection, DbTransaction t
  var reduceIndex = Index as ReduceIndex;
  if (reduceIndex != null)
  {
- var bridgeTableName = type.Name + "_Document";
+ var documentTable = CollectionHelper.Current.GetPrefixedName(Store.DocumentTable);
+ var bridgeTableName = type.Name + "_" + documentTable;
  var columnList = $"[{type.Name}Id], [DocumentId]";
  var parameterList = $"@Id, @DocumentId";
  var bridgeSqlAdd = $"insert into [{_tablePrefix}{bridgeTableName}] ({columnList}) values ({parameterList});";

src/YesSql.Core/Indexes/IIndexProvider.cs

@@ -6,6 +6,7 @@ public interface IIndexProvider
  {
  void Describe(IDescriptor context);
  Type ForType();
+ string CollectionName { get; set; }
  }
 
  public abstract class IndexProvider<T> : IIndexProvider
@@ -17,6 +18,8 @@ void IIndexProvider.Describe(IDescriptor context)
  Describe((DescribeContext<T>)context);
  }
 
+ public string CollectionName { get; set; }
+
  public Type ForType()
  {
  return typeof(T);