DocSpace-buildtools/common/ASC.Core.Common/EF/Model/Tenant/DbButton.cs

using System.ComponentModel.DataAnnotations.Schema;
using ASC.Core.Common.EF.Model;
using Microsoft.EntityFrameworkCore;

namespace ASC.Core.Common.EF
{
    [Table("tenants_buttons")]
    public class DbButton : BaseEntity
    {
        [Column("button_url")]
        public string ButtonUrl { get; set; }

        [Column("tariff_id")]
        public int TariffId { get; set; }

        [Column("partner_id")]
        public string PartnerId { get; set; }

        public override object[] GetKeys()
        {
            return new object[] { TariffId, PartnerId };
        }
    }

    public static class DbButtonExtension
    {
        public static ModelBuilderWrapper AddDbButton(this ModelBuilderWrapper modelBuilder)
        {
            modelBuilder
                .Add(MySqlAddDbButton, Provider.MySql)
                .Add(PgSqlAddDbButton, Provider.Postrge);
            return modelBuilder;
        }
        public static void MySqlAddDbButton(this ModelBuilder modelBuilder)
        {
            modelBuilder.Entity<DbButton>(entity =>
            {
                entity.HasKey(e => new { e.TariffId, e.PartnerId })
                    .HasName("PRIMARY");

                entity.ToTable("tenants_buttons");

                entity.Property(e => e.TariffId).HasColumnName("tariff_id");

                entity.Property(e => e.PartnerId)
                    .HasColumnName("partner_id")
                    .HasColumnType("varchar(50)")
                    .HasCharSet("utf8")
                    .HasCollation("utf8_general_ci");

                entity.Property(e => e.ButtonUrl)
                    .IsRequired()
                    .HasColumnName("button_url")
                    .HasColumnType("text")
                    .HasCharSet("utf8")
                    .HasCollation("utf8_general_ci");
            });
        }
        public static void PgSqlAddDbButton(this ModelBuilder modelBuilder)
        {
            modelBuilder.Entity<DbButton>(entity =>
            {
                entity.HasKey(e => new { e.TariffId, e.PartnerId })
                    .HasName("tenants_buttons_pkey");

                entity.ToTable("tenants_buttons", "onlyoffice");

                entity.Property(e => e.TariffId).HasColumnName("tariff_id");

                entity.Property(e => e.PartnerId)
                    .HasColumnName("partner_id")
                    .HasMaxLength(50);

                entity.Property(e => e.ButtonUrl)
                    .IsRequired()
                    .HasColumnName("button_url");
            });
        }
    }
}