软件编程
位置:首页>> 软件编程>> java编程>> JAVA实现心跳检测(长连接)

JAVA实现心跳检测(长连接)

作者:菜鸟---程序员  发布时间:2022-12-16 04:26:30 

标签:JAVA,心跳检测

1、心跳机制简介

在分布式系统中,分布在不同主机上的节点需要检测其他节点的状态,如服务器节点需要检测从节点是否失效。为了检测对方节点的有效性,每隔固定时间就发送一个固定信息给对方,对方回复一个固定信息,如果长时间没有收到对方的回复,则断开与对方的连接。

发包方既可以是服务端,也可以是客户端,这要看具体实现。因为是每隔固定时间发送一次,类似心跳,所以发送的固定信息称为心跳包。心跳包一般为比较小的包,可根据具体实现。心跳包主要应用于长连接的保持与短线链接。

一般而言,应该客户端主动向服务器发送心跳包,因为服务器向客户端发送心跳包会影响服务器的性能。

实现原理:长连接的维持,是要客户端程序,定时向服务端程序,发送一个维持连接包的。如果,长时间未发送维持连接包,服务端程序将断开连接。

2、心跳机制实现方式

心跳机制有两种实现方式,一种基于TCP自带的心跳包,TCP的SO_KEEPALIVE选项可以,系统默认的默认跳帧频率为2小时,超过2小时后,本地的TCP 实现会发送一个数据包给远程的 Socket. 如果远程Socket 没有发回响应, TCP实现就会持续尝试 11 分钟, 直到接收到响应为止。 否则就会自动断开Socket连接。但TCP自带的心跳包无法检测比较敏感地知道对方的状态,默认2小时的空闲时间,对于大多数的应用而言太长了。可以手工开启KeepAlive功能并设置合理的KeepAlive参数。

另一种在应用层自己进行实现,基本步骤如下:

1、Client使用定时器,不断发送心跳;

2、Server收到心跳后,回复一个包;

3、Server为每个Client启动超时定时器,如果在指定时间内没有收到Client的心跳包,则Client失效。

3、客户端

Client通过持有Socket的对象,可以随时(使用sendObject方法)发送Massage Object(消息)给服务端。

如果keepAliveDelay毫秒(程序中是2秒)内未发送任何数据,则自动发送一个KeepAlive Object(心跳)给服务端,用于维持连接。

由于,我们向服务端,可以发送很多不同的消息对象,服务端也可以返回不同的对象。所以,对于返回对象的处理,要编写具体的ObjectAction实现类进行处理。通过Client.addActionMap方法进行添加。这样,程序会回调处理

4 、服务端

由于客户端会定时(keepAliveDelay毫秒)发送维持连接的信息过来,所以,服务端要有一个检测机制。

即当服务端receiveTimeDelay毫秒(程序中是3秒)内未接收任何数据,则自动断开与客户端的连接。

ActionMapping的原理与客户端相似(相同)。

通过添加相应的ObjectAction实现类,可以实现不同对象的响应、应答过程。

5、代码实现

5.1 KeepAlive.java

package com.zyz.mynative.demo02;

/**
* @author zyz
* @version 1.0
* @data 2023/2/15 11:05
* @Description:
*/
import java.io.Serializable;
import java.text.SimpleDateFormat;
import java.util.Date;

/**
*
* 维持连接的消息对象(心跳对象)
* @author zyz
* @version 1.0
* @data 2023/2/15 11:05
* @Description:
*/
public class KeepAlive implements Serializable{

private static final long serialVersionUID = -2813120366138988480L;

/* 覆盖该方法,仅用于测试使用。
    * @see java.lang.Object#toString()
    */

@Override
   public String toString() {
       return new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date())+"\t维持连接包";
   }

}

5.2 MyClient.java

package com.zyz.mynative.demo02;

/**
* @author zyz
* @version 1.0
* @data 2023/2/15 11:06
* @Description:
*/

import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.Socket;
import java.net.UnknownHostException;
import java.util.concurrent.ConcurrentHashMap;

