JAVA NIO

原创 2006年06月05日 22:41:00

import java.net.ServerSocket;
import java.net.InetSocketAddress;
import java.net.Socket;

import java.util.*;
import java.io.IOException;

import java.nio.ByteBuffer;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.Channel;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.SocketChannel;
import java.nio.charset.*;
import java.nio.CharBuffer;

import com.guanda.qqserver.util.CommandDefine;
import com.guanda.qqserver.bussiness.*;
import com.guanda.qqserver.util.Functions;
import com.guanda.qqserver.protocol.OnLineClientInfo;
import com.guanda.qqserver.util.OnLineList;

public class QQServer implements Runnable
{
  private static int port;

//  private final ByteBuffer buffer = ByteBuffer.allocate(16348);
  /**
   * 构造函数
   * @param port  服务器的监听端口
   */
  public QQServer()
  {
    port = Integer.parseInt(Functions.getProperties("tcpserver.prot"));
    new Thread(this).start();
  }

  public void run()
  {
    try
    {
      ServerSocketChannel ssc = ServerSocketChannel.open();
      ssc.configureBlocking( false );
      ServerSocket ss = ssc.socket();
      InetSocketAddress isa = new InetSocketAddress(Functions.getProperties("tcpserver.ip"),port);
      ss.bind(isa);
      Selector selector = Selector.open();
      ssc.register( selector, SelectionKey.OP_ACCEPT );
      System.out.println( "Listening on port "+port );
      while (true)
      {
        int num = selector.select();
//        System.out.println("Return Set Size: "+num);
        if (num == 0)
          continue;
        Set keys = selector.selectedKeys();

        Iterator it = keys.iterator();
        while (it.hasNext())
        {
          SelectionKey key = (SelectionKey)it.next();
          it.remove();
          if(key.isAcceptable())
          {
            //System.out.println( "accept-----" );
            Socket s = ss.accept();
            Functions.writeLog("Got connection from "+s );
            SocketChannel sc = s.getChannel();
            sc.configureBlocking(false);
            sc.register(selector,SelectionKey.OP_READ);
          }
          else if(key.isReadable())
          {
           // System.out.println("Read-----");
            SocketChannel sc = null;
            try
            {
              sc = (SocketChannel)key.channel();
              boolean ok = ProcessInput( sc );
              if(!ok)
              {
                new ChangeUserStateBuss(sc).sendOffLineMessage();
                OnLineList.removeUserFromList(sc);
                key.cancel();
                Socket s = null;
                try
                {
                  s = sc.socket();
                  s.close();
                }
                catch( IOException ie )
                {
                  Functions.writeLog(ie);
                  System.err.println("Error closing socket ");
                }
              }
            }
            catch(IOException ie)
            {
              key.cancel();
              try
              {
                sc.close();
                new ChangeUserStateBuss(sc).sendOffLineMessage();
                OnLineList.removeUserFromList(sc);
              }
              catch(IOException ie2)
              {
                Functions.writeLog(ie2);
                System.out.println( ie2 );
              }
              System.out.println( "Closed channnel:"+sc );
            }
          }
        }
      System.out.println("");
  //keys.clear();
      }
    }
    catch(IOException ie)
    {
      Functions.writeLog(ie);
      System.err.println( ie );
    }
 catch(java.nio.channels.CancelledKeyException ce)
 {
   Functions.writeLog(ce);
   System.err.println(ce);
 }
 catch(Exception ee)
 {
   Functions.writeLog("unKowned Exception !");
   System.err.println(ee);
   Functions.writeLog(ee);
 }
  }
  /**
   * 处理指定通道的输入流
   * @param 需要处理的Socket通道
   * @return true:读取到数据    false:收到客户端断开的消息
   */
  private boolean ProcessInput(SocketChannel sc) throws IOException
  {
    CharBuffer cb;
    ///包头:Command_ID
    String command_ID ="";
    ByteBuffer headpacket = ByteBuffer.allocate(8);
    sc.read(headpacket);
    headpacket.flip();
    if (headpacket.limit() == 0)
      return false;
    command_ID =  Functions.ChangeByteToString(headpacket);
    if (command_ID.equals(CommandDefine.GUANDA_CONNECT_CHECKED))
    {
      System.out.println("链路检测-------");
      return true; //链路检测
    }
    System.out.println("Command_ID = "+command_ID);
    //包头:数据包大小
    ///在此处加入消息头判断对无效消息格式中断处理
    int packetSize = this.GetPacketSize(sc);
    if (packetSize == -1)
        return false;
    System.out.println("Parcket_SIZE:"+packetSize);
    //包体
    ByteBuffer temp3 = ByteBuffer.allocate(packetSize);
    sc.read(temp3);
    temp3.flip();
    if (temp3.limit() == 0)
      return false;
    cb = Functions.ChangeByteToChar(temp3);
    if (cb != null)
    {
      System.out.print("cb:");
      PrintCharBufferInfo(cb);
      System.out.println("Packet_Content:" + cb.toString());
      //////根据COMMAND_ID选择对应的处理方法,然后将处理结果回显到客户端。
      StringBuffer packetBuffer = new StringBuffer();
      packetBuffer.append("Command_ID=").append(command_ID).append("/r/n");
      packetBuffer.append("BodySize=").append(packetSize).append("/r/n");
      packetBuffer.append("BodyContent=").append(cb.toString()).append("/r/n");
      Functions.writeLog(packetBuffer.toString());
      if (command_ID.equals(CommandDefine.GUANDA_USER_AUTH))
      {
        //客户端发送的登陆验证消息
        new UserLoginBuss(sc,cb.toString()).login();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_MESSAGE_SEND))
      {
        //客户端发送的文字信息交流消息
        new TalkBuss(sc,cb.toString()).transferMessage();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_BROADCAST_SEND))
      {
        //客户端发送过来的发送发送系统公告消息
        new BroadCastBuss(sc,cb.toString()).transferMessage();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_DEPTBROADCAST_SEND))
      {
        //客户端发送过来的发送部门公告消息
        new DeptBroadCastBuss(sc,cb.toString()).transferMessage();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_SET_USERINFO))
      {
        //客户端发送过来的饿个人设置消息
        new UserInfoBuss(sc,cb.toString()).userInfoSet();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_GET_ONLINEUSERS))
      {
        //客户端发送过来的获取在线用户列表的消息
        new UserInfoBuss(sc,cb.toString()).getOnLineUserList();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_GET_OPERATOR))
      {
        //客户端发送过来的权限验证消息
        new UserOperatorBuss(sc,cb.toString()).CheckUserOperator();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_GET_OFFLINEMESSAGE))
      {
        //客户端发送过来的获取离线消息
        new OffLineMessageSendBuss(sc,cb.toString()).chooseTranactProcess();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_CHANGESTATE_SEND))
      {
        //客户端发送过来的状态切换消息
        new ChangeUserStateBuss(sc,cb.toString()).changeUserState();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_MESSAGE_SYSTEM))
      {
        //应用系统推送过来的系统消息
        new SystemMessageBuss(sc,cb.toString()).transferMessage();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_SYSTEM_CHANGED))
      {
        //应用系统推送过来的人事变动消息
        new PCMessageBuss(sc,cb.toString()).transferMessage();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_GET_ALLIMAGEINDEX))
      {
        //得到全部用户的头象编号
        new UserInfoBuss(sc,cb.toString()).getAllImageIndex();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_GET_USERINFO))
      {
        //得到指定用户的详细信息
        new UserInfoBuss(sc,cb.toString()).getUserInfobyID();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_GET_USEROPERATOR))
      {
        //系统维护人员读取指定用户的操作权限
        new UserOperatorBuss(sc,cb.toString()).sysManReadUserOperators();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_SET_USEROPERATOR))
      {//系统维护人员设置指定用户的操作权限
        new UserOperatorBuss(sc,cb.toString()).sysManWriteUserOperators();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_SYSGET_USERIFNO))
      {////系统管理员读取用户信息
        new UserInfoBuss(sc,cb.toString()).sysManGetUserInfo();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_SYSSET_USERINFO))
      {///系统管理员设置用户信息
        new UserInfoBuss(sc,cb.toString()).sysManSetUserInfo();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_SYSGET_DUTYINFOS))
      {///读取系统职务信息
        new DutyInfoBuss(sc,cb.toString()).getDutyInfos();
      }
      else if (command_ID.equals(CommandDefine.GUANDA_DEPT_IMAGE_STATUS))
      {//读取特定部门下的用户信息和头像信息
        new UserInfoBuss(sc,cb.toString()).getDeptUserImageAndStatus();
      }
      else
      {
        System.out.println("UnKnowed Command!");
        Functions.writeLog("UnKnowed Command!");
        System.out.println("Content:" + cb.toString());
      }
    }
    PrintByteBufferInfo(temp3);
    return true;
  }
  /**
   * 打印出字符流的相关信息:
   * 1.流指针的位置 2.流指针的上限活动范围 3.字符流的容量
   * @param 需要打印的字符流
   */
  private void PrintCharBufferInfo(CharBuffer bb)
  {
    System.out.println("CharBuffer Infomation:" + bb.position() + "/"
      + bb.limit() +"/" + bb.capacity());
  }
  /**
   *打印出字节流的相关信息:
   *   1.流指针的位置 2.流指针的上限活动范围 3.字节流的容量
   * @param 需要打印的字节流
   */
  private void PrintByteBufferInfo(ByteBuffer bb)
  {
    System.out.println("ByteBuffer Infomation:" + bb.position()+"/"
      +bb.limit()+"/"+bb.capacity());
  }
  //得到数据包长度
  private int GetPacketSize(SocketChannel sc)
  {
    ByteBuffer buffer = ByteBuffer.allocate(4);
    CharBuffer cb = null;
    try
    {
      sc.read(buffer);
      int PacketSize;
      buffer.flip();
      cb = Functions.ChangeByteToChar(buffer);
      String sPacketSize = cb.toString();
      if (sPacketSize.length() < 4)
        return -1;
      int i0,i1,i2,i3,ByteCount;

      i3 = Integer.parseInt(sPacketSize.substring(3));
      i2 = Integer.parseInt(sPacketSize.substring(2,3));
      i1 = Integer.parseInt(sPacketSize.substring(1,2));
      i0 = Integer.parseInt(sPacketSize.substring(0,1));
      ByteCount = i3 + i2*10 +i1*100 + i0*1000;
      return ByteCount;
    }
    catch(NumberFormatException ex)
    {
      Functions.writeLog(ex);
      ex.printStackTrace();
      System.out.println("cb package size:" + cb.toString());
      return -1;
    }
    catch(IOException ex)
    {
      Functions.writeLog(ex);
      ex.printStackTrace();
      return -1;
    }
  }
  public static void main(String[] args)
  {
    new QQServer();
  }
}

 

