1 using System.Web.Http;
 2 using WebActivatorEx;
 3 using DemoAPI;
 4 using Swashbuckle.Application;
 5 using Swashbuckle.Swagger;
 6 using System.Collections.Concurrent;
 7 using System.Collections.Generic;
 8 using System.Xml;
 9 using System.IO;
10
11 [assembly: PreApplicationStartMethod(typeof(SwaggerConfig), "Register")]
12
13 namespace DemoAPI
14 {
15     public class SwaggerConfig
16     {
17         public static void Register()
18         {
19             var thisAssembly = typeof(SwaggerConfig).Assembly;
20
21             GlobalConfiguration.Configuration
22                .EnableSwagger(c =>
23                {
24                    c.SingleApiVersion("v1", "DemoAPI");
25                    c.IncludeXmlComments(System.AppDomain.CurrentDomain.BaseDirectory + "/bin/DemoAPI.XML");
26                    c.CustomProvider((defaultProvider) => new CachingSwaggerProvider(defaultProvider));
27                })
28                .EnableSwaggerUi(c =>
29                {
30                    //路径规则：项目命名空间.文件夹名称.js文件名称
31                    c.InjectJavaScript(thisAssembly, "DemoAPI.Scripts.swagger.js");
32                });
33         }
34     }
35
36     public class CachingSwaggerProvider : ISwaggerProvider
37     {
38         private static ConcurrentDictionary<string, SwaggerDocument> _cache =
39                  new ConcurrentDictionary<string, SwaggerDocument>();
40
41         private readonly ISwaggerProvider _swaggerProvider;
42
43         public CachingSwaggerProvider(ISwaggerProvider swaggerProvider)
44         {
45             _swaggerProvider = swaggerProvider;
46         }
47
48         public SwaggerDocument GetSwagger(string rootUrl, string apiVersion)
49         {
50             var cacheKey = string.Format("{0}_{1}", rootUrl, apiVersion);
51             SwaggerDocument srcDoc = null;
52             //只读取一次
53             if (!_cache.TryGetValue(cacheKey, out srcDoc))
54             {
55                 srcDoc = _swaggerProvider.GetSwagger(rootUrl, apiVersion);
56
57                 srcDoc.vendorExtensions = new Dictionary<string, object> { { "ControllerDesc", GetControllerDesc() } };
58                 _cache.TryAdd(cacheKey, srcDoc);
59             }
60             return srcDoc;
61         }
62
63         /// <summary>
64         /// 从API文档中读取控制器描述
65         /// </summary>
66         /// <returns>所有控制器描述</returns>
67         public static ConcurrentDictionary<string, string> GetControllerDesc()
68         {
69             string xmlpath = string.Format("{0}/bin/DemoAPI.XML", System.AppDomain.CurrentDomain.BaseDirectory);
70             ConcurrentDictionary<string, string> dic = new ConcurrentDictionary<string, string>();
71             if (!File.Exists(xmlpath))
72             {
73                 return dic;
74             }
75             XmlDocument doc = new XmlDocument();
76             doc.Load(xmlpath);
77             XmlNodeList list = doc.SelectNodes("//member[contains(@name,'T:')]");
78             foreach (XmlNode item in list)
79             {
80                 string val = item.Attributes["name"].Value;
81                 string key = val.Substring(val.LastIndexOf('.') + 1).Replace("Controller", "");
82                 XmlNode node = item.FirstChild;
83                 if (!dic.ContainsKey(key))
84                 {
85                     dic.TryAdd(key, node.InnerText.Trim());
86                 }
87             }
88             return dic;
89         }
90
91     }
92 }