src/Library/Consul.AspNetCore/ServiceCollectionExtensions.cs

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

using System;
using System.Linq;
using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Hosting.Server.Features;
using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.DependencyInjection.Extensions;
using Microsoft.Extensions.Options;

namespace Consul.AspNetCore
{
    public static class ServiceCollectionExtensions
    {
        /// <summary>
        /// Add default <see cref="IConsulClient"/>.
        /// First client can be accessed from DI, to create multiple named clients use <see cref="IConsulClientFactory"/>
        /// </summary>
        public static IServiceCollection AddConsul(this IServiceCollection services)
        {
            return services.AddConsul(options => { });
        }

        /// <summary>
        /// Add default <see cref="IConsulClient"/> with configured <see cref="IOptions{ConsulClientConfiguration}"/>.
        /// First client can be accessed from DI, to create multiple named clients use <see cref="IConsulClientFactory"/>
        /// </summary>
        public static IServiceCollection AddConsul(
            this IServiceCollection services,
            Action<ConsulClientConfiguration> configure)
        {
            return services.AddConsul(Options.DefaultName, configure);
        }

        /// <summary>
        /// Add named <see cref="IConsulClient"/> with configured <see cref="IOptions{ConsulClientConfiguration}"/>.
        /// First client can be accessed from DI, to create multiple named clients use <see cref="IConsulClientFactory"/>
        /// </summary>
        public static IServiceCollection AddConsul(
            this IServiceCollection services,
            string name,
            Action<ConsulClientConfiguration> configure)
        {
            services.Configure(name, configure);
            services.TryAddSingleton<IConsulClientFactory, ConsulClientFactory>();
            services.TryAddSingleton(sp => sp.GetRequiredService<IConsulClientFactory>().CreateClient(name));

            return services;
        }

        /// <summary>
        /// Register consul service with default and support only ttl <see cref="IConsulClient"/>.
        /// First client can be accessed from DI, to create multiple named clients use <see cref="IConsulClientFactory"/>
        /// </summary>
        public static IServiceCollection AddConsulServiceRegistration(
            this IServiceCollection services,
            Action<AgentServiceRegistration> configure)
        {
            var registration = new AgentServiceRegistration();

            configure.Invoke(registration);