手机端小强原创文章,java小强个人博客站点
当前位置: 首页 >> 开源 >> Mina控制IP连接量和连接总数量

Mina控制IP连接量和连接总数量

59630 开源 | 2014-7-30

要求就是可以控制某个IP只能连接一次客户端,或连接的总数量不能超过设定的值。这里使用了apache-mina-2.0.7版本,请注意。

其实这个可以参考官方的一个聊天示例,来一个客户端的时候就缓存下来,走的时候移除。

这里就稍作改造,这里会有一个缓存客户端session的Map,还有一个连接总数的计数器,设定一个IP只能连接一个客户端,然后连接总数不能超过2个。示例中客户端会连接三个连接,休眠后中断连接。

服务端代码:

package com.test;
import java.io.IOException;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.util.HashMap;
import java.util.Map;
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.transport.socket.SocketSessionConfig;
import org.apache.mina.transport.socket.nio.NioSocketAcceptor;
public class TcpServer extends IoHandlerAdapter {
 public static final int PORT = 18567;
 private static Map<String, IoSession> sessions_ip = new HashMap<String, IoSession>(); // 控制某个IP的连接
 private static int sessionSize = 0; // 控制连接总量
 @Override
 public void exceptionCaught(IoSession session, Throwable cause) throws Exception {
  cause.printStackTrace();
  session.close(true);
 }
 @Override
 public void messageReceived(IoSession session, Object message) throws Exception {
 }
 @Override
 public void sessionClosed(IoSession session) throws Exception {
  InetSocketAddress inetSocketAddress = (InetSocketAddress)session.getRemoteAddress();
  InetAddress address = inetSocketAddress.getAddress();
  String ip = address.getHostAddress();
  int port = inetSocketAddress.getPort();
  // 这里是有这个IP才能移除且总数减一,如果你不想以IP来控制,只是要控制总数,那么在客户端来的时候就要以(IP+端口)为依据来缓存该客户端,然后做出减一操作
  if(null != sessions_ip.get(ip)) {
   System.out.println("客户端离开:" + ip + ":" + port);
   sessions_ip.remove(ip + ":" + port);
   sessionSize--;
  }
  System.out.println("Session closed...");
 }
 @Override
 public void sessionCreated(IoSession session) throws Exception {
  System.out.println("Session created...");
 }
 @Override
 public void sessionIdle(IoSession session, IdleStatus status) throws Exception {
  System.out.println("Session idle...");
 }
 @Override
 public void sessionOpened(IoSession session) throws Exception {
  if(sessionSize > 2){
   System.err.println("客户端超出最大数量");
   session.close(true);
   return;
  }
  InetSocketAddress inetSocketAddress = (InetSocketAddress)session.getRemoteAddress();
  InetAddress address = inetSocketAddress.getAddress();
  String ip = address.getHostAddress();
  int port = inetSocketAddress.getPort();
  if(null != sessions_ip.get(ip)) {
   System.err.println("该IP已有客户端连接,禁止新链接");
   session.close(true);
   return;
  }
  System.out.println("新增客户端:" + ip + ":" + port);
  sessions_ip.put(ip, session);
  sessionSize++;
  System.out.println("Session Opened...");
 }
 public TcpServer() throws IOException {
  NioSocketAcceptor acceptor = new NioSocketAcceptor();
  acceptor.setHandler(this);
  SocketSessionConfig scfg = acceptor.getSessionConfig();
  acceptor.bind(new InetSocketAddress(PORT));
  System.out.println("Server started...");
 }
 public static void main(String[] args) throws IOException {
  new TcpServer();
 }
}

客户端代码:

