diff --git a/pom.xml b/pom.xml
index 5df0585..007bdca 100644
--- a/pom.xml
+++ b/pom.xml
@@ -48,6 +48,15 @@
org.springframework.boot
spring-boot-starter-security
+
+ org.springframework.security
+ spring-security-messaging
+
+
+ org.springframework.boot
+ spring-boot-starter-websocket
+ provided
+
org.springframework.boot
spring-boot-devtools
diff --git a/src/main/java/com/xkrs/FirePointApplication.java b/src/main/java/com/xkrs/FirePointApplication.java
index 3222b01..8de176b 100644
--- a/src/main/java/com/xkrs/FirePointApplication.java
+++ b/src/main/java/com/xkrs/FirePointApplication.java
@@ -3,6 +3,9 @@ package com.xkrs;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
+/**
+ * @author XinYi Song
+ */
@SpringBootApplication
public class FirePointApplication {
diff --git a/src/main/java/com/xkrs/common/config/RestTemplateConfig.java b/src/main/java/com/xkrs/common/config/RestTemplateConfig.java
new file mode 100644
index 0000000..93eb92b
--- /dev/null
+++ b/src/main/java/com/xkrs/common/config/RestTemplateConfig.java
@@ -0,0 +1,91 @@
+package com.xkrs.common.config;
+
+/**
+ * @author xkrs
+ */
+import org.apache.http.client.HttpClient;
+import org.apache.http.config.Registry;
+import org.apache.http.config.RegistryBuilder;
+import org.apache.http.conn.socket.ConnectionSocketFactory;
+import org.apache.http.conn.socket.PlainConnectionSocketFactory;
+import org.apache.http.conn.ssl.NoopHostnameVerifier;
+import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
+import org.apache.http.impl.client.DefaultHttpRequestRetryHandler;
+import org.apache.http.impl.client.HttpClientBuilder;
+import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
+import org.apache.http.ssl.SSLContextBuilder;
+import org.apache.http.ssl.TrustStrategy;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;
+import org.springframework.web.client.DefaultResponseErrorHandler;
+import org.springframework.web.client.RestTemplate;
+
+import javax.net.ssl.HostnameVerifier;
+import javax.net.ssl.SSLContext;
+import java.security.KeyManagementException;
+import java.security.KeyStoreException;
+import java.security.NoSuchAlgorithmException;
+import java.security.cert.CertificateException;
+import java.security.cert.X509Certificate;
+
+/**
+ * @Author: HongZhi Wang
+ * @Date: 2020/8/17 16:45
+ *
+ * 创建RestTemplate配置类,设置连接池大小、超时时间、重试机制等。
+ */
+@Configuration
+public class RestTemplateConfig {
+
+ @Bean
+ public RestTemplate restTemplate(){
+ RestTemplate restTemplate = new RestTemplate();
+ restTemplate.setRequestFactory(clientHttpRequestFactory());
+ restTemplate.setErrorHandler(new DefaultResponseErrorHandler());
+ return restTemplate;
+ }
+
+ @Bean
+ public HttpComponentsClientHttpRequestFactory clientHttpRequestFactory() {
+ try {
+ HttpClientBuilder httpClientBuilder = HttpClientBuilder.create();
+ SSLContext sslContext = new SSLContextBuilder().loadTrustMaterial(null, new TrustStrategy() {
+ @Override
+ public boolean isTrusted(X509Certificate[] chain, String authType) throws CertificateException {
+ return true;
+ }
+ }).build();
+ httpClientBuilder.setSSLContext(sslContext);
+ HostnameVerifier hostnameVerifier = NoopHostnameVerifier.INSTANCE;
+ SSLConnectionSocketFactory sslConnectionSocketFactory = new SSLConnectionSocketFactory(sslContext, hostnameVerifier);
+ Registry socketFactoryRegistry = RegistryBuilder.create()
+ .register("http", PlainConnectionSocketFactory.getSocketFactory())
+ //注册http和https请求
+ .register("https", sslConnectionSocketFactory).build();
+ //开始设置连接池
+ //创建连接池
+ PoolingHttpClientConnectionManager poolingHttpClientConnectionManager = new PoolingHttpClientConnectionManager(socketFactoryRegistry);
+ // 最大连接数500
+ poolingHttpClientConnectionManager.setMaxTotal(500);
+ // 同路由并发数100
+ poolingHttpClientConnectionManager.setDefaultMaxPerRoute(100);
+ httpClientBuilder.setConnectionManager(poolingHttpClientConnectionManager);
+ // 重试次数
+ httpClientBuilder.setRetryHandler(new DefaultHttpRequestRetryHandler(3, true));
+ HttpClient httpClient = httpClientBuilder.build();
+ // httpClient连接配置
+ HttpComponentsClientHttpRequestFactory clientHttpRequestFactory = new HttpComponentsClientHttpRequestFactory(httpClient);
+ // 连接超时
+ clientHttpRequestFactory.setConnectTimeout(20000);
+ // 数据读取超时时间
+ clientHttpRequestFactory.setReadTimeout(30000);
+ // 连接不够用的等待时间
+ clientHttpRequestFactory.setConnectionRequestTimeout(20000);
+ return clientHttpRequestFactory;
+ } catch (KeyManagementException | NoSuchAlgorithmException | KeyStoreException e) {
+ System.out.println("初始化HTTP连接池出错"+ e);
+ }
+ return null;
+ }
+}
diff --git a/src/main/java/com/xkrs/common/config/WebSocketSecurityConfig.java b/src/main/java/com/xkrs/common/config/WebSocketSecurityConfig.java
new file mode 100644
index 0000000..5ea9f85
--- /dev/null
+++ b/src/main/java/com/xkrs/common/config/WebSocketSecurityConfig.java
@@ -0,0 +1,25 @@
+package com.xkrs.common.config;
+
+import org.springframework.context.annotation.Configuration;
+import org.springframework.security.config.annotation.web.messaging.MessageSecurityMetadataSourceRegistry;
+import org.springframework.security.config.annotation.web.socket.AbstractSecurityWebSocketMessageBrokerConfigurer;
+
+import static org.springframework.messaging.simp.SimpMessageType.MESSAGE;
+import static org.springframework.messaging.simp.SimpMessageType.SUBSCRIBE;
+
+
+/**
+ * @author xkrs
+ */
+@Configuration
+public class WebSocketSecurityConfig extends AbstractSecurityWebSocketMessageBrokerConfigurer {
+ @Override
+ protected void configureInbound(MessageSecurityMetadataSourceRegistry messages) {
+ messages
+ .nullDestMatcher().authenticated()
+ .simpSubscribeDestMatchers("/user/queue/errors").permitAll()
+ .simpTypeMatchers(MESSAGE, SUBSCRIBE).denyAll()
+ .anyMessage().denyAll();
+
+ }
+}
\ No newline at end of file
diff --git a/src/main/java/com/xkrs/utils/Query.java b/src/main/java/com/xkrs/utils/Query.java
index 032aa3c..ec382e7 100644
--- a/src/main/java/com/xkrs/utils/Query.java
+++ b/src/main/java/com/xkrs/utils/Query.java
@@ -57,7 +57,7 @@ public class Query {
}
/**
- * 查询近3天的火点
+ * 查询近一周的火点
* @param startTime
* @param endTime
* @return
diff --git a/src/main/java/com/xkrs/websocket/config/WebSocketBrokerConfig.java b/src/main/java/com/xkrs/websocket/config/WebSocketBrokerConfig.java
new file mode 100644
index 0000000..bb8d377
--- /dev/null
+++ b/src/main/java/com/xkrs/websocket/config/WebSocketBrokerConfig.java
@@ -0,0 +1,25 @@
+package com.xkrs.websocket.config;
+
+import org.springframework.context.annotation.Configuration;
+import org.springframework.messaging.simp.config.MessageBrokerRegistry;
+import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
+import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
+import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
+
+/**
+ * websocket广播配置
+ * @author xkrs
+ */
+@Configuration
+@EnableWebSocketMessageBroker
+public class WebSocketBrokerConfig implements WebSocketMessageBrokerConfigurer {
+ @Override
+ public void registerStompEndpoints(StompEndpointRegistry registry) {
+ registry.addEndpoint("/endPointClientServer").withSockJS();
+ }
+
+ @Override
+ public void configureMessageBroker(MessageBrokerRegistry registry) {
+ registry.enableSimpleBroker("/ws");
+ }
+}
diff --git a/src/main/java/com/xkrs/websocket/config/WebSocketConfig.java b/src/main/java/com/xkrs/websocket/config/WebSocketConfig.java
new file mode 100644
index 0000000..261844d
--- /dev/null
+++ b/src/main/java/com/xkrs/websocket/config/WebSocketConfig.java
@@ -0,0 +1,16 @@
+package com.xkrs.websocket.config;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.socket.server.standard.ServerEndpointExporter;
+
+/**
+ * @author xkrs
+ */
+@Configuration
+public class WebSocketConfig {
+ @Bean
+ public ServerEndpointExporter serverEndpointExporter(){
+ return new ServerEndpointExporter();
+ }
+}
diff --git a/src/main/java/com/xkrs/websocket/controller/TestController.java b/src/main/java/com/xkrs/websocket/controller/TestController.java
new file mode 100644
index 0000000..aad8385
--- /dev/null
+++ b/src/main/java/com/xkrs/websocket/controller/TestController.java
@@ -0,0 +1,18 @@
+package com.xkrs.websocket.controller;
+
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+/**
+ * @author xkrs
+ */
+@RestController
+@RequestMapping("/test")
+public class TestController {
+
+ @GetMapping("/hello")
+ public String sayHello(){
+ return "asdasd";
+ }
+}
diff --git a/src/main/java/com/xkrs/websocket/controller/websocket/WebSocketController.java b/src/main/java/com/xkrs/websocket/controller/websocket/WebSocketController.java
new file mode 100644
index 0000000..d9e689a
--- /dev/null
+++ b/src/main/java/com/xkrs/websocket/controller/websocket/WebSocketController.java
@@ -0,0 +1,24 @@
+package com.xkrs.websocket.controller.websocket;
+
+import com.xkrs.websocket.service.WebSocketServer;
+import org.springframework.web.bind.annotation.*;
+
+/**
+ * @author xkrs
+ */
+@RestController
+@RequestMapping("/api/socket")
+public class WebSocketController {
+
+ @GetMapping("/sendAll")
+ public String sendAll(@RequestParam String msg) {
+ WebSocketServer.broadInfo(msg);
+ return "success";
+ }
+
+ @RequestMapping(value = "/sendOne", method = RequestMethod.GET)
+ public String sendOne(@RequestParam String msg, @RequestParam String session) {
+ WebSocketServer.sendMsg(msg, session);
+ return "success";
+ }
+}
diff --git a/src/main/java/com/xkrs/websocket/controller/websocketbroker/ViewController.java b/src/main/java/com/xkrs/websocket/controller/websocketbroker/ViewController.java
new file mode 100644
index 0000000..4bd39b1
--- /dev/null
+++ b/src/main/java/com/xkrs/websocket/controller/websocketbroker/ViewController.java
@@ -0,0 +1,17 @@
+package com.xkrs.websocket.controller.websocketbroker;
+
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.servlet.config.annotation.ViewControllerRegistry;
+import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;
+
+/**
+ * @author xkrs
+ */
+@Configuration
+public class ViewController extends WebMvcConfigurerAdapter {
+
+ @Override
+ public void addViewControllers(ViewControllerRegistry registry) {
+ registry.addViewController("/ws").setViewName("/ws");
+ }
+}
diff --git a/src/main/java/com/xkrs/websocket/controller/websocketbroker/WebSocketBrokerController.java b/src/main/java/com/xkrs/websocket/controller/websocketbroker/WebSocketBrokerController.java
new file mode 100644
index 0000000..9700d44
--- /dev/null
+++ b/src/main/java/com/xkrs/websocket/controller/websocketbroker/WebSocketBrokerController.java
@@ -0,0 +1,21 @@
+package com.xkrs.websocket.controller.websocketbroker;
+
+import com.xkrs.websocket.websocket.ClientServerMessage;
+import com.xkrs.websocket.websocket.ServerClientMessage;
+import org.springframework.messaging.handler.annotation.MessageMapping;
+import org.springframework.messaging.handler.annotation.SendTo;
+import org.springframework.stereotype.Controller;
+
+/**
+ * @author xkrs
+ */
+@Controller
+public class WebSocketBrokerController {
+
+ @MessageMapping("/hello")
+ @SendTo("/ws/getResponse")
+ public ServerClientMessage say(ClientServerMessage message) throws InterruptedException {
+ Thread.sleep(3000);
+ return new ServerClientMessage("hello," + message.getName() + "!");
+ }
+}
diff --git a/src/main/java/com/xkrs/websocket/service/WebSocketServer.java b/src/main/java/com/xkrs/websocket/service/WebSocketServer.java
new file mode 100644
index 0000000..c8a3685
--- /dev/null
+++ b/src/main/java/com/xkrs/websocket/service/WebSocketServer.java
@@ -0,0 +1,128 @@
+package com.xkrs.websocket.service;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.PostConstruct;
+import javax.websocket.*;
+import javax.websocket.server.ServerEndpoint;
+import java.io.IOException;
+import java.util.Map;
+import java.util.concurrent.CopyOnWriteArraySet;
+import java.util.concurrent.atomic.AtomicInteger;
+
+/**
+ * @author xkrs
+ */
+@Component
+@ServerEndpoint(value = "/ws/asset")
+public class WebSocketServer {
+
+ @PostConstruct
+ public void init() {
+ System.out.println("websocketBroker loading......");
+ }
+
+ private static Logger log = LoggerFactory.getLogger(WebSocketServer.class);
+ private static final AtomicInteger ONLINE_COUNT = new AtomicInteger(0);
+ private static CopyOnWriteArraySet sessionSet = new CopyOnWriteArraySet<>();
+
+ /**
+ * 连接建立
+ */
+ @OnOpen
+ public void onOpen(Session session) {
+ sessionSet.add(session);
+ //在线加1
+ int cnt = ONLINE_COUNT.incrementAndGet();
+ //log.info("have a new connect:{}", cnt);
+ //sendMsg(session, "connect success");
+ }
+
+ /**
+ * 连接关闭
+ *
+ * @param session
+ */
+ @OnClose
+ public void onClose(Session session) {
+ sessionSet.remove(session);
+ //在线减1
+ int cnt = ONLINE_COUNT.decrementAndGet();
+ log.info("have a connect close:{}", cnt);
+ }
+
+ /**
+ * 收到客户端消息后调用的方法
+ *
+ * @param session
+ * @param msg
+ */
+ @OnMessage
+ public void onMessage(String msg, Session session) {
+ log.info("receive the msg:{}", msg);
+ }
+
+ private static void sendMsg(Session session, String msg) {
+ try {
+ session.getBasicRemote().sendText(String.format(msg));
+ } catch (IOException e) {
+ log.error("发送信息出错:{}", e.getMessage());
+ e.printStackTrace();
+ }
+ }
+
+ private static void sendDate(Session session, Map map) {
+ try {
+ session.getBasicRemote().sendObject(map);
+ } catch (IOException | EncodeException e) {
+ log.error("发送信息出错:{}", e.getMessage());
+ e.printStackTrace();
+ }
+ }
+
+ /**
+ * 指定session发送
+ */
+ public static void sendMsg(String msg, String sessionId) {
+ Session session = null;
+ for (Session s : sessionSet) {
+ if (s.getId().equals(sessionId)) {
+ session = s;
+ break;
+ }
+ }
+ if (session != null) {
+ sendMsg(session, msg);
+ } else {
+ log.warn("not find your session");
+ }
+ }
+
+ /**
+ * 群发消息
+ */
+ public static void broadInfo(String msg) {
+ for (Session s : sessionSet) {
+ synchronized (sessionSet){
+ if (s.isOpen()) {
+ sendMsg(s, msg);
+ }
+ }
+ }
+ }
+
+ /**
+ * 群发map消息
+ * @param map
+ */
+ public static void broadInfoMap(Map map) {
+ for (Session s : sessionSet) {
+ if (s.isOpen()) {
+ sendDate(s, map);
+ }
+ }
+ }
+
+}
diff --git a/src/main/java/com/xkrs/websocket/websocket/ClientServerMessage.java b/src/main/java/com/xkrs/websocket/websocket/ClientServerMessage.java
new file mode 100644
index 0000000..11c054e
--- /dev/null
+++ b/src/main/java/com/xkrs/websocket/websocket/ClientServerMessage.java
@@ -0,0 +1,18 @@
+package com.xkrs.websocket.websocket;
+
+/**
+ * @author xkrs
+ */
+public class ClientServerMessage {
+
+ public String getName() {
+ return name;
+ }
+
+ public void setName(String name) {
+ this.name = name;
+ }
+
+ private String name;
+
+}
diff --git a/src/main/java/com/xkrs/websocket/websocket/ServerClientMessage.java b/src/main/java/com/xkrs/websocket/websocket/ServerClientMessage.java
new file mode 100644
index 0000000..b803856
--- /dev/null
+++ b/src/main/java/com/xkrs/websocket/websocket/ServerClientMessage.java
@@ -0,0 +1,21 @@
+package com.xkrs.websocket.websocket;
+
+/**
+ * @author xkrs
+ */
+public class ServerClientMessage {
+
+ public String getResponseMsg() {
+ return responseMsg;
+ }
+
+ public void setResponseMsg(String responseMsg) {
+ this.responseMsg = responseMsg;
+ }
+
+ private String responseMsg;
+
+ public ServerClientMessage(String responseMsg) {
+ this.responseMsg = responseMsg;
+ }
+}
diff --git a/src/main/resources/application.properties b/src/main/resources/application.properties
index 5d80227..238054a 100644
--- a/src/main/resources/application.properties
+++ b/src/main/resources/application.properties
@@ -47,6 +47,8 @@ spring.servlet.multipart.max-file-size = 64MB
# 最大请求大小
spring.servlet.multipart.max-request-size = 70MB
+spring.main.allow-bean-definition-overriding = true
+
# Geoserver服务器地址
my.GeoserverAdress = http://139.199.98.175:9080/geoserver/