使用.NET6打造動態API

known發表於2021-11-02

ApiLite是直接將Service層自動生成api路由,可以不用新增Controller,支援模組外掛化,在專案開發中能夠提高工作效率,降低程式碼量。

開發環境

  • .NET SDK 6.0.100-rc.2.21505.57
  • VS2022 Preview 7.0

示例地址

示例目標

  • 根據Service動態生成api
  • 支援自定義路由模板(通過Route特性定義)
  • 支援模組外掛化
  • 支援不同模組,相同Service名稱的路由(名稱空間需要有3級以上,例如:Com.Mod.XXX)
  • 自動根據方法名稱判斷請求方式,Get開頭的方法名為GET請求,其他為POST請求

編碼約定

  • 模組類庫必須包含繼承IModule介面的類
  • 需要生成api的Service必須繼承IService介面
  • GET請求的方法必須以Get開頭

核心程式碼

主要是ApiFeatureProvider和ApiConvention這兩個自定義類來動態生成api,ApiFeatureProvider繼承ControllerFeatureProvider,覆寫IsController方法,判斷服務型別是否符合Controller。ApiConvention實現了IApplicationModelConvention介面,動態新增Action。下面是主要程式碼,完整程式碼請在GitHub上下載。

static class ServiceExtension
{
    internal static WebApplicationBuilder AddKApp(this WebApplicationBuilder builder, Action<AppOption>? action = null)
    {
        var option = new AppOption();
        action?.Invoke(option);
        ...
        AddDynamicApi(mvcBuilder, option);//新增動態api
        return builder;
    }

    private static void AddDynamicApi(IMvcBuilder builder, AppOption option)
    {
        builder.ConfigureApplicationPartManager(m =>
        {
            m.ApplicationParts.Add(new AssemblyPart(typeof(IService).Assembly));
            foreach (var item in option.Modules)
            {
                item.Initialize();//初始化模組
                //將模組新增到ApplicationParts,這樣才能發現服務類
                var assembly = item.GetType().Assembly;
                m.ApplicationParts.Add(new AssemblyPart(assembly));
            }
            m.FeatureProviders.Add(new ApiFeatureProvider());
        });

        builder.Services.Configure<MvcOptions>(o =>
        {
            o.Conventions.Add(new ApiConvention());
        });
    }

    internal static WebApplication UseKApp(this WebApplication app)
    {
        ...
        return app;
    }
}

//判斷服務型別是否為Controller
class ApiFeatureProvider : ControllerFeatureProvider
{
    protected override bool IsController(TypeInfo typeInfo)
    {
        if (!typeof(IService).IsAssignableFrom(typeInfo) ||
            !typeInfo.IsPublic ||
            typeInfo.IsAbstract ||
            typeInfo.IsGenericType)
            return false;

        return true;
    }
}

class ApiConvention : IApplicationModelConvention
{
    public void Apply(ApplicationModel application)
    {
        foreach (var controller in application.Controllers)
        {
            var type = controller.ControllerType;
            if (typeof(IService).IsAssignableFrom(type))
            {
                ConfigureApiExplorer(controller);
                ConfigureSelector(controller);
            }
        }
    }

    ...

    //構造路由模板
    private string GetRouteTemplate(ActionModel action)
    {
        if (action.Attributes != null && action.Attributes.Count > 0)
        {
            foreach (var item in action.Attributes)
            {
                if (item is RouteAttribute attribute)
                {
                    return attribute.Path;//返回自定義路由
                }
            }
        }

        var routeTemplate = new StringBuilder();
        //routeTemplate.Append("api");
        var names = action.Controller.ControllerType.Namespace.Split('.');
        if (names.Length > 2)
        {
            //支援不同模組相同類名,新增名稱空間模組名作字首
            routeTemplate.Append(names[^2]);
        }

        // Controller
        var controllerName = action.Controller.ControllerName;
        if (controllerName.EndsWith("Service"))
            controllerName = controllerName[0..^7];

        routeTemplate.Append($"/{controllerName}");

        // Action
        var actionName = action.ActionName;
        if (actionName.EndsWith("Async"))
            actionName = actionName[..^"Async".Length];

        if (!string.IsNullOrEmpty(actionName))
            routeTemplate.Append($"/{actionName}");

        return routeTemplate.ToString();
    }
}

使用示例

KHost.Run(args, o =>
{
    o.Modules.Add(new TestModule());//新增模組
});

class TestModule : IModule
{
    public void Initialize()
    {
    }
}

public class TestService : IService
{
    public string GetName(string name)
    {
        return $"Hello {name}";
    }

    public string SaveData(string data)
    {
        return $"{DateTime.Now:yyyy-MM-dd HH:mm:ss} {data}";
    }

    [Route("api/test")]
    public string GetCustMethod(string id)
    {
        return id;
    }
}

相關文章