用户上线及广播功能

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

}