Java使用mina开发TCP服务

今晚哒老虎

已于 2023-08-07 09:43:02 修改

阅读量270

点赞数 2

分类专栏： Java 文章标签： java 开发语言

于 2023-08-04 14:43:46 首次发布

本文链接：https://blog.csdn.net/guo__hang/article/details/132103836

版权

Java 专栏收录该内容

24 篇文章 0 订阅

订阅专栏

1.导入依赖

        <!-- https://mvnrepository.com/artifact/org.apache.mina/mina-core -->
        <dependency>
            <groupId>org.apache.mina</groupId>
            <artifactId>mina-core</artifactId>
            <version>2.2.1</version>
        </dependency>

2.编写TCP服务端

import org.apache.log4j.Logger;
import org.apache.mina.core.service.IoAcceptor;
import org.apache.mina.core.session.IdleStatus;
import org.apache.mina.filter.codec.ProtocolCodecFilter;
import org.apache.mina.filter.codec.serialization.ObjectSerializationCodecFactory;
import org.apache.mina.transport.socket.nio.NioSocketAcceptor;

import java.io.IOException;
import java.net.InetSocketAddress;

public class TCPServer implements Runnable {

    private static final Logger logger = Logger.getLogger(TCPServer.class);

    private final int port = 55555;

    @Override
    public void run() {

        IoAcceptor acceptor = new NioSocketAcceptor();
        acceptor.getSessionConfig().setReadBufferSize(2048);
        acceptor.getSessionConfig().setIdleTime(IdleStatus.BOTH_IDLE, 10);

        // 编写过滤器
//        acceptor.getFilterChain().addLast("codec",
//                new ProtocolCodecFilter(new TextLineCodecFactory(Charset.forName("UTF-8"),
//                        LineDelimiter.WINDOWS.getValue(),
//                        LineDelimiter.WINDOWS.getValue()))
//        );
        //加入解码器
        acceptor.getFilterChain().addLast("objectFilter",new ProtocolCodecFilter(new ObjectSerializationCodecFactory()));

        // 注册IoHandler
        acceptor.setHandler(new TCPServerHandler());

        // 绑定端口
        try {
            acceptor.bind(new InetSocketAddress(port));
        } catch (IOException e) {
            logger.error("服务端口被占用:" + port);
        }


        logger.info("TCPServer启动成功 port:" + port);
    }
}

2.注册服务端IoHandler，继承 IoHandlerAdapter 重写父类方法

import org.apache.log4j.Logger;
import org.apache.mina.core.service.IoHandlerAdapter;
import org.apache.mina.core.session.IdleStatus;
import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.FilterEvent;

import java.net.InetSocketAddress;
import java.util.HashMap;

public class TCPServerHandler extends IoHandlerAdapter {

    private static final Logger logger = Logger.getLogger(TCPServerHandler.class);

    @Override
    public void sessionCreated(IoSession session) throws Exception {
        super.sessionCreated(session);
    }

    @Override
    public void sessionOpened(IoSession session) throws Exception {
        logger.info(session.getRemoteAddress() + "已连接...");
    }

    @Override
    public void sessionClosed(IoSession session) throws Exception {
        super.sessionClosed(session);
    }

    @Override
    public void sessionIdle(IoSession session, IdleStatus status) throws Exception {
        super.sessionIdle(session, status);
    }

    @Override
    public void exceptionCaught(IoSession session, Throwable cause) throws Exception {
        // 连接异常处理
        logger.error(session.getRemoteAddress() + cause.getMessage());
    }

    @Override
    public void messageReceived(IoSession session, Object message) throws Exception {
        if (message instanceof User) {
            User request = (User) message;
        }
    }

    @Override
    public void messageSent(IoSession session, Object message) throws Exception {
        super.messageSent(session, message);
    }

    @Override
    public void inputClosed(IoSession session) throws Exception {
        super.inputClosed(session);
    }

    @Override
    public void event(IoSession session, FilterEvent event) throws Exception {
        super.event(session, event);
    }
}

3.编写TCP客户端

import org.apache.log4j.Logger;
import org.apache.mina.core.future.ConnectFuture;
import org.apache.mina.core.service.IoConnector;
import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.codec.ProtocolCodecFilter;
import org.apache.mina.filter.codec.serialization.ObjectSerializationCodecFactory;
import org.apache.mina.transport.socket.nio.NioSocketConnector;

import java.net.InetSocketAddress;

public class TCPClient implements Runnable {

    private static final Logger logger = Logger.getLogger(TCPClient.class);

    private final String ip = "127.0.0.1";

    private final int port = 55555;

    //创建连接客户端
    private static final IoConnector connector = new NioSocketConnector();

