2020-09-02 15:23:39 +00:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
2020-11-17 10:47:17 +00:00
|
|
|
|
using System.Linq;
|
2020-10-14 09:11:04 +00:00
|
|
|
|
using System.Linq.Expressions;
|
2020-09-02 15:23:39 +00:00
|
|
|
|
|
|
|
|
|
using Microsoft.EntityFrameworkCore;
|
2020-10-14 09:11:04 +00:00
|
|
|
|
using Microsoft.EntityFrameworkCore.Query.SqlExpressions;
|
2020-09-02 15:23:39 +00:00
|
|
|
|
|
|
|
|
|
namespace ASC.Core.Common.EF.Model
|
|
|
|
|
{
|
|
|
|
|
public class ModelBuilderWrapper
|
|
|
|
|
{
|
2020-10-05 10:02:28 +00:00
|
|
|
|
private ModelBuilder ModelBuilder { get; set; }
|
|
|
|
|
private Provider Provider { get; set; }
|
2020-09-02 15:23:39 +00:00
|
|
|
|
|
|
|
|
|
private ModelBuilderWrapper(ModelBuilder modelBuilder, Provider provider)
|
|
|
|
|
{
|
|
|
|
|
ModelBuilder = modelBuilder;
|
|
|
|
|
Provider = provider;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static ModelBuilderWrapper From(ModelBuilder modelBuilder, Provider provider)
|
|
|
|
|
{
|
|
|
|
|
return new ModelBuilderWrapper(modelBuilder, provider);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public ModelBuilderWrapper Add(Action<ModelBuilder> action, Provider provider)
|
|
|
|
|
{
|
2020-10-05 10:02:28 +00:00
|
|
|
|
if (provider == Provider)
|
2020-09-02 15:23:39 +00:00
|
|
|
|
{
|
2020-10-05 10:02:28 +00:00
|
|
|
|
action(ModelBuilder);
|
2020-09-02 15:23:39 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return this;
|
|
|
|
|
}
|
|
|
|
|
|
2020-10-05 09:56:07 +00:00
|
|
|
|
public ModelBuilderWrapper HasData<T>(params T[] data) where T : class
|
|
|
|
|
{
|
|
|
|
|
ModelBuilder.Entity<T>().HasData(data);
|
|
|
|
|
|
|
|
|
|
return this;
|
|
|
|
|
}
|
2020-10-14 09:11:04 +00:00
|
|
|
|
|
|
|
|
|
public void AddDbFunction()
|
|
|
|
|
{
|
|
|
|
|
ModelBuilder
|
|
|
|
|
.HasDbFunction(typeof(JsonExtensions).GetMethod(nameof(JsonExtensions.JsonValue)))
|
|
|
|
|
.HasTranslation(e =>
|
|
|
|
|
{
|
|
|
|
|
var res = new List<SqlExpression>();
|
|
|
|
|
if (e is List<SqlExpression> list)
|
|
|
|
|
{
|
|
|
|
|
if (list[0] is SqlConstantExpression key)
|
|
|
|
|
{
|
|
|
|
|
res.Add(new SqlFragmentExpression($"`{key.Value}`"));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (list[1] is SqlConstantExpression val)
|
|
|
|
|
{
|
|
|
|
|
res.Add(new SqlConstantExpression(Expression.Constant($"$.{val.Value}"), val.TypeMapping));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2020-11-17 10:47:17 +00:00
|
|
|
|
return new SqlFunctionExpression("JSON_EXTRACT", res, true, res.Select((SqlExpression a) => false), typeof(string), null);
|
2020-10-14 09:11:04 +00:00
|
|
|
|
});
|
|
|
|
|
}
|
2020-09-02 15:23:39 +00:00
|
|
|
|
}
|
|
|
|
|
}
|