GrainStorageOptions.cs

using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Metadata;
using Orleans;
using Orleans.Runtime;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace Pole.Orleans.Provider.EntityframeworkCore
{
    public abstract class GrainStorageOptions
    {
        internal string KeyPropertyName { get; set; }

        internal string KeyExtPropertyName { get; set; }

        internal string ETagPropertyName { get; set; }

        internal string PersistenceCheckPropertyName { get; set; }

        internal IProperty ETagProperty { get; set; }

        internal bool CheckForETag { get; set; }

        internal Func<object, string> ConvertETagObjectToStringFunc { get; set; }

        internal Type ETagType { get; set; }

        public bool ShouldUseETag { get; set; }

        internal bool IsConfigured { get; set; }

        internal bool PreCompileReadQuery { get; set; } = true;
    }
    public class GrainStorageOptions<TContext, TGrain, TEntity> : GrainStorageOptions
       where TContext : DbContext
       where TEntity : class
    {
        internal Func<TContext, IQueryable<TEntity>> DbSetAccessor { get; set; }

        internal Func<TContext, IQueryable<TEntity>> DbSetNoTrackingAccessor { get; set; }

        internal Func<TEntity, bool> IsPersistedFunc { get; set; }

        internal Func<TEntity, string> GetETagFunc { get; set; }

        internal Func<TEntity, Guid> GuidKeySelector { get; set; }

        internal Func<TEntity, string> KeyExtSelector { get; set; }

        internal Func<TEntity, long> LongKeySelector { get; set; }

        public bool IsRelatedData { get; set; }

        internal Func<TContext, IAddressable, Task<TEntity>> ReadStateAsync { get; set; }
        internal Func<TContext, IAddressable, Task<TEntity>> ReadStateNoTrackingAsync { get; set; }

        internal Action<IGrainState, TEntity> SetEntity { get; set; }

        internal Func<IGrainState, TEntity> GetEntity { get; set; }
    }
}