Java Socket长连接示例代码

SocketListenerPusher.java代码如下: 

import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

import org.apache.commons.configuration.ConfigurationException;
import org.directwebremoting.impl.DaemonThreadFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.shihuan.dragonkeeper.common.utils.PropertiesUtil;
import com.shihuan.dragonkeeper.global.ConfigFile;

public class SocketListenerPusher implements Runnable {

    protected static Logger logger = LoggerFactory.getLogger(SocketListenerPusher.class);

    public static String socketlistenerserver_CONFIG = ConfigFile.SOCKETLISTENERSERVER__CONFIG + ConfigFile.SUFFIX_NAME;

    private ServerSocket serverSocket;
    private ExecutorService pool;


    public SocketListenerPusher() {
        int port = 0;
        int poolsize = 0;

        try {
            port = Integer.parseInt(PropertiesUtil.getPropertiesValue(socketlistenerserver_CONFIG, "serverport"));
            poolsize = Integer.parseInt(PropertiesUtil.getPropertiesValue(socketlistenerserver_CONFIG, "poolsize"));

            serverSocket = new ServerSocket();
            serverSocket.setReuseAddress(true);
            serverSocket.bind(new InetSocketAddress(port));
            pool = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors() * poolsize);

            //下面两句循环执行run()方法, 相当于while(true){...}
            ScheduledThreadPoolExecutor executor = new ScheduledThreadPoolExecutor(1, new DaemonThreadFactory());
            executor.scheduleAtFixedRate(this, 1L, 1L, TimeUnit.MILLISECONDS);

        } catch (NumberFormatException e) {
            logger.error(e.getMessage(), e);
            e.printStackTrace();
        } catch (ConfigurationException e) {
            logger.error(e.getMessage(), e);
            e.printStackTrace();
        } catch (IOException e) {
            logger.error(e.getMessage(), e);
            e.printStackTrace();
        }

    }

    public void run() {
        Socket socket = null;
        try {
            socket = serverSocket.accept();
            pool.execute(new SocketListenerHandler(socket));
        } catch (IOException e) {
            System.out.println("线程池被关闭!!!!!!!!!!!");
            pool.shutdown();
            logger.error(e.getMessage(), e);
            e.printStackTrace();
        }

    }

SocketListenerHandler.java代码如下: 

import java.io.BufferedInputStream;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.ObjectInputStream;
import java.net.Socket;
import java.sql.Connection;
import java.sql.SQLException;

import org.apache.commons.configuration.ConfigurationException;
import org.apache.commons.dbutils.DbUtils;
import org.apache.commons.dbutils.QueryRunner;
import org.apache.commons.io.IOUtils;
import org.directwebremoting.Browser;
import org.directwebremoting.ScriptSessions;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.alibaba.fastjson.JSON;
import com.shihuan.dragonkeeper.common.dto.DataSourceInfo;
import com.shihuan.dragonkeeper.common.utils.ByteArrayUtil;
import com.shihuan.dragonkeeper.common.utils.DataSourceMapUtil;
import com.shihuan.dragonkeeper.common.utils.DateFormatterUtil;
import com.shihuan.dragonkeeper.common.utils.PropertiesUtil;
import com.shihuan.dragonkeeper.global.ConfigFile;
import com.shihuan.dragonkeeper.server.bean.ActivityServiceBean;

public class SocketListenerHandler implements Runnable {

    protected static Logger logger = LoggerFactory.getLogger(SocketListenerHandler.class);

    private static String jdbc_CONFIG = ConfigFile.JDBC_CONFIG + ConfigFile.SUFFIX_NAME;

    public static final int timeOut = 0*1000 ;  //设置读取操作异常为1秒
    private final String dataRealTimeAction_id = "Agentdata_" + Math.random();

    private static final String noData = "{'nodata':'心跳信息'}";
    private static final String errorData = "{'error':'无法解析的请求'}";


    private Socket connectedsocket = null;

