-
Notifications
You must be signed in to change notification settings - Fork 3.6k
/
main.go
115 lines (102 loc) · 1.87 KB
/
main.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
104
105
106
107
108
109
110
111
112
113
114
115
package main
import (
"bufio"
"fmt"
"io"
"log"
"net"
)
type User struct {
Name string
Output chan Message
}
type Message struct {
Username string
Text string
}
type ChatServer struct {
Users map[string]User
Join chan User
Leave chan User
Input chan Message
}
func (cs *ChatServer) Run() {
for {
select {
case user := <-cs.Join:
cs.Users[user.Name] = user
go func() {
cs.Input <- Message{
Username: "SYSTEM",
Text: fmt.Sprintf("%s joined", user.Name),
}
}()
case user := <-cs.Leave:
delete(cs.Users, user.Name)
go func() {
cs.Input <- Message{
Username: "SYSTEM",
Text: fmt.Sprintf("%s left", user.Name),
}
}()
case msg := <-cs.Input:
for _, user := range cs.Users {
select {
case user.Output <- msg:
default:
}
}
}
}
}
func handleConn(chatServer *ChatServer, conn net.Conn) {
defer conn.Close()
io.WriteString(conn, "Enter your Username:")
scanner := bufio.NewScanner(conn)
scanner.Scan()
user := User{
Name: scanner.Text(),
Output: make(chan Message, 10),
}
chatServer.Join <- user
defer func() {
chatServer.Leave <- user
}()
// Read from conn
go func() {
for scanner.Scan() {
ln := scanner.Text()
chatServer.Input <- Message{user.Name, ln}
}
}()
// write to conn
for msg := range user.Output {
if msg.Username != user.Name {
_, err := io.WriteString(conn, msg.Username+": "+msg.Text+"\n")
if err != nil {
break
}
}
}
}
func main() {
server, err := net.Listen("tcp", ":9000")
if err != nil {
log.Fatalln(err.Error())
}
defer server.Close()
chatServer := &ChatServer{
Users: make(map[string]User),
Join: make(chan User),
Leave: make(chan User),
Input: make(chan Message),
}
go chatServer.Run()
for {
conn, err := server.Accept()
if err != nil {
log.Fatalln(err.Error())
}
go handleConn(chatServer, conn)
}
}