main.go:
package main
import "fmt"
func main() {
server := NewServer("127.0.0.1",8888)
if !server.Start() {
fmt.Println("服务器启动失败")
return
}
}
server.go:
监听线程发现新连接时,把每个连接记录到map中(用锁保证数据安全),当有新连接过来时则通知已存在的连接。
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.L