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

 

  • 1
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
以下是一个简单的 Java Socket 客户连接代码示例: ```java import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; import java.net.Socket; public class SocketClient { private static final String SERVER_HOST = "localhost"; // 服务地址 private static final int SERVER_PORT = 8888; // 服务口号 public static void main(String[] args) { try { Socket socket = new Socket(SERVER_HOST, SERVER_PORT); // 发送消息 OutputStream outputStream = socket.getOutputStream(); outputStream.write("Hello, Server!".getBytes()); // 接收消息 InputStream inputStream = socket.getInputStream(); byte[] buffer = new byte[1024]; int length; while ((length = inputStream.read(buffer)) != -1) { String message = new String(buffer, 0, length); System.out.println("Receive message from server: " + message); // 发送消息 outputStream.write(("I received: " + message).getBytes()); } // 关闭连接 socket.close(); } catch (IOException e) { e.printStackTrace(); } } } ``` 在该示例中,我们通过 `Socket` 类连接到一个指定的服务器地址和口。然后,我们使用 `OutputStream` 发送消息,使用 `InputStream` 接收消息,并在接收到消息后再次使用 `OutputStream` 发送响应。最后,我们关闭了与服务器的连接。 该代码示例演示了如何建立一个简单的 Java Socket 客户连接。然而,在实际开发中,你需要考虑更多的因素,如线程管理、异常处理、心跳包等。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值