go-chat/process/userManager.go

package process

import (
	"fmt"
	"net"
)

type UserProcess struct {
	Uid  int
	Conn net.Conn
}

var onlineUsers = make(map[int]*UserProcess)

func GetOnlineUsers() map[int]*UserProcess {
	return onlineUsers
}

func Push(process *UserProcess) {
	onlineUsers[process.Uid] = process
}

func Get(id int) *UserProcess {
	u, ok := onlineUsers[id]
	if ok {
		return u
	}
	return nil
}

func Del(id int) {
	delete(onlineUsers, id)
}

func Disconnect(conn net.Conn) int {
	for u, process := range GetOnlineUsers() {
		if conn.RemoteAddr() == process.Conn.RemoteAddr() {
			fmt.Println("用户", u, "已下线")
			delete(onlineUsers, u)
			return u
		}
	}
	return 0
}