    private static ConnectFuture connect = null;

    @Override
    public void run() {
        logger.info("连接服务:" + ip);

        //设置连接超时
        connector.setConnectTimeoutMillis(30000);

//        connector.getFilterChain().addLast("codec",
//                new ProtocolCodecFilter(new TextLineCodecFactory(Charset.forName("UTF-8"),
//                        LineDelimiter.WINDOWS.getValue(),
//                        LineDelimiter.WINDOWS.getValue())));
        connector.getFilterChain().addLast("objectFilter", new ProtocolCodecFilter(new ObjectSerializationCodecFactory()));

        connector.setHandler(new TCPClientHandler());

        // 设置默认访问地址
        connector.setDefaultRemoteAddress(new InetSocketAddress(ip, port));
        // 添加重连监听
        connector.addListener(new TCPClientIoServiceListener());

        connect = connector.connect(new InetSocketAddress(ip, port));

        // 阻塞检测是否连接成功
        if (!status()) {
            reconnection();
        }
        logger.info("connected...");

    }

    /**
     * 服务连接状态
     * @return
     */
    public static boolean status() {
        try {
            connect.awaitUninterruptibly();// 等待连接创建成功
            IoSession session = connect.getSession();// 获取会话
            return session.isConnected();
        } catch (Exception e) {
            return false;
        }
    }

    /**
     * 尝试重连
     */
    public static void reconnection() {
        while (true) {
            try {
                ThreadUtil.sleep(1000 * 3);
                ConnectFuture future = connector.connect();
                future.awaitUninterruptibly();// 等待连接创建成功
                IoSession session = future.getSession();// 获取会话
                if (session.isConnected()) {
                    logger.info("断线重连[" + connector.getDefaultRemoteAddress() + "]成功...");
                    break;
                }
            } catch (Exception ex) {
                logger.info("重连服务器登录失败,3秒后重试:" + ex.getMessage());
            }
        }
    }

}

4.同样需要注册客户端IoHandler

import org.apache.log4j.Logger;
import org.apache.mina.core.service.IoHandlerAdapter;
import org.apache.mina.core.session.IdleStatus;
import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.FilterEvent;

public class TCPClientHandler extends IoHandlerAdapter {

    private static final Logger logger = Logger.getLogger(TCPClientHandler.class);

    @Override
    public void sessionCreated(IoSession session) throws Exception {
        super.sessionCreated(session);
    }

    @Override
    public void sessionOpened(IoSession session) throws Exception {
        // 向TCP服务发送消息
        session.write(new User());
    }

    @Override
    public void sessionClosed(IoSession session) throws Exception {
        super.sessionClosed(session);
    }

    @Override
    public void sessionIdle(IoSession session, IdleStatus status) throws Exception {
        super.sessionIdle(session, status);
    }

    @Override
    public void exceptionCaught(IoSession session, Throwable cause) throws Exception {
        // 断开连接异常处理
        logger.error(session.getRemoteAddress() + cause.getMessage());
    }

    @Override
    public void messageReceived(IoSession session, Object message) throws Exception {
        super.messageReceived(session, message);
    }

    @Override
    public void messageSent(IoSession session, Object message) throws Exception {
        super.messageSent(session, message);
    }

    @Override
    public void inputClosed(IoSession session) throws Exception {
        super.inputClosed(session);
    }

    @Override
    public void event(IoSession session, FilterEvent event) throws Exception {
        super.event(session, event);
    }
}

5.监听服务（用于掉线自动尝试重连）

import org.apache.log4j.Logger;
import org.apache.mina.core.service.IoService;
import org.apache.mina.core.service.IoServiceListener;
import org.apache.mina.core.session.IdleStatus;
import org.apache.mina.core.session.IoSession;

import java.net.InetSocketAddress;

public class TCPClientIoServiceListener implements IoServiceListener {

    private static final Logger logger = Logger.getLogger(TCPClientIoServiceListener.class);

    @Override
    public void serviceActivated(IoService ioService) throws Exception {

    }

    @Override
    public void serviceIdle(IoService ioService, IdleStatus idleStatus) throws Exception {

    }

    @Override
    public void serviceDeactivated(IoService ioService) throws Exception {

    }

    @Override
    public void sessionCreated(IoSession ioSession) throws Exception {

    }

    @Override
    public void sessionClosed(IoSession ioSession) throws Exception {

    }

    @Override
    public void sessionDestroyed(IoSession ioSession) throws Exception {
        logger.info(((InetSocketAddress) ioSession.getRemoteAddress()).getHostString() + "会话已销毁...");
        TCPClient.reconnection();
    }

}

完事了...