From 99a1a65939e5f728ba1637ff990f194c794875c2 Mon Sep 17 00:00:00 2001
From: 12888491 <295228902@qq.com>
Date: Thu, 3 Nov 2022 15:11:15 +0800
Subject: [PATCH] =?UTF-8?q?api=E6=96=87=E6=A1=A3=E6=98=BE=E7=A4=BA?=
 =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=9D=83=E9=99=90=E8=AE=A4=E8=AF=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 WaterCloud.Web/Startup.cs | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/WaterCloud.Web/Startup.cs b/WaterCloud.Web/Startup.cs
index b7ae33c..fb09228 100644
--- a/WaterCloud.Web/Startup.cs
+++ b/WaterCloud.Web/Startup.cs
@@ -29,6 +29,7 @@ using Microsoft.AspNetCore.ResponseCompression;
 using System.IO.Compression;
 using WaterCloud.Domain.SystemOrganize;
 using Microsoft.OpenApi.Models;
+using System.Collections.Generic;
 
 namespace WaterCloud.Web
 {
@@ -54,7 +55,26 @@ namespace WaterCloud.Web
 				config.SwaggerDoc("v1", new OpenApiInfo { Title = "WaterCloud Api", Version = "v1" });
 				var xmlFile = $"{Assembly.GetExecutingAssembly().GetName().Name}.xml";
 				var xmlPath = Path.Combine(AppContext.BaseDirectory, xmlFile);
-				config.IncludeXmlComments(xmlPath, true); //添加控制器层注释（true表示显示控制器注释）                
+				config.IncludeXmlComments(xmlPath, true); //添加控制器层注释（true表示显示控制器注释）
+				config.AddSecurityDefinition(GlobalContext.SystemConfig.TokenName, new OpenApiSecurityScheme
+				{
+					Description = "header token",
+					Name = GlobalContext.SystemConfig.TokenName,
+					In = ParameterLocation.Header,
+					Scheme = "",
+					Type = SecuritySchemeType.ApiKey,//设置类型
+					BearerFormat = ""
+				});
+				config.AddSecurityRequirement(new OpenApiSecurityRequirement
+				{
+					{
+						new OpenApiSecurityScheme
+						{
+							Reference = new OpenApiReference { Type = ReferenceType.SecurityScheme, Id = GlobalContext.SystemConfig.TokenName }
+						},
+						new List<string>()
+					}
+				});
 			});
 			services.AddSession();
             //代替HttpContext.Current