通過Dapr實現一個簡單的基於.net的微服務電商系統(十七)——服務保護之動態配置與熱過載

apiVersion: dapr.io/v1alpha1

kind: Component

metadata:

  name: sentinel

  namespace: dapreshop

spec:

  type: middleware.http.sentinel

  version: v1

  metadata:

  - name: appName

    value: "rules"

  - name: logDir

    value: "/tmp"

  - name: flowRules

    value: >-

      []

  - name: circuitbreakerRules

    value: >-

      []

 static Kubernetes kubernetes = new Kubernetes(KubernetesClientConfiguration.BuildConfigFromConfigFile(SentinelComponentBaseConfig.kubeconfig));

        /// <summary>

        /// 註冊規則

        /// </summary>

        /// <param name="aliSentinelConfig"></param>

        public static async Task RegisterSentinelConfig(SentinelConfigList aliSentinelConfigList)

        {

            await GetAndSaveSentinelComponent(component =>

            {

                component.FlowRules = aliSentinelConfigList.FlowRules.GetDistinct();

                component.BreakingRules = aliSentinelConfigList.BreakingRules.GetDistinct();

            });

        }

        /// <summary>

        /// 獲取所有註冊規則

        /// </summary>

        /// <returns></returns>

        public static async Task<SentinelConfigList> GetAll()

        {

            var component = await GetDefaultSentinelComponent();

            return new SentinelConfigList()

            {

                FlowRules = component.FlowRules,

                BreakingRules = component.BreakingRules

            };

        }

        #region 本地方法

        /// <summary>

        /// 獲取預設的SentinelComponent

        /// </summary>

        /// <returns></returns>

        static async Task<SentinelComponent> GetDefaultSentinelComponent()

        {

            var component = new SentinelComponent();

            await component.Create(kubernetes);

            return component;

        }

        /// <summary>

        /// 傳遞委託變更預設SentinelComponent

        /// </summary>

        /// <param name="operatorComponent"></param>

        static async Task GetAndSaveSentinelComponent(Action<SentinelComponent> operatorComponent)

        {

            var component = await GetDefaultSentinelComponent();

            operatorComponent(component);

            component.SetMetaData();

            Patch(component);

            ReloadDeploy();

        }

        /// <summary>

        /// Patch SentinelComponent到k8s環境

        /// </summary>

        /// <param name="component"></param>

        static void Patch(SentinelComponent component)

        {

            var patch = new JsonPatchDocument<SentinelComponent>();

            patch.Replace(x => x.spec.metadata, component.spec.metadata);

            kubernetes.PatchNamespacedCustomObject(new V1Patch(patch, V1Patch.PatchType.JsonPatch), SentinelComponentBaseConfig.Group, SentinelComponentBaseConfig.Version, SentinelComponentBaseConfig.NamespaceParameter, SentinelComponentBaseConfig.Plural, SentinelComponentBaseConfig.ComponentName);

        }

        /// <summary>

        /// 重啟相關deploy更新SentinelComponent

        /// </summary>

        static void ReloadDeploy()

        {

            var deploy = kubernetes.ReadNamespacedDeployment(SentinelComponentBaseConfig.DeploymentName, SentinelComponentBaseConfig.NamespaceParameter);

            deploy.Spec.Template.Metadata.Annotations[SentinelComponentBaseConfig.restart] = DateTime.UtcNow.ToString("s");

            var patch = new JsonPatchDocument<V1Deployment>();

            patch.Replace(e => e.Spec.Template.Metadata.Annotations, deploy.Spec.Template.Metadata.Annotations);

            kubernetes.PatchNamespacedDeployment(new V1Patch(patch, V1Patch.PatchType.JsonPatch), SentinelComponentBaseConfig.DeploymentName, SentinelComponentBaseConfig.NamespaceParameter);

        }