JiShe.CollectBus/src/JiShe.CollectBus.Application/Samples/SampleAppService.cs

using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using Apache.IoTDB.DataStructure;
using Apache.IoTDB;
using Confluent.Kafka;
using JiShe.CollectBus.Ammeters;
using JiShe.CollectBus.FreeSql;
using JiShe.CollectBus.IoTDBProvider;
using JiShe.CollectBus.IotSystems.PrepayModel;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Options;
using JiShe.CollectBus.IoTDBProvider.Context;
using Microsoft.Extensions.Logging;
using JiShe.CollectBus.Common.Helpers;

namespace JiShe.CollectBus.Samples;

public class SampleAppService : CollectBusAppService, ISampleAppService
{
    private readonly ILogger<SampleAppService> _logger;
    private readonly IIoTDBProvider _iotDBProvider;
    private readonly IoTDBRuntimeContext _dbContext;
    private readonly IoTDBOptions _options;

    public SampleAppService(IIoTDBProvider iotDBProvider, IOptions<IoTDBOptions> options,
        IoTDBRuntimeContext dbContext, ILogger<SampleAppService> logger)
    {
        _iotDBProvider = iotDBProvider;
        _options = options.Value;
        _dbContext = dbContext;
        _logger = logger;
    }


    [HttpGet]
    public async Task UseSessionPool(long timestamps)
    {
        string? messageHexString = null;
        if (timestamps == 0)
        {
            timestamps = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
            _logger.LogError($"timestamps_{timestamps}");
        }
        else
        {
            messageHexString = messageHexString + timestamps;
        }

        ElectricityMeter meter = new ElectricityMeter()
        {
            SystemName = "energy",
            DeviceId = "402440506",
            DeviceType = "Ammeter",
            Current = 10,
            MeterModel = "DDZY-1980",
            ProjectCode = "10059",
            Voltage = 10,
            IssuedMessageHexString = messageHexString,
            Timestamps = timestamps,
        };
        await _iotDBProvider.InsertAsync(meter);
    }


    [HttpGet]
    public async Task UseTableSessionPool()
    {
        //_dbContext.UseTableSessionPool = true;
        _iotDBProvider.SwitchSessionPool(true);

        ElectricityMeter meter = new ElectricityMeter()
        {
            SystemName = "energy",
            DeviceId = "402440506",
            DeviceType = "Ammeter",
            Current = 10,
            MeterModel = "DDZY-1980",
            ProjectCode = "10059",
            Voltage = 10,
            Timestamps = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds(),
        };
        await _iotDBProvider.InsertAsync(meter);
    }

    /// <summary>
    /// 测试设备分组均衡控制算法
    /// </summary>
    /// <param name="deviceCount"></param>
    /// <returns></returns>
    [HttpGet]
    public async Task TestDeviceGroupBalanceControl(int deviceCount = 200000)
    {
        var deviceList = new List<string>();
        for (int i = 0; i < deviceCount; i++)
        {
            deviceList.Add($"Device_{Guid.NewGuid()}");
        }

        // 初始化缓存
        DeviceGroupBalanceControl.InitializeCache(deviceList);

        // 打印分布统计
        DeviceGroupBalanceControl.PrintDistributionStats();

        await Task.CompletedTask;
    }


    public Task<SampleDto> GetAsync()
    {
        return Task.FromResult(
            new SampleDto
            {
                Value = 42
            }
        );
    }

    [Authorize]
    public Task<SampleDto> GetAuthorizedAsync()
    {
        return Task.FromResult(
            new SampleDto
            {
                Value = 42
            }
        );
    }

    [AllowAnonymous]
    public async Task<List<Vi_BaseAmmeterInfo>> Test()
    {

        var ammeterList = await SqlProvider.Instance.Change(DbEnum.PrepayDB).Select<Vi_BaseAmmeterInfo>().Where(d => d.TB_CustomerID == 5).Take(10).ToListAsync();
        return ammeterList;
    }
}