聊天2 用户上线及广播功能

main.go

package main


func main(){
	server := NewServer("1.1.1.1",10830)
	server.Start()
}

server.go

package main

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

type  Server struct {
	Ip string
	Port int
	UserMap map[string]*User
	maplock sync.RWMutex
	C chan string
}

func NewServer(ip string,port int) *Server {
	server := &Server{
		Ip:ip,
		Port: port,
		UserMap: make(map[string]*User),
		C: make(chan string),
	}
	return server
}

func (this *Server) UserWriteC(user *User,msg string){
	msg2 := "["+user.UserName+"]:"+msg
	fmt.Println("sie ru begin")
	this.C <- msg2
	fmt.Println("xie ru end")
}

func (this *Server) GuangBo(){
	for {
		msg := <-this.C
		this.maplock.Lock()
		for _, u := range this.UserMap {
			fmt.Println("-----",u.UserName)
			u.C <- msg
		}
		this.maplock.Unlock()
	}
}

func (this *Server) LoginHello(conn net.Conn){
	fmt.Println("Hello,",conn.RemoteAddr().String())
	conn.Write([]byte("登陆成功！\n"))

	user := NewUser(conn)
	this.maplock.Lock()
	this.UserMap[user.UserName]=user
	this.maplock.Unlock()
	this.UserWriteC(user,"已上线")

	select {

	}
}

func (this *Server) Start() {
	listener, err := net.Listen("tcp", fmt.Sprintf("%s:%d", this.Ip, this.Port))
	if err != nil {
		fmt.Println("net.Listen err: ", err)
	}
	defer listener.Close()
	go this.GuangBo()

	for {
		conn, err := listener.Accept()
		if err != nil {
			fmt.Println("listener.Accept() err: ", err)
			continue
		}
		go this.LoginHello(conn)
	}
}

user.go

package main

import (
	"net"
)

type User struct {
	UserName string
	UserAddr string
	C chan string
	Conn net.Conn
}

func NewUser(conn net.Conn) *User{
	var user User
	user.UserName=conn.RemoteAddr().String()
	user.UserAddr=conn.RemoteAddr().String()
	user.C = make(chan string)
	user.Conn = conn

	go user.UserReadmsg()
	//fmt.Println(user.UserName)
	//fmt.Println(user.UserAddr)
	//fmt.Println(user.C)
	//fmt.Println(user.Conn)
	//fmt.Printf("%v\n",user)

	return  &user
}

func (user *User) UserReadmsg() {
	for {
		msg := <-user.C
		user.Conn.Write([]byte(msg + "\n"))
	}
}