using JiShe.CollectBus.IotSystems.Devices;
using JiShe.CollectBus.IotSystems.MessageIssueds;
using JiShe.CollectBus.IotSystems.MessageReceiveds;
using JiShe.CollectBus.IotSystems.MeterReadingRecords;
using JiShe.CollectBus.IotSystems.Protocols;
using MongoDB.Bson;
using MongoDB.Driver;
using System;
using System.Collections.Generic;
using Volo.Abp.Data;
using Volo.Abp.MongoDB;
using Volo.Abp.MultiTenancy;

namespace JiShe.CollectBus.MongoDB;

[IgnoreMultiTenancy]
[ConnectionStringName(CollectBusDbProperties.MongoDbConnectionStringName)]
public class CollectBusMongoDbContext : AbpMongoDbContext, ICollectBusMongoDbContext
{
    /* Add mongo collections here. Example:
     * public IMongoCollection<Question> Questions => Collection<Question>();
     */

    public IMongoCollection<MessageReceived> MessageReceiveds => Collection<MessageReceived>();
    public IMongoCollection<MessageReceivedLogin> MessageReceivedLogins => Collection<MessageReceivedLogin>();
    public IMongoCollection<MessageReceivedHeartbeat> MessageReceivedHeartbeats => Collection<MessageReceivedHeartbeat>();
    public IMongoCollection<Device> Devices => Collection<Device>();
    public IMongoCollection<ProtocolInfo> ProtocolInfos => Collection<ProtocolInfo>();

    /// <summary>
    /// 抄表记录，默认按天分表
    /// </summary>
    public IMongoCollection<MeterReadingRecords> MeterReadingRecordInfo => Database.GetCollection<MeterReadingRecords>(DateTime.Now.GetCollectionName<MeterReadingRecords>());


    protected override void CreateModel(IMongoModelBuilder modelBuilder)
    {
        base.CreateModel(modelBuilder);

        modelBuilder.Entity<MeterReadingRecords>(builder =>
        {
            // 创建索引
            builder.ConfigureIndexes(index =>
            {
                List<CreateIndexModel<BsonDocument>> createIndexModels = new List<CreateIndexModel<BsonDocument>>();
                createIndexModels.Add(new CreateIndexModel<BsonDocument>(
                        Builders<BsonDocument>.IndexKeys.Ascending(nameof(MeterReadingRecords)),
                        new CreateIndexOptions
                        {
                            Unique = true
                        }
                    ));
                index.CreateMany(createIndexModels);
            });

        });

        modelBuilder.ConfigureCollectBus();
    }
}