forked from Tnze/go-mc
/
playerlist.go
103 lines (87 loc) · 2.29 KB
/
playerlist.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package server
import (
"sync"
"github.com/google/uuid"
"github.com/apepenkov/go-mc/chat"
)
type PlayerListClient interface {
SendDisconnect(reason chat.Message)
}
// PlayerList is a player list based on linked-list.
// This struct should not be copied after used.
type PlayerList struct {
maxPlayer int
players map[PlayerListClient]PlayerSample
// Only the field players is protected by this Mutex.
// Because others field never change after created.
playersLock sync.Mutex
}
// NewPlayerList create a PlayerList which implement ListPingHandler.
func NewPlayerList(maxPlayers int) *PlayerList {
return &PlayerList{
maxPlayer: maxPlayers,
players: make(map[PlayerListClient]PlayerSample),
}
}
func (p *PlayerList) ClientJoin(client PlayerListClient, player PlayerSample) {
p.playersLock.Lock()
defer p.playersLock.Unlock()
if len(p.players) >= p.maxPlayer {
client.SendDisconnect(chat.TranslateMsg("multiplayer.disconnect.server_full"))
return
}
p.players[client] = player
}
func (p *PlayerList) ClientLeft(client PlayerListClient) {
p.playersLock.Lock()
defer p.playersLock.Unlock()
delete(p.players, client)
}
// CheckPlayer implements LoginChecker for PlayerList
func (p *PlayerList) CheckPlayer(string, uuid.UUID, int32) (ok bool, reason chat.Message) {
p.playersLock.Lock()
defer p.playersLock.Unlock()
if len(p.players) >= p.maxPlayer {
return false, chat.TranslateMsg("multiplayer.disconnect.server_full")
}
return true, chat.Message{}
}
func (p *PlayerList) MaxPlayer() int {
return p.maxPlayer
}
func (p *PlayerList) OnlinePlayer() int {
p.playersLock.Lock()
defer p.playersLock.Unlock()
return len(p.players)
}
func (p *PlayerList) PlayerSamples() (sample []PlayerSample) {
p.playersLock.Lock()
defer p.playersLock.Unlock()
// Up to 10 players can be returned
length := len(p.players)
if length > 10 {
length = 10
}
sample = make([]PlayerSample, length)
var i int
for _, player := range p.players {
sample[i] = player
i++
if i >= length {
break
}
}
return
}
func (p *PlayerList) Len() int {
p.playersLock.Lock()
defer p.playersLock.Unlock()
return len(p.players)
}
func (p *PlayerList) Range(f func(PlayerListClient, PlayerSample)) {
p.playersLock.Lock()
defer p.playersLock.Unlock()
for client, player := range p.players {
f(client, player)
}
}