package com.guanda.qqserver.util;

/**
 * <p>Title:封装好的系统公用函数 </p>
 * <p>Description: </p>
 * <p>Copyright: Copyright (c) 2004</p>
 * <p>Company: GuanDa Tech</p>
 * @author Liu QinDong
 * @version 1.0
 */

import java.nio.ByteBuffer;
import java.nio.CharBuffer;
import java.nio.charset.*;

import java.io.DataInputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.io.File;
import java.io.InputStream;
import java.io.BufferedInputStream;
import java.io.BufferedReader;
import java.io.InputStreamReader;

import java.net.URL;
import java.net.MalformedURLException;

import java.util.Properties;
import java.util.Vector;

public class Functions {
  /**
   * 将字节流转换成字符流
   * @param 需要转换的字节流
   * @return 转换后的字符流
  */
  public static synchronized CharBuffer ChangeByteToChar(ByteBuffer buffer)
  {
    Charset ch = Charset.forName("GBK");
    CharsetDecoder  decoder = ch.newDecoder();
    try
    {
      CharBuffer cha = decoder.decode(buffer);
      return cha;
    }
    catch(CharacterCodingException ce)
    {
      Functions.writeLog(ce);
      System.out.println();
      return null;
    }
  }

  /**
   * 将字节流转换成字符串
   * @param 需要转换的字节流
   * @return
   */
  public static synchronized String ChangeByteToString(ByteBuffer buffer)
  {
    Charset ch = Charset.forName("GBK");
    CharsetDecoder  decoder = ch.newDecoder();
    try
    {
      CharBuffer cha = decoder.decode(buffer);
      return cha.toString();
    }
    catch(CharacterCodingException ce)
    {
      Functions.writeLog(ce);
      System.out.println();
      return null;
    }
  }