/**
* C/S架构的客户端对象,持有该对象,可以随时向服务端发送消息。
*
* @author zyz
* @version 1.0
* @data 2023/2/15 11:06
* @Description:
*/
public class MyClient {

/**
    * 处理服务端发回的对象,可实现该接口。
    */
   public static interface ObjectAction {
       void doAction(Object obj, MyClient client);
   }

public static final class DefaultObjectAction implements ObjectAction {
       @Override
       public void doAction(Object obj, MyClient client) {
           System.out.println("处理:\t" + obj.toString());
       }
   }

private String serverIp;
   private int port;
   private Socket socket;
   //连接状态
   private boolean running = false;

/**
    * 最后一次发送数据的时间
    */
   private long lastSendTime;

/**
    * 用于保存接收消息对象类型及该类型消息处理的对象
    */
   private ConcurrentHashMap<Class, ObjectAction> actionMapping = new ConcurrentHashMap<Class, ObjectAction>();

public MyClient(String serverIp, int port) {
       this.serverIp = serverIp;
       this.port = port;
   }

public void start() throws UnknownHostException, IOException {
       if (running) {
           return;
       }
       socket = new Socket(serverIp, port);
       System.out.println("本地端口:" + socket.getLocalPort());
       lastSendTime = System.currentTimeMillis();
       running = true;

//保持长连接的线程,每隔2秒项服务器发一个一个保持连接的心跳消息
       new Thread(new KeepAliveWatchDog()).start();
       //接受消息的线程,处理消息
       new Thread(new ReceiveWatchDog()).start();
   }

public void stop() {
       if (running) {
           running = false;
       }
       ;
   }

/**
    * 添加接收对象的处理对象。
    *
    * @param cls    待处理的对象,其所属的类。
    * @param action 处理过程对象。
    */
   public void addActionMap(Class<Object> cls, ObjectAction action) {
       actionMapping.put(cls, action);
   }

public void sendObject(Object obj) throws IOException {
       ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream());
       oos.writeObject(obj);
       System.out.println("发送:\t" + obj);
       oos.flush();
   }

class KeepAliveWatchDog implements Runnable {
       long checkDelay = 10;
       //两秒钟检测一次
       long keepAliveDelay = 2000;

@Override
       public void run() {
           while (running) {
               if (System.currentTimeMillis() - lastSendTime > keepAliveDelay) {
                   try {
                       MyClient.this.sendObject(new KeepAlive());
                   } catch (IOException e) {
                       e.printStackTrace();
                       MyClient.this.stop();
                   }
                   lastSendTime = System.currentTimeMillis();
               } else {
                   try {
                       Thread.sleep(checkDelay);
                   } catch (InterruptedException e) {
                       e.printStackTrace();
                       MyClient.this.stop();
                   }
               }
           }
       }
   }

class ReceiveWatchDog implements Runnable {
       @Override
       public void run() {
           while (running) {
               try {
                   InputStream in = socket.getInputStream();
                   if (in.available() > 0) {
                       ObjectInputStream ois = new ObjectInputStream(in);
                       Object obj = ois.readObject();
                       System.out.println("接收:\t" + obj);
                       ObjectAction oa = actionMapping.get(obj.getClass());
                       oa = oa == null ? new DefaultObjectAction() : oa;
                       oa.doAction(obj, MyClient.this);
                   } else {
                       Thread.sleep(10);
                   }
               } catch (Exception e) {
                   e.printStackTrace();
                   MyClient.this.stop();
               }
           }
       }
   }

public static void main(String[] args) throws UnknownHostException, IOException {
       String serverIp = "127.0.0.1";
       int port = 65432;
       MyClient client = new MyClient(serverIp, port);
       client.start();
   }
}

5.3 MyServer

package com.zyz.mynative.demo02;

import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.concurrent.ConcurrentHashMap;

