package com.tiger.biz.websocket; import lombok.extern.slf4j.Slf4j; import org.java_websocket.client.WebSocketClient; import org.java_websocket.drafts.Draft_6455; import org.java_websocket.handshake.ServerHandshake; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; import org.springframework.data.redis.core.RedisTemplate; import org.springframework.stereotype.Component; import javax.websocket.server.ServerEndpoint; import java.net.URI; import java.text.SimpleDateFormat; import java.util.Date; import java.util.Map; import java.util.concurrent.Executors; import java.util.concurrent.ScheduledExecutorService; import java.util.concurrent.TimeUnit; /** * websocket的客户端 */ @Slf4j public class DemoWebSocketClient extends WebSocketClient { @Autowired private RedisTemplate<String, String> redisTemplate; public static final String HEARTBEAT_CMD = "此处为商定的保活命令"; public DemoWebSocketClient(URI serverUri) { super(serverUri, new Draft_6455()); } @Override public void onOpen(ServerHandshake serverHandshake) { //开启心跳保活 heartbeat(this); log.info("===建立连接,心跳保活开启==="); } @Override public void onMessage(String s) { log.info("{}时来自服务端的消息:{}", new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date()), s); try { Map<Object, Object> map = redisTemplate.opsForHash().entries("zd:location"); System.out.println(map.size()+"dd"); } catch (Exception e) { e.printStackTrace(); } } @Override public void onClose(int a, String s, boolean b) { //重连 log.info("由于:{},连接被关闭,开始尝试重新连接", s); DemoReconnectThreadEnum.getInstance().reconnectWs(this); } @Override public void onError(Exception e) { log.error("====websocket出现错误====" + e.getMessage()); } /** * 心跳保活 * * @param var1 */ private void heartbeat(DemoWebSocketClient var1) { Schedul 改进以上代码怎么让 redistimplate正常注入 且URI serverUri不报错
时间: 2024-01-31 14:02:14 浏览: 219
WebSocket-client.zip_DEMO_websocket_websocket client_websocket客户
首先,需要将该类注入到 Spring 容器中,可以使用 @Component 注解进行标记。同时,需要在 DemoWebSocketClient 类中添加一个空的构造函数,并在其中调用 super() 方法,以便能够正常地注入 RedisTemplate 对象。具体代码如下:
```
@Component
@Slf4j
public class DemoWebSocketClient extends WebSocketClient {
private RedisTemplate<String, String> redisTemplate;
@Value("${websocket.server-uri}")
private String serverUri;
public DemoWebSocketClient() {
super(URI.create(serverUri), new Draft_6455());
}
@Autowired
public DemoWebSocketClient(RedisTemplate<String, String> redisTemplate) {
super(URI.create(serverUri), new Draft_6455());
this.redisTemplate = redisTemplate;
}
// 省略其他方法不变
}
```
在构造函数中添加了一个 RedisTemplate 类型的参数,并在该参数上添加了 @Autowired 注解,这样 Spring 就会自动将 RedisTemplate 对象注入到 DemoWebSocketClient 中。同时,在添加了 @Value 注解的 serverUri 变量上,可以通过配置文件进行注入,避免了硬编码的问题。
阅读全文