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();
	}

}



源代码在笔者[email protected]邮箱网盘中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

猜你喜欢

转载自shihuan830619.iteye.com/blog/2041774