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
|
package main
import (
"encoding/json"
"errors"
)
type MsgType int
const (
MsgChat MsgType = iota // Default
MsgJoin
MsgLeave
)
func (mt *MsgType) MarshalJSON() ([]byte, error) {
switch *mt {
case MsgChat:
return json.Marshal("chat")
case MsgJoin:
return json.Marshal("join")
case MsgLeave:
return json.Marshal("leave")
}
return nil, errors.New("Unknown message type")
}
type Message struct {
Type MsgType `json:"type"`
User string `json:"user"`
Text string `json:"text,omitempty"`
}
type Room struct {
Messages chan Message
Buddies map[string]Buddy
}
func NewRoom() (r *Room) {
r = new(Room)
r.Messages = make(chan Message)
r.Buddies = make(map[string]Buddy)
go r.Broadcast()
}
func (r *Room) Leave(nick string) {
if _, ok := r.Buddies[nick]; !ok {
return
}
delete(r.Buddies[nick])
if len(r.Buddies) == 0 {
close(r.Messages)
} else {
r.Messages <- Message{
Type: MsgLeave,
From: nick,
}
}
}
func (r *Room) Broadcast() {
for m := range r.Messages {
for _, buddy := range r.Buddies {
buddy.Receive <- m // TODO: What happens when this locks?
}
}
}
var rooms = make(map[string]Room)
func Join(room, nick string) (*Buddy, error) {
r, ok := rooms[room]
if !ok {
r = NewRoom()
rooms[room] = r
}
if _, there := r.Buddies[nick]; there {
return nil, errors.New("Nickname is already in use")
}
if len(r.Buddies) >= *perroom {
return nil, errors.New("Room is full")
}
r.Messages <- Message{
Type: MsgJoin,
From: nick,
}
b := NewBuddy(nick, r)
r.Buddies[nick] = b
return b, nil
}
|