package com.test;
import java.net.InetSocketAddress;
import org.apache.mina.core.future.ConnectFuture;
import org.apache.mina.core.service.IoConnector;
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.transport.socket.SocketSessionConfig;
import org.apache.mina.transport.socket.nio.NioSocketConnector;
public class TcpClient extends IoHandlerAdapter {
 private IoConnector connector;
 private static IoSession session;
 public TcpClient() {
  connector = new NioSocketConnector();
  connector.setHandler(this);
  SocketSessionConfig dcfg = (SocketSessionConfig) connector.getSessionConfig();
  ConnectFuture connFuture = connector.connect(new InetSocketAddress("localhost", TcpServer.PORT));
  connFuture.awaitUninterruptibly();
  session = connFuture.getSession();
 }
 @Override
 public void exceptionCaught(IoSession session, Throwable cause) throws Exception {
  cause.printStackTrace();
 }
 @Override
 public void messageReceived(IoSession session, Object message) throws Exception {
 }
 @Override
 public void messageSent(IoSession session, Object message) throws Exception {
 }
 @Override
 public void sessionClosed(IoSession session) throws Exception {
 }
 @Override
 public void sessionCreated(IoSession session) throws Exception {
 }
 @Override
 public void sessionIdle(IoSession session, IdleStatus status) throws Exception {
 }
 @Override
 public void sessionOpened(IoSession session) throws Exception {
 }
 public static void main(String[] args) throws Exception {
  TcpClient client1 = new TcpClient();
  TcpClient client2 = new TcpClient();
  TcpClient client3 = new TcpClient();
  Thread.sleep(3000);
  client1.connector.dispose(true);
  client2.connector.dispose(true);
  client3.connector.dispose(true);
 }
}

看服务端和客户端的打印:

SLF4J: Failed to load class "org.slf4j.impl.StaticLoggerBinder".
SLF4J: Defaulting to no-operation (NOP) logger implementation
SLF4J: See http://www.slf4j.org/codes.html#StaticLoggerBinder for further details.
Server started...
Session created...
新增客户端:127.0.0.1:57246
Session Opened...
该IP已有客户端连接,禁止新链接
Session created...
客户端离开:127.0.0.1:57255
Session closed...
该IP已有客户端连接,禁止新链接
Session created...
客户端离开:127.0.0.1:57264
Session closed...
客户端离开:127.0.0.1:57246
Session closed...

客户端不打印。

可以看到本地一个连接连接上去以后,就会开始拒绝新的连接了。至于连接总数的控制,我的注释里面有解释,代码也有这个功能,可以参考下。

推荐您阅读更多有关于“ 网络编程 开源 mina 连接总数 ”的文章

上一篇:windows service2003 64位 配置使用PLSQL Developer 下一篇:Mina重连 千万别这么干

猜你喜欢

发表评论:

个人资料
blogger

java小强
没有思考,人生的路会越走越难!

搜索
分类
最新微语
  • 车也学了,年也过了,生日也过了,村里的会也赶了,这次,是真的,年过去了。不过我没回京,也没有在家找工作,我在等什么吗?反正现在正合了我这个懒人的要求,不过,我歇不住,思考下人生。

    2018-03-20 00:11

  • 8月1日,我已离开奋斗多年的北京。不知道是暂时的离开,还是永久的离别,反正已经离职在家,告别每日上班,每天苦累的煎熬,过一段属于自己的生活。以前是专职工作,现在专职生活。

    2017-08-18 12:47

  • 又弄完一个项目,累成狗,但是感觉又进步不少,除了很多坑已经踩过,做起来也是轻车熟路。同时也认识到,程序不在于你多牛逼,而是在乎你的细节把控度,而细节的关注,是一个优秀程序员必须要注意的。另外,要相信自己,勇敢向前,没人生下来就是成功的,而且,成功的路,比成功本身更重要。

    2017-06-30 09:46

  • 今日北京再次沙尘暴来袭,吃了几年细粮,终于能来口粗粮了,不过大早上看见这场景,还是吓我一跳,不过随后就平静了,毕竟是老朋友了。进公司又发现一股烤糊的味道,真是祸不单行啊,例外都是污染。发了两个口罩,开启保护模式。

    2017-05-04 10:16

  • 今天同学问我,最近还在写代码吗?我想了想,这个问题怎么回答呢,我好像确实很长时间,虽然写了一些,但是主要内容已经不是写代码了。然后再想想,自己也7年多了,这么多年了,我收获了什么,我的目标到底是什么。眼看就奔三了,人生啊,开启感叹模式。

    2017-03-30 22:52

  • 更多»

最新文章
热门文章
随机文章