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