Golang TCP服务器群聊消息

Posted 2021-05-20 卑微的小李

main.go:

 

package main

import (
	"fmt"
)

func main()  {
	server := NewServer("127.0.0.1",8888)
	if !server.Start() {
		fmt.Println("服务器启动失败")
		return
	}

}

 

 

server.go

增加广播函数，轮询往当前在线的设备中发送数据。

 

package main

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

type Server struct {
	Ip string
	Port int

	users map[string]*User
	userMutex sync.RWMutex
	msgC chan string
}

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

func (this *Server) handler(conn net.Conn)  {
	fmt.Println(fmt.Sprintf("%s: 连接成功",conn.RemoteAddr().String()))
	this.userMutex.Lock()
	this.users[conn.RemoteAddr().String()] = NewUser(conn,this)
	this.userMutex.Unlock()
}

func (this *Server) Start() bool {
	listener, err := net.Listen("tcp", fmt.Sprintf("%s:%d",this.Ip,this.Port))
	defer listener.Close()

	if err != nil {
		fmt.Println("net listen err:",err)
		return false
	}

	go this.listenMsg()

	for  {
		con, err := listener.Accept()
		if err != nil {
			fmt.Println("net Accept err:",err)
			continue
		}
		this.userOnline(con.RemoteAddr().String())
		go this.handler(con)
	}

	return true
}

func (this *Server) listenMsg()  {
	for  {
		msg:= <- this.msgC
		this.brodCast(msg)
	}
}

func (this *Server) brodCast(msg string )  {
	this.userMutex.Lock()
	for _, v := range this.users {
		sendMsg := "[" + v.Conn.RemoteAddr().String()+"]"+v.Name+":"+msg+"\\n"
		_,err := v.Conn.Write([]byte(sendMsg))
		if err != nil {
			fmt.Println("server brodCast err:",err)
		}
	}
	this.userMutex.Unlock()
}

func (this *Server) userOnline(name string)  {
	this.userMutex.Lock()
	for _, v := range this.users {
		_,err := v.Conn.Write([]byte(fmt.Sprintf("user:%s online",name)))
		if err != nil {
			fmt.Println("server brodCast err:",err)
		}
	}
	this.userMutex.Unlock()
}

func (this *Server) userOffline(ip string)  {
	this.userMutex.Lock()
	username := this.users[ip].Name
	delete(this.users, ip)
	for _, v := range this.users {
		_,err := v.Conn.Write([]byte(fmt.Sprintf("user:%s offline",username)))
		if err != nil {
			fmt.Println("server brodCast err:",err)
		}
	}
	this.userMutex.Unlock()
}

 

 

user.go

在消息监听函数中不断的阻塞读取消息，如果数据为0设备下线。

如果数据开头是all：则判断为广播数据，轮询往当前在线的设备中发送数据。

 

package main

import (
	"fmt"
	"net"
	"runtime"
	"strings"
)

type User struct {
	Name string
	Address string
	C chan string
	Conn net.Conn
	server *Server
}

func NewUser(conn net.Conn,ser *Server) *User  {
	user := &User{
		Name: conn.RemoteAddr().String(),
		Address: conn.RemoteAddr().String(),
		C: make(chan string),
		Conn: conn,
		server: ser,
	}

	go user.listenMessage()

	return user
}

func (this *User) listenMessage()  {

	for  {
			msg :=make([]byte,4096)
			num,err :=this.Conn.Read(msg)
			if err != nil {
				fmt.Println("user Read err:",err)
			}
			if num==0 {
				fmt.Println(fmt.Sprintf("user:%s device offline.",this.Conn.RemoteAddr().String()))
				this.server.userOffline(this.Conn.RemoteAddr().String())
				_ = this.Conn.Close()
				close(this.C)
				runtime.Goexit()
			}

		msgStr := string(msg[:num])
		if len(msgStr)>4 && msgStr[:4]=="all:" {
			//fmt.Println(strings.Split(msgStr,":")[1])
			this.server.brodCast(strings.Split(msgStr,":")[1])
		}
	}
}

 

 

 

 

 

目录：

Golang 从入门到放弃