using JiShe.CollectBus.IotSystems.Devices;
using JiShe.CollectBus.IotSystems.MessageReceiveds;
using JiShe.CollectBus.IotSystems.MeterReadingRecords;
using JiShe.CollectBus.IotSystems.Protocols;
using JiShe.CollectBus.ShardingStrategy;
using MongoDB.Bson;
using MongoDB.Bson.Serialization;
using MongoDB.Driver;
using System;
using System.Collections.Concurrent;
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>();
       
    protected override void CreateModel(IMongoModelBuilder modelBuilder)
    {
        base.CreateModel(modelBuilder); 
        modelBuilder.ConfigureCollectBus();

        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
                //        }
                //    ));


                //var indexKeys = Builders<BsonDocument>.IndexKeys
                //.Ascending("CreationTime")
                //.Ascending("OrderNumber");

                //var indexOptions = new CreateIndexOptions
                //{
                //    Background = true,
                //    Name = "IX_CreationTime_OrderNumber"
                //};
                //index.CreateOne(
                //new CreateIndexModel<BsonDocument>(indexKeys, indexOptions));

                index.CreateOne(new CreateIndexModel<BsonDocument>(
                        Builders<BsonDocument>.IndexKeys.Ascending(nameof(MeterReadingRecords)),
                        new CreateIndexOptions
                        {
                            Unique = true
                        }
                    ));
            });

        });
    } 
}