    public SocketListenerHandler(Socket socket){
        this.connectedsocket = socket;
    }

@Override
    public void run() {
        BufferedReader in = null;
String resultData = "";

        try {
            connectedsocket.setSoTimeout(timeOut);  //表示接收数据时的等待超时数据, 此方法必须在接收数据之前执行才有效. 此外, 当输入流的 read()方法抛出 SocketTimeoutException后, Socket仍然是连接的, 可以尝试再次读数据, 单位为毫秒, 它的默认值为 0(表示会无限等待, 永远不会超时)
            connectedsocket.setKeepAlive(false);   //表示对于长时间处于空闲状态的Socket, 是否要自动把它关闭.

            in = new BufferedReader(new InputStreamReader(connectedsocket.getInputStream()));
if (in.ready()) {  //判断流中是否有数据

                resultData = getNoHeadData(in.readLine());   //从Agent端接收到的数据
                logger.info("#### 结果DATA = "+resultData);

                if (resultData==null || "".equals(resultData)) {
                    logger.info(dataRealTimeAction_id + " -->>> " + "内容为空!");
                } else if (resultData.charAt(0) != '{') {  //要在客户端定时维持心跳信息
                    logger.info(dataRealTimeAction_id + " -->>> " + noData);
                } else {
                    ActivityServiceBean asb = JSON.parseObject(resultData, ActivityServiceBean.class);
                    System.out.println("打印预处理信息Start......");
                    System.out.println(asb.getProxyname() + " -- " + asb.getIp() + " -- " + asb.getCalltime() + " -- " + asb.getAnswertime() + " -- " + asb.getCpu() + " -- " + asb.getThread() + " -- " + asb.getStatus() + " -- " + asb.getAccessaddress() + " -- " + asb.getAccessfilename() + " -- " + asb.getSql() + " -- " + asb.getContent());
                    System.out.println("打印预处理信息End......");
//                  parseData(ois);

                    logger.info(dataRealTimeAction_id + ": 成功处理了接收到的数据!");
                }

            }
} catch (IOException e) {
            logger.error(e.getMessage() + " " + errorData, e);
            e.printStackTrace();
        } catch (NumberFormatException e) {
            logger.error(e.getMessage(), e);
            e.printStackTrace();
        } finally {

            if (in != null) {
                try {
                    in.close();
                } catch (IOException e) {
                    logger.error(e.getMessage(), e);
                    e.printStackTrace();
                }
            }
}

    }

TestSocketListenerPusher.java请求端代码如下: 

import java.io.BufferedOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.Socket;
import java.net.UnknownHostException;
import java.util.Date;

import org.apache.commons.configuration.ConfigurationException;

import com.alibaba.fastjson.JSON;
import com.shihuan.dragonkeeper.common.utils.ByteArrayUtil;
import com.shihuan.dragonkeeper.common.utils.PropertiesUtil;
import com.shihuan.dragonkeeper.global.ConfigFile;
import com.shihuan.dragonkeeper.server.bean.ActivityServiceBean;

public class TestSocketListenerPusher implements Runnable {

    private static String socketlistenerserver_CONFIG = ConfigFile.SOCKETLISTENERSERVER__CONFIG + ConfigFile.SUFFIX_NAME;

    private Socket socketclient = null;

    @Override
    public void run() {

        String serverip = "";
        int port = 0;

        OutputStream os = null;

        try {
            serverip = PropertiesUtil.getPropertiesValue(socketlistenerserver_CONFIG, "serverip");
            port = Integer.parseInt(PropertiesUtil.getPropertiesValue(socketlistenerserver_CONFIG, "serverport"));


            ActivityServiceBean asb = null;

            for (int i=0; i<2; i++) {
                asb = new ActivityServiceBean();
                asb.setProxyname("testProxyname"+i);
                asb.setIp("testIp"+i);
                Date curdate = new Date();
                asb.setCalltime(curdate);
                asb.setAnswertime(curdate);
                asb.setCpu("testCpu"+i);
                asb.setThread("testThread"+i);
                asb.setStatus("testStatus"+i);
                asb.setAccessaddress("testAccessaddress"+i);
                asb.setAccessfilename("testAccessfilename"+i);
                asb.setSql("testSql"+i);
                asb.setContent("testContent"+i);

                String jsonStr = JSON.toJSONString(asb).trim();
                byte[] information = (new String(ByteArrayUtil.getIntToByte(jsonStr.length()))+jsonStr).getBytes();

                System.out.println(information.length);

                socketclient = new Socket(serverip, port);
                socketclient.setSoTimeout(0);
                socketclient.setKeepAlive(false);

                os = new BufferedOutputStream(socketclient.getOutputStream());
                os.write(information);
                os.flush();

                System.out.println("Client" + i + " -->>> " + new String(ByteArrayUtil.getIntToByte(jsonStr.length()))+jsonStr);

                os.close();
                Thread.sleep(3000);
            }


        } catch (ConfigurationException e) {
            e.printStackTrace();
        } catch (UnknownHostException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            /*
            try {
                if (os != null) {
                    os.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
            */
        }


    }

