socket长连接客户端管理

今天看到有人在ask上提问关于socket长连接发送消息的问题，所以突然就想着写一篇博客。

为刚接触socket长连接开发，并且没有用mina或netty这种nio框架的开发者提出的一些思路。

可以根据自己的业务进行改变。

下面就是具体代码（只有对连接的管理，具体的连接没有提供）：

 

package littlehow.socket;

import java.io.IOException;
import java.io.OutputStream;
import java.net.Socket;
import java.nio.charset.Charset;
import java.util.Enumeration;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * SocketClientManager
 *   如果管理socket客户端长连接
 *   假设连接在10000以内,可以使用10个concurrenthashmap来进行管理
 *   如果使用的是apache mina或者netty的，socket连接id就不用自己管理，
 *   其内部有实现id的一套管理方法
 *
 * @author littlehow
 * @time 2016-06-17 09:37
 */
public class SocketClientManager {
    //全局字符编码
    private static final Charset charset;
    //数组长度
    private static int table_length = 10;
    //初始id为从1开始
    private final static AtomicInteger common_id = new AtomicInteger(1);
    //在线人数
    private final static AtomicInteger onlineCount = new AtomicInteger(0);

    /**
     * 客户端管理类
     */
    static ConcurrentHashMap<Integer, Session>[] clients = new ConcurrentHashMap[table_length];
    static {
        for (int i = 0; i < table_length; i ++) {
            clients[i] = new ConcurrentHashMap<>();
        }
        charset = Charset.forName("UTF-8");
    }
    /**
     * socket客户端轻装类
     */
    static class Session {
        /**
         * id标识
         */
        private int id;
        /**
         * socket连接
         */
        private Socket socket;
        public Session(Socket socket) {
            id = common_id.getAndAdd(1);
            this.socket = socket;
        }
        public void close() throws IOException {
            this.socket.close();
        }
        public Integer getId() {
            return id;
        }
        public Socket getSocket() {
            return socket;
        }
    }

    /**
     * 获取对应的map
     * @param id
     * @return
     */
    private static ConcurrentHashMap<Integer, Session> getTable (int id) {
        return clients[id % table_length];
    }

    /**
     * 进入一个连接,因为id是均匀增加的，所以连接会均匀分布到10个map中
     * @param socket
     */
    public static Integer online (Socket socket) {
        Session session = new Session(socket);
        ConcurrentHashMap<Integer, Session> client = getTable(session.getId());
        client.put(session.getId(), session);
        onlineCount.incrementAndGet();
        return session.getId();
    }

    /**
     * 某一个id下线
     * @param id
     */
    public static void offline (int id) throws IOException {
        Session session = getTable(id).remove(id);
        if (session != null) {
            onlineCount.decrementAndGet();
            session.close();
        }
    }

    /**
     * 这里可以扩展字符编码
     * @param id
     * @param message
     * @throws IOException
     */
    public static void sendMessageOne (Integer id, String message) throws IOException {
        sendMessageOne(id, message.getBytes(charset));
    }

    public static void sendMessageOne(Integer id, byte[] message) throws IOException {
        Session session = getTable(id).get(id);
        if (session != null) {
            writeMsg(message, session.getSocket());
        }
    }

    /**
     * 千万不要关闭，不然长连接会断开
     * @param message
     * @param socket
     * @throws IOException
     */
    private static void writeMsg (byte[] message, Socket socket) throws IOException {
        OutputStream os = socket.getOutputStream();
        os.write(message);
        os.flush();
    }

    /**
     * 为一个map下群发消息
     * @param index
     */
    public static void sendMessageOneMap(int index, String message) {
        ConcurrentHashMap<Integer, Session> client = clients[index];
        Enumeration<Integer> keys = client.keys();
        while (keys.hasMoreElements()) {
            try {
                writeMsg(message.getBytes(charset), client.get(keys.nextElement()).getSocket());
            } catch (IOException e) {
                //可以对异常的连接进行处理，最终清理出离线人员。
            }
        }
    }

    /**
     * 广播消息
     * @param message
     */
    public static void broadcast(String message) {
        for (int i = 0; i < table_length; i ++) {
            sendMessageOneMap(i, message);
        }
    }

    /**
     * 获取当前在线人数
     * @return
     */
    public static int currentOnline() {
        return onlineCount.get();
    }
}