配置中心重构主键ConfigId
智能大石头 authored at 2024-04-01 23:12:36
2.15 KiB
Stardust
using NewLife;
using NewLife.Configuration;
using NewLife.Log;
using NewLife.Threading;
using Stardust.Data.Configs;
using XCode;

namespace Stardust.Server.Services;

public class ApolloService : IHostedService
{
    private TimerX _timer;

    public Task StartAsync(CancellationToken cancellationToken)
    {
        _timer = new TimerX(DoApolloSync, null, 5_000, 300_000) { Async = true };

        return Task.CompletedTask;
    }

    public Task StopAsync(CancellationToken cancellationToken)
    {
        _timer.TryDispose();

        return Task.CompletedTask;
    }

    private void DoApolloSync(Object state)
    {
        foreach (var item in AppConfig.FindAllWithCache())
        {
            if (!item.Enable || !item.EnableApollo || item.ApolloMetaServer.IsNullOrEmpty()) continue;

            var httpConfig = new ApolloConfigProvider
            {
                Server = item.ApolloMetaServer,
                AppId = item.ApolloAppId
            };

            var nameSpace = item.ApolloNameSpace;
            if (nameSpace.IsNullOrEmpty()) nameSpace = "application";
            httpConfig.SetApollo(nameSpace);

            try
            {
                // 一次性加载所有配置
                httpConfig.LoadAll();

                if (httpConfig.Keys.Count > 0)
                {
                    // 配置匹配到本地
                    var ds = ConfigData.FindAllByApp(item.Id);
                    foreach (var elm in httpConfig.Keys)
                    {
                        var cfg = ds.FirstOrDefault(e => e.Key.EqualIgnoreCase(elm));
                        if (cfg == null)
                        {
                            cfg = new ConfigData { ConfigId = item.Id, Key = elm, Enable = true, };
                            ds.Add(cfg);
                        }

                        cfg.Value = httpConfig[elm];

                        if (cfg is IEntity entity && entity.HasDirty) cfg.Version = item.AcquireNewVersion();
                    }
                    ds.Save(true);
                }
            }
            catch (Exception ex)
            {
                XTrace.WriteException(ex);
            }
        }
    }
}