    public static void main(String[] args) {
        Thread t = new Thread(new TestSocketListenerPusher());
        t.start();
    }

}

源代码在笔者shihuan8@163.com邮箱网盘中J2EE代码文件夹里。 

---------------------------------------------------------------------------------- 
如果是按byte[]传输数据的情况,请参考如下代码: 

SimpleSocketServer.java代码如下: 

package com.shihuan.socket;

import java.io.BufferedInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.net.Socket;

public class SimpleSocketServer {

    public static void main(String[] args) {
        try {
            ServerSocket ss = new ServerSocket();
            ss.setReuseAddress(true);  //两个进程共用同一个端口的时候,一个进程关闭后,另一个进程还能够立刻重用相同端口
            ss.setReceiveBufferSize(128*1024);  //缓冲区中允许接收的最大字节数,默认是8192
            ss.bind(new InetSocketAddress(19990));

            Socket client = ss.accept();
            InputStream in = new BufferedInputStream(client.getInputStream());

            byte tmpb = (byte)in.read();
            System.out.println("第一个字节的byte值 --->> " + tmpb);
            System.out.println("接收字节 --->> " + in.available());
            byte[] bc = new byte[in.available()+1];
            bc[0] = tmpb;
            in.read(bc, 1, in.available());

            System.out.println(bc.length);
            System.out.println(new String(bc));
            in.close();
        } catch (IOException e) {
            System.out.println(e.getMessage());
            e.printStackTrace();
        }

    }

}

SimpleSocketClient.java代码如下: 

package com.shihuan.socket;

import java.io.BufferedOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.Socket;
import java.net.UnknownHostException;

public class SimpleSocketClient {

    public static void main(String[] args) throws UnknownHostException {
        try {
            Socket s = new Socket("192.168.1.10", 19990);
            OutputStream os = new BufferedOutputStream(s.getOutputStream());
            String info = "abc!";
            info = "大家好!";
            byte[] bi = info.getBytes();
            os.write(bi);
            os.flush();
            os.close();
        } catch (IOException e) {
            System.out.println(e.getMessage());
            e.printStackTrace();
        }
    }

}

稍微复杂一点儿代码示例，处理了粘包问题： 
StartListenerTcpThread.java代码: 

import java.io.BufferedInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.SocketAddress;
import java.util.Vector;
import java.util.concurrent.ExcutorService;
import java.util.concurrent.Excutors;

import org.apache.commons.io.IUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.shihuan.dragonkeeper.common.utils.ByteArrayUtil;
import com.shihuan.dragonkeeper.global.ConfigFile;

public class StartListenerTcpThread implements Runnable {
   public static Logger logger = LoggerFactory.getLogger(StartListenerTcpThread.class);

   private static ExcutorService Threadpool = Excutors.newCachedThreadPool();
   private static boolean businessflag = true;
   private static final int receiveBufferSize = 128;
   private static Vector<byte[]> tmpbytes = new Vector<byte[]>();
   private ServerSocket serverSocket = null;

   public StartListenerTcpThread(String ip, int port){
      try{
         serverSocket = new ServerSocket();
         serverSocket.setReuseAddress(true);
         serverSocket.setReceiveBufferSize(receiveBufferSize*1024);
         serverSocket.setSoTimeout(0);
         SocketAddress sa = new InetSocketAddress(port);
         serverSocket.bind(sa, 20);
      }catch(IOException e){
         logger.error(e.getMessage(), e);
      }
   }

   public void run(){
      Socket socket = null;
      while(true){
         if(businessflag){
            try{
               socket = serverSocket.accept();
               System.out.println("New connection accepted " + socket.getInetAddress() + ":" + socket.getPort());
               InputStream socketIn = new BufferedInputStream(socket.getInputStream());
               byte tmpb = (byte)socketIn.read();
               byte[] currentbytes = null;
               if(tmpbytes.size() > 0){  //上一次IO流中有未处理的剩余包
                  int oldBytesLen = tmpbytes.get(0).length;
                  int socketBytesLen = socketIn.available()+1;
                  int currentLength = oldByteLen + socketBytesLen;
                  currentbytes = new byte[currentLength];
                  System.arraycopy(tmpbytes.get(0), 0, currentbytes, oldBytesLen);
                  currentbytes[oldBytesLen] = tmpb;
                  socketIn.read(currentbytes, oldBytesLen+1, socketBytesLen-1);
                  socketIn.close();
                  splitInputStreamByte(currentbytes);
               }else{  //正常未粘包情况
                  int socketBytesLen = socketIn.available()+1;
                  currentbytes = new byte[socketBytesLen];
                  currentbytes[0] = tmpb;
                  socketIn.read(currentbytes, 1, socketBytesLen-1);
                  socketIn.close();
                  splitInputStreamByte(currentbytes);
               }
            }catch(IOException e){
               logger.error(e.getMessage(), e);
            }
         }
      }
   }