/**
* C/S架构的服务端对象。
*
* @author zyz
* @version 1.0
* @data 2023/2/15 11:08
* @Description:
*/
public class MyServer {

/**
    * 要处理客户端发来的对象,并返回一个对象,可实现该接口。
    */
   public interface ObjectAction {
       Object doAction(Object rev, MyServer server);
   }

public static final class DefaultObjectAction implements ObjectAction {
       @Override
       public Object doAction(Object rev, MyServer server) {
           System.out.println("处理并返回:" + rev);
           return rev;
       }
   }

private int port;
   private volatile boolean running = false;
   private long receiveTimeDelay = 3000;
   /**
    * ConcurrentHashMap是线程安全的,ConcurrentHashMap并非锁住整个方法,
    * 而是通过原子操作和局部加锁的方法保证了多线程的线程安全,且尽可能减少了性能损耗。
    */
   private ConcurrentHashMap<Class, ObjectAction> actionMapping = new ConcurrentHashMap<Class, ObjectAction>();
   private Thread connWatchDog;

public MyServer(int port) {
       this.port = port;
   }

/**
    * 通过继承Runnable ,开启线程
    */
   public void start() {
       if (running) {
           return;
       }
       ;
       running = true;
       connWatchDog = new Thread(new ConnWatchDog());
       connWatchDog.start();
   }

@SuppressWarnings("deprecation")
   public void stop() {
       if (running) {
           running = false;
       }
       ;
       if (connWatchDog != null) {
           connWatchDog.stop();
       }
       ;
   }

public void addActionMap(Class<Object> cls, ObjectAction action) {
       actionMapping.put(cls, action);
   }

/**
    * 继承Runnable 重写run方法、实现线程
    */
   class ConnWatchDog implements Runnable {
       @Override
       public void run() {
           try {
               ServerSocket ss = new ServerSocket(port, 5);
               while (running) {
                   Socket s = ss.accept();
                   new Thread(new SocketAction(s)).start();
               }
           } catch (IOException e) {
               e.printStackTrace();
               MyServer.this.stop();
           }

}
   }

/**
    * 继承Runnable 重写run方法、实现线程,通过匿名内部类
    */
   class SocketAction implements Runnable {
       Socket s;
       boolean run = true;
       long lastReceiveTime = System.currentTimeMillis();

public SocketAction(Socket s) {
           this.s = s;
       }

@Override
       public void run() {
           while (running && run) {
               if (System.currentTimeMillis() - lastReceiveTime > receiveTimeDelay) {
                   overThis();
               } else {
                   try {
                       InputStream in = s.getInputStream();
                       if (in.available() > 0) {
                           ObjectInputStream ois = new ObjectInputStream(in);
                           Object obj = ois.readObject();
                           lastReceiveTime = System.currentTimeMillis();
                           System.out.println("接收:\t" + obj);
                           ObjectAction oa = actionMapping.get(obj.getClass());
                           oa = oa == null ? new DefaultObjectAction() : oa;
                           Object out = oa.doAction(obj, MyServer.this);
                           if (out != null) {
                               ObjectOutputStream oos = new ObjectOutputStream(s.getOutputStream());
                               oos.writeObject(out);
                               oos.flush();
                           }
                       } else {
                           Thread.sleep(10);
                       }
                   } catch (Exception e) {
                       e.printStackTrace();
                       overThis();
                   }
               }
           }
       }

private void overThis() {
           if (run) {
               run = false;
           }
           ;
           if (s != null) {
               try {
                   s.close();
               } catch (IOException e) {
                   e.printStackTrace();
               }
           }
           System.out.println("关闭:" + s.getRemoteSocketAddress());
       }

}

public static void main(String[] args) {
       int port = 65432;
       MyServer server = new MyServer(port);
       server.start();
   }

}

5.4 测试结果

先开启服务端,在开启客户端

JAVA实现心跳检测(长连接)

参考资料:

1、java实现心跳机制

2、java实现长连接

来源:https://blog.csdn.net/weixin_43304253/article/details/129051919

0
投稿

猜你喜欢

手机版 软件编程 asp之家 www.aspxhome.com