gofmt

main.go

@@ -2,27 +2,27 @@ package main
 
 import (
 	"fmt"
-//	"time"
+	//	"time"
 	"sync"
 )
 
-func main(){
+func main() {
 	fmt.Println("start up server")
 	ln, err := bind_socket()
-	if err != nil{
+	if err != nil {
 		return
 	}
 	network_manager := new(NetworkManager)
 	network_manager.control_channels = make(map[string]chan string)
 	network_manager.mutex = new(sync.Mutex)
 	//start_the_world()
 	/*
-	go func(){
-		timer := time.NewTimer(time.Second * 10)
-		<-timer.C
-		fmt.Println("quit all channel")
-		network_manager.QuitAllChannel()
-	}()
+		go func(){
+			timer := time.NewTimer(time.Second * 10)
+			<-timer.C
+			fmt.Println("quit all channel")
+			network_manager.QuitAllChannel()
+		}()
 	*/
 	for {
 		conn, err := ln.Accept()

msg_queue.go

@@ -1,47 +1,46 @@
 package main
 
 import (
-	"net"
 	"fmt"
 	"io"
+	"net"
 )
 
-
 const BUFF_LEN = 4096
-func HandleConnection(c net.Conn, manager *NetworkManager){
+
+func HandleConnection(c net.Conn, manager *NetworkManager) {
 	defer c.Close()
 	fmt.Println("Connection from: ", c.RemoteAddr())
 	quit := manager.RegisterControlChannel(c)
 	buff := make([]byte, BUFF_LEN)
-	loop:
-	for{
-		select{
-			case <-quit:
-				break loop
-			default:
-				count, err := c.Read(buff)
-				if err != nil{
-					if err == io.EOF{
-						fmt.Println("client disconnect")
-						return
-					}
-					fmt.Println(err)
+loop:
+	for {
+		select {
+		case <-quit:
+			break loop
+		default:
+			count, err := c.Read(buff)
+			if err != nil {
+				if err == io.EOF {
+					fmt.Println("client disconnect")
 					return
 				}
-				if count != 0{
-					fmt.Print(string(buff))
-					c.Write(buff)
-				}
+				fmt.Println(err)
+				return
+			}
+			if count != 0 {
+				fmt.Print(string(buff))
+				c.Write(buff)
+			}
 		}
 	}
 	manager.UnregisterControlChannel(c)
 }
 
-func bind_socket() (net.Listener, error){
+func bind_socket() (net.Listener, error) {
 	ln, err := net.Listen("tcp", ":19840")
 	if err != nil {
 		fmt.Println("bind failed")
 	}
 	return ln, err
 }
-

network_manager.go

@@ -6,13 +6,13 @@ import (
 	"sync"
 )
 
-type NetworkManager struct{
-	online_count int //do i need mutex to protect online count?
+type NetworkManager struct {
+	online_count     int //do i need mutex to protect online count?
 	control_channels map[string]chan string
-	mutex *sync.Mutex
+	mutex            *sync.Mutex
 }
 
-func (this *NetworkManager) RegisterControlChannel(c net.Conn) chan string{
+func (this *NetworkManager) RegisterControlChannel(c net.Conn) chan string {
 	defer this.mutex.Unlock()
 	this.mutex.Lock()
 	this.online_count++
@@ -21,18 +21,17 @@ func (this *NetworkManager) RegisterControlChannel(c net.Conn) chan string{
 	return this.control_channels[key]
 }
 
-func (this *NetworkManager) UnregisterControlChannel(c net.Conn){
+func (this *NetworkManager) UnregisterControlChannel(c net.Conn) {
 	this.online_count--
 	key := c.RemoteAddr().String()
 	delete(this.control_channels, key)
 }
 
-func (this *NetworkManager) QuitAllChannel(){
+func (this *NetworkManager) QuitAllChannel() {
 	for k, v := range this.control_channels {
-		go func(ch chan string){
+		go func(ch chan string) {
 			fmt.Println(k)
 			ch <- "quit" //block here
 		}(v)
 	}
 }
-