标签：上线 string chan 用户 广播 User net conn user

package main

import "net"

type User struct {
	Name string
	Addr string

	C    chan string
	conn net.Conn
}

func newUser(conn net.Conn) *User {
	userAddr := conn.RemoteAddr().String()

	user := &User{
		Name: userAddr,
		Addr: userAddr,
		C:    make(chan string),
		conn: conn,
	}

	//开启goroutine
	go user.ListenMessage()

	return user
}

//监听当前用户channel
func (this *User) ListenMessage() {
	for {
		msg := <-this.C

		this.conn.Write([]byte(msg + "\n"))
	}
}

package main

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

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) ChrodCast(user *User, msg string) {
	sendMsg := "[" + user.Addr + "]" + user.Name + ":" + msg

	this.Message <- sendMsg
}

func (this *Server) Handler(conn net.Conn) {
	// fmt.Println("已连接")

	user := newUser(conn)

	this.maplock.Lock()
	this.OnlineMap[user.Name] = user
	this.maplock.Unlock()

	//发送消息
	this.ChrodCast(user, "已上线")

	select {}
}

//监听当前channel 消息
func (this *Server) ListenMessager() {

	for {
		msg := <-this.Message

		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("listen 监听err：", err)
	}

	defer listener.Close()

	go this.ListenMessager()

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

		//handler
		go this.Handler(conn)
	}

}

标签：上线,string,chan,用户,广播,User,net,conn,user
来源： https://www.cnblogs.com/alway-learn/p/16083044.html

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