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