  /**
   * 将一个字符串转换成字节流
   * @param 需要转换的字符串
   * @return 转换后的字节流
  */
  public static synchronized ByteBuffer ChangeStrToByteBuffer(String source)
  {
    CharBuffer cha = CharBuffer.wrap(source);

    Charset ch = Charset.forName("GBK");
    CharsetEncoder encoder = ch.newEncoder();
    try
    {
      ByteBuffer bf = encoder.encode(cha);
      return bf;
    }
    catch(CharacterCodingException ce)
    {
      Functions.writeLog(ce);
      System.out.println("ChangeStrToByteBuffer Error:source=" + source);
      return null;
    }
  }

}

 

 

JAVA NIO SOCKET大文件上传服务器

当前很多手机应用或者是网络应用都需要支持大文件上传功能,有些用FTP来实现上传但是FTP存在许多的问题。比如FTP的安全问题还有不支持GZIP压缩等问题。采用SOCKET来实现文件上传,很轻松就可以实...
  • reaganjava
  • reaganjava
  • 2013年03月19日 12:48
  • 2500

Java NIO 简单例子

服务器端: package nioT; import java.io.IOException; import java.net.InetSocketAddress; import java.nio...
  • whb123andy
  • whb123andy
  • 2013年12月10日 14:22
  • 3069

Java NIO 详解(一)

NIO即新的输入输出,这个库是在JDK1.4中才引入的。它在标准java代码中提供了高速的面向块的IO操作。 一、基本概念描述1.1 I/O简介I/O即输入输出,是计算机与外界世界的一个借口。IO操作...
  • suifeng3051
  • suifeng3051
  • 2015年09月14日 11:07
  • 21234

攻破JAVA NIO技术壁垒

现在使用NIO的场景越来越多,很多网上的技术框架或多或少的使用NIO技术,譬如Tomcat,Jetty。学习和掌握NIO技术已经不是一个JAVA攻城狮的加分技能,而是一个必备技能。再者,现在互联网的面...
  • u013256816
  • u013256816
  • 2016年05月19日 21:25
  • 20685

Java NIO开发需要注意的陷阱(转)

转自 陷阱1:处理事件忘记移除key 在select返回值大于0的情况下,循环处理 Selector.selectedKeys集合,每处理一个必须从Set中移除 I...
  • martin_liang
  • martin_liang
  • 2014年11月17日 23:30
  • 810

Java NIO实战之聊天室

在工作之余花了两个星期看完了《Java NIO》,总体来说这本书把NIO写的很详细,没有过多的废话,讲的都是重点,只是翻译的中文版看的确实吃力,英文水平太低也没办法,总算也坚持看完了。《Java NI...
  • abc_key
  • abc_key
  • 2014年06月07日 01:44
  • 7535

Java NIO-非阻塞通信

相对于非阻塞通信的复杂性,通常客户端并不需要使用非阻塞通信以提高性能,故这里只有服务端使用非阻塞通信方式实现...
  • a19881029
  • a19881029
  • 2013年10月17日 10:31
  • 4280

面试题:关于一些nio的问题资料

Java NIO提供了与标准IO不同的IO工作方式:  Channels and Buffers(通道和缓冲区):标准的IO基于字节流和字符流进行操作的,而NIO是基于通道(Channel)...
  • hxpjava1
  • hxpjava1
  • 2017年02月21日 14:38
  • 3936

7.Java NIO系列教程之Server/Client完整示例

TCPServer类: package com.gw.demo; import java.io.IOException; import java.net.InetSocketAddress; imp...
  • shihuacai
  • shihuacai
  • 2015年01月23日 15:22
  • 2178

Java 网络IO编程总结(BIO、NIO、AIO均含完整实例代码)

本文会从传统的BIO到NIO再到AIO自浅至深介绍,并附上完整的代码讲解。 1、BIO编程 1.1、传统的BIO编程 网络编程的基本模型是C/S模型,即两个进程间的通信。 服务...
  • anxpp
  • anxpp
  • 2016年05月29日 01:46
  • 85113
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:JAVA NIO
举报原因:
原因补充:

(最多只允许输入30个字)