فهرست منبع

增加非`DEBUG`期间日志

mxd 3 سال پیش
والد
کامیت
b8694c512a

+ 9 - 2
magic-api-spring-boot-starter/src/main/java/org/ssssssss/magicapi/spring/boot/starter/MagicAPIAutoConfiguration.java

@@ -594,9 +594,15 @@ public class MagicAPIAutoConfiguration implements WebMvcConfigurer, WebSocketCon
 		return configuration;
 	}
 
+	private DefaultAuthorizationInterceptor defaultAuthorizationInterceptor;
+
 	public AuthorizationInterceptor createAuthorizationInterceptor() {
+		if(defaultAuthorizationInterceptor != null){
+			return defaultAuthorizationInterceptor;
+		}
 		SecurityConfig securityConfig = properties.getSecurityConfig();
-		return new DefaultAuthorizationInterceptor(securityConfig.getUsername(), securityConfig.getPassword());
+		defaultAuthorizationInterceptor = new DefaultAuthorizationInterceptor(securityConfig.getUsername(), securityConfig.getPassword());
+		return defaultAuthorizationInterceptor;
 	}
 
 	private RestTemplate createRestTemplate() {
@@ -620,7 +626,8 @@ public class MagicAPIAutoConfiguration implements WebMvcConfigurer, WebSocketCon
 		if (web != null && !registerWebsocket) {
 			registerWebsocket = true;
 			MagicWebSocketDispatcher dispatcher = new MagicWebSocketDispatcher(properties.getClusterConfig().getInstanceId(), magicNotifyServiceProvider.getObject(), Arrays.asList(
-					new MagicDebugHandler()
+					new MagicDebugHandler(),
+					new MagicWorkbenchHandler(authorizationInterceptorProvider.getIfAvailable(this::createAuthorizationInterceptor))
 			));
 			WebSocketHandlerRegistration registration = webSocketHandlerRegistry.addHandler(dispatcher, web + "/console");
 			if (properties.isSupportCrossDomain()) {