   /**
    * 拆分byte数组并分多线程处理
    * @param parambytes 原byte数组
    * @return 处理后剩余部分的byte数组
   */
   private static void splitInputStreamByte(byte[] parambytes) {
      if(parambytes != null){
         if(parambytes.length > 4){
            byte[] head = new byte[4];  //单包长度
            System.arraycopy(parambytes, 0, head, 0, 4);
            int bodyLength = ByteArrayUtil.getint(head);
            if(bodyLength <= parambytes.length-4){
               final byte[] body = new byte[bodyLength];
               System.arraycopy(parambytes, 4, body, 0, bodyLength);
               ThreadPool.execute(new Runnable(){
                  public void run(){
                     byte[] processDatas = body;
                     try{
                        System.out.println(IOUtils.toString(processDatas, "UTF-8").trim());
                     }catch(IOException e){
                        logger.error(e.getMessage(), e);
                     }
                  }
               });

               int resultLen = parambytes.length-4-bodyLength;
               if(resultLen == 0){
                  splitInputStreamByte(null);
               }else{
                  byte[] resultbytes = new byte[resultLen];
                  System.arraycopy(parambytes, 4+bodyLength, resultbytes, 0, resultLen);
                  splitInputStreamByte(resultbytes);
               }
            }else{
               tmpbytes.clear();
               tmpbytes.add(parambytes);
            }
         }else{
            tmpbytes.clear();
            tmpbytes.add(parambytes);
         }
      }
   }

   public static void openflag(){
      businessflag = true;
   }

   public static void closeflag(){
      businessflag = false;
   }

}

TestTcpSocket.java代码： 

import java.io.IOException;
import java.io.OutputStream;
import java.net.Socket;
import java.net.UnknownHostException;

import com.shihuan.dragonkeeper.common.utils.ByteArrayUtil;
import com.shihuan.dragonkeeper.global.ConfigFile;

public class TestTcpSocket implements Runnable{
   private Socket socketClient = null;

   public void run(){
      String serverip = "192.168.1.10";
      int port = 19990;
      try{
         while(true){
           System.out.println("SocketClient start......");
           String mystr = "hello everyone!";
           socketClient = new Socket(serverip, port);
           OutputStream os = socketClient.getOutputStream();
           byte[] head = ByteArrayUtil.int2byte(mystr.length());
           byte[] body = mystr.getBytes();
           byte[] total = ByteArrayUtil.byteMerge(head, body);

           os.write(total);
           os.flush();
           os.close();

           Thread.sleep(1000);

           System.out.println("SocketClient end......");
         }
      }catch(Exception e){
         logger.error(e.getMessage(), e);
      }
   }

   public static void main(String[] args){
      Thread t = new Thread(new TestTcpSocket());
      t.start();
   }
}

下面写ByteArrayUtil.java代码： 

package com.shihuan.dragonkeeper.common.utils;

public class ByteArrayUtil {

   /**
    * 将int型的数据类型转换成byte[]类型
   */
   public static final byte[] int2byte(int paramInt){
      byte[] resultByte = new byte[4];
      resultByte[3] = ((byte)(paramInt & 0xFF));
      resultByte[2] = ((byte)(paramInt >>> 8 & 0xFF));
      resultByte[1] = ((byte)(paramInt >>> 16 & 0xFF));
      resultByte[0] = ((byte)(paramInt >>> 24 & 0xFF));

      return resultByte;
   }

   /**
    * 将byte型的数据类型转换成int类型
   */
   public static final int getint(byte[] paramArrayOfByte){
      int result = (paramArrayOfByte[0] & 0xFF) << 24 | (paramArrayOfByte[1] & 0xFF) << 16 | (paramArrayOfByte[2] & 0xFF) << 8 | paramArrayOfByte[3] & 0xFF;
      return result;
   }

   /**
    * 合并两个byte数组到一个byte数组中
   */
   public static byte[] byteMerge(byte[] byte1, byte[] byte2){
      byte[] result = new byte[byte1.length+byte2.length];
      System.arraycopy(byte1, 0, result, 0, byte1.length);
      System.arraycopy(byte2, 0, result, byte1.length, byte2.length);
      return result;
   }
}

http://blog.csdn.net/defonds/article/details/8782785