From a092d5396f5eecbe46d4a04b9f70742faf356ccd Mon Sep 17 00:00:00 2001
From: pandaapo <35672972+pandaapo@users.noreply.github.com>
Date: Wed, 14 Dec 2022 09:44:10 +0800
Subject: [PATCH] Fix issue: prompt no registration if web dependence was not
 imported. (#777)

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

diff --git a/CHANGELOG.md b/CHANGELOG.md
index bf3751aa2..722625864 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -18,3 +18,4 @@
 - [Bugfix: get service instances by Flux.blockLast() to resolve concurrent problem](https://github.com/Tencent/spring-cloud-tencent/pull/763)
 - [Test:add sct-stater-polaris-router juint.](https://github.com/Tencent/spring-cloud-tencent/pull/767)
 - [Feature: support nacos registry](https://github.com/Tencent/spring-cloud-tencent/pull/757)
+- [Fix issue: prompt no registration if web dependence was not imported.](https://github.com/Tencent/spring-cloud-tencent/pull/777)
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 739a04361..60b665356 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
@@ -73,4 +73,9 @@ public class PolarisServiceRegistryAutoConfiguration {
 			AutoServiceRegistrationProperties autoServiceRegistrationProperties, PolarisRegistration registration) {
 		return new PolarisAutoServiceRegistration(registry, autoServiceRegistrationProperties, registration);
 	}
+
+	@Bean
+	public PolarisWebApplicationCheck polarisWebApplicationCheck() {
+		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..2ed34d256
--- /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.warn("This service instance will not be registered, because it is not a servlet-based or reactive web application.");
+		}
+	}
+}