mirror of
https://github.com/unknwon/the-way-to-go_ZH_CN.git
synced 2025-11-13 17:36:12 +08:00
fix: coding style and file format for chapter 11, 12, 13, 14 and 15.
This commit is contained in:
@@ -1,75 +1,75 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"net"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Map of the clients: contains: clientname - 1 (active) / 0 - (inactive)
|
||||
var mapUsers map[string]int
|
||||
|
||||
func main() {
|
||||
var listener net.Listener
|
||||
var error error
|
||||
var conn net.Conn
|
||||
mapUsers = make(map[string]int)
|
||||
|
||||
fmt.Println("Starting the server ...")
|
||||
|
||||
// create listener:
|
||||
listener, error = net.Listen("tcp", "localhost:50000")
|
||||
checkError(error)
|
||||
// listen and accept connections from clients:
|
||||
for {
|
||||
conn, error = listener.Accept()
|
||||
checkError(error)
|
||||
go doServerStuff(conn)
|
||||
}
|
||||
}
|
||||
|
||||
func doServerStuff(conn net.Conn) {
|
||||
var buf []byte
|
||||
var error error
|
||||
|
||||
for {
|
||||
buf = make([]byte, 512)
|
||||
_, error = conn.Read(buf)
|
||||
checkError(error)
|
||||
input := string(buf)
|
||||
if strings.Contains(input, ": SH") {
|
||||
fmt.Println("Server shutting down.")
|
||||
os.Exit(0)
|
||||
}
|
||||
// op commando WHO: write out mapUsers
|
||||
if strings.Contains(input, ": WHO") {
|
||||
DisplayList()
|
||||
}
|
||||
// extract clientname:
|
||||
ix := strings.Index(input, "says")
|
||||
clName := input[0:ix-1]
|
||||
//fmt.Printf("The clientname is ---%s---\n", string(clName))
|
||||
// set clientname active in mapUsers:
|
||||
mapUsers[string(clName)] = 1
|
||||
fmt.Printf("Received data: --%v--", string(buf))
|
||||
}
|
||||
}
|
||||
|
||||
// advantage: code is cleaner,
|
||||
// disadvantage: the server process has to stop at any error:
|
||||
// a simple return continues in the function where we came from!
|
||||
func checkError(error error) {
|
||||
if error != nil {
|
||||
panic("Error: " + error.Error()) // terminate program
|
||||
}
|
||||
}
|
||||
|
||||
func DisplayList() {
|
||||
fmt.Println("--------------------------------------------")
|
||||
fmt.Println("This is the client list: 1=active, 0=inactive")
|
||||
for key, value := range mapUsers {
|
||||
fmt.Printf("User %s is %d\n", key, value)
|
||||
}
|
||||
fmt.Println("--------------------------------------------")
|
||||
}
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net"
|
||||
"os"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Map of the clients: contains: clientname - 1 (active) / 0 - (inactive)
|
||||
var mapUsers map[string]int
|
||||
|
||||
func main() {
|
||||
var listener net.Listener
|
||||
var error error
|
||||
var conn net.Conn
|
||||
mapUsers = make(map[string]int)
|
||||
|
||||
fmt.Println("Starting the server ...")
|
||||
|
||||
// create listener:
|
||||
listener, error = net.Listen("tcp", "localhost:50000")
|
||||
checkError(error)
|
||||
// listen and accept connections from clients:
|
||||
for {
|
||||
conn, error = listener.Accept()
|
||||
checkError(error)
|
||||
go doServerStuff(conn)
|
||||
}
|
||||
}
|
||||
|
||||
func doServerStuff(conn net.Conn) {
|
||||
var buf []byte
|
||||
var error error
|
||||
|
||||
for {
|
||||
buf = make([]byte, 512)
|
||||
_, error = conn.Read(buf)
|
||||
checkError(error)
|
||||
input := string(buf)
|
||||
if strings.Contains(input, ": SH") {
|
||||
fmt.Println("Server shutting down.")
|
||||
os.Exit(0)
|
||||
}
|
||||
// op commando WHO: write out mapUsers
|
||||
if strings.Contains(input, ": WHO") {
|
||||
DisplayList()
|
||||
}
|
||||
// extract clientname:
|
||||
ix := strings.Index(input, "says")
|
||||
clName := input[0 : ix-1]
|
||||
//fmt.Printf("The clientname is ---%s---\n", string(clName))
|
||||
// set clientname active in mapUsers:
|
||||
mapUsers[string(clName)] = 1
|
||||
fmt.Printf("Received data: --%v--", string(buf))
|
||||
}
|
||||
}
|
||||
|
||||
// advantage: code is cleaner,
|
||||
// disadvantage: the server process has to stop at any error:
|
||||
// a simple return continues in the function where we came from!
|
||||
func checkError(error error) {
|
||||
if error != nil {
|
||||
panic("Error: " + error.Error()) // terminate program
|
||||
}
|
||||
}
|
||||
|
||||
func DisplayList() {
|
||||
fmt.Println("--------------------------------------------")
|
||||
fmt.Println("This is the client list: 1=active, 0=inactive")
|
||||
for key, value := range mapUsers {
|
||||
fmt.Printf("User %s is %d\n", key, value)
|
||||
}
|
||||
fmt.Println("--------------------------------------------")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user