ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

go 及时通信-用户消息广播

2022-04-02 14:31:25  阅读:144  来源: 互联网

标签:string int 用户 Server 广播 User go Message server


package main

import (
	"fmt"
	"io"
	"net"
	"sync"
)

//构件server
type Server struct {
	Ip   string
	Port int

	//在线用户列表
	OnlineMap map[string]*User
	MapLock   sync.RWMutex
	//消息
	Message chan string
}

//提供一个对外的接口
func NewServer(ip string, port int) *Server {

	server := &Server{
		Ip:   ip,
		Port: port,

		OnlineMap: make(map[string]*User),
		Message:   make(chan string),
	}

	return server
}

//发送消息
func (this *Server) BroadCast(user *User, msg string) {
	sendMsg := "[" + user.Addr + "]" + ":" + msg

	this.Message <- sendMsg
}

func (this *Server) Handler(conn net.Conn) {
	// fmt.Println("链接成功")

	user := NewUser(conn)

	//用户上线
	this.MapLock.Lock()
	this.OnlineMap[user.Addr] = user
	this.MapLock.Unlock()

	//发送消息
	this.BroadCast(user, "shangxianle")

	//接收客户端消息并发送
	go func() {
		buf := make([]byte, 5000)

		for {
			//读取输入
			n, err := conn.Read(buf)
			if n == 0 {
				//用户下线
				this.BroadCast(user, "xianxianle")
				return
			}
			if err != nil && err != io.EOF {
				fmt.Println("conn read err:", err)
				return
			}

			msg := string(buf)
			this.BroadCast(user, msg)
		}

	}()

	//阻塞
	select {}
}

//监听channel
func (this *Server) ListMessager() {
	for {
		msg := <-this.Message
		//发送每个用户channel
		this.MapLock.Lock()
		for _, cli := range this.OnlineMap {
			cli.C <- msg
		}
		this.MapLock.Unlock()
	}
}

//启动
func (this *Server) start() {

	//监听客户端
	listener, err := net.Listen("tcp", fmt.Sprintf("%s:%d", this.Ip, this.Port))
	if err != nil {
		fmt.Println("listener err:", err)
		return
	}

	//关闭
	defer listener.Close()

	go this.ListMessager()

	for {
		//连接
		conn, err := listener.Accept()
		if err != nil {
			fmt.Println("conn err:", err)
			continue
		}

		//启动go handler操作
		go this.Handler(conn)
	}
}

  

标签:string,int,用户,Server,广播,User,go,Message,server
来源: https://www.cnblogs.com/alway-learn/p/16091939.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有