From 63e21f9779be8df5c9e08b3ce32d90526ca2edff Mon Sep 17 00:00:00 2001
From: pandaapo <1052156701@qq.com>
Date: Sat, 10 Dec 2022 21:59:41 +0800
Subject: [PATCH] Fix issue: prompt no registration if web dependence was not
 imported.

---
 ...larisServiceRegistryAutoConfiguration.java |  5 +++
 .../registry/PolarisWebApplicationCheck.java  | 45 +++++++++++++++++++
 2 files changed, 50 insertions(+)
 create mode 100644 spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisWebApplicationCheck.java

diff --git a/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistryAutoConfiguration.java b/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistryAutoConfiguration.java
index 129450bca..41c69e8e6 100644
--- a/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistryAutoConfiguration.java
+++ b/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistryAutoConfiguration.java
@@ -72,4 +72,9 @@ public class PolarisServiceRegistryAutoConfiguration {
 			AutoServiceRegistrationProperties autoServiceRegistrationProperties, PolarisRegistration registration) {
 		return new PolarisAutoServiceRegistration(registry, autoServiceRegistrationProperties, registration);
 	}
+
+	@Bean
+	public PolarisWebApplicationCheck polarisApplicationReady() {
+		return new PolarisWebApplicationCheck();
+	}
 }
diff --git a/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisWebApplicationCheck.java b/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisWebApplicationCheck.java
new file mode 100644
index 000000000..69016a4f3
--- /dev/null
+++ b/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisWebApplicationCheck.java
@@ -0,0 +1,45 @@
+/*
+ * Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
+ *
+ * Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
+ *
+ * Licensed under the BSD 3-Clause License (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://opensource.org/licenses/BSD-3-Clause
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed
+ * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
+ * CONDITIONS OF ANY KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations under the License.
+ */
+
+package com.tencent.cloud.polaris.registry;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.WebApplicationType;
+import org.springframework.boot.context.event.ApplicationReadyEvent;
+import org.springframework.context.ApplicationListener;
+
+/**
+ * Prompt user if web dependence was not imported.
+ *
+ * @author Daifu Wu
+ */
+public class PolarisWebApplicationCheck implements ApplicationListener<ApplicationReadyEvent> {
+
+	private static final Logger LOGGER = LoggerFactory.getLogger(PolarisWebApplicationCheck.class);
+
+	@Override
+	public void onApplicationEvent(ApplicationReadyEvent event) {
+		SpringApplication springApplication = event.getSpringApplication();
+		WebApplicationType webApplicationType = springApplication.getWebApplicationType();
+		if (webApplicationType.equals(WebApplicationType.NONE)) {
+			LOGGER.info("This service instance will not be registered, because it is not a servlet-based or reactive web application.");
+		}
+	}
+}