【原】使用Golang语言编写echo程序

网络教程的echo程序，在server断开链接的时候，会出现卡死的情况，如何防止这样的问题出现呢？

我自己优化了一把，异步的echo程序，在client端进行容错判断，在server重启的时候能够进行重新链接。

啥也不说了，上代码。 

 

//server.go 

package main

import (
    "bufio"
    "fmt"
    "net"
)

func Echo(c net.Conn) {
    defer c.Close()
    for {
        line, err := bufio.NewReader(c).ReadString('\n')
        if err != nil {
            fmt.Printf("Failure to read:%s\n", err.Error())
            return
        }
        _, err = c.Write([]byte(line))
        if err != nil {
            fmt.Printf("Failure to write: %s\n", err.Error())
            return
        }
    }
}

func main() {
    fmt.Printf("Server is ready...\n")
    l, err := net.Listen("tcp", ":8053")
    if err != nil {
        fmt.Printf("Failure to listen: %s\n", err.Error())
    }

    for {
        if c, err := l.Accept(); err == nil {
            go Echo(c) //new thread
        }
    }
}

 //client.go

package main

import (
     " bufio "
     " fmt "
     " net "
     " os "
     " time "
)

type Clienter  struct {
    client  net.Conn
    isAlive  bool
    SendStr chan  string
    RecvStr chan  string
}

func (c *Clienter) Connect()  bool {
     if c.isAlive {
         return  true
    }  else {
         var err error
        c.client, err = net.Dial( " tcp ",  " 127.0.0.1:8053 ")
         if err != nil {
            fmt.Printf( " Failure to connet:%s\n ", err.Error())
             return  false
        }
        c.isAlive =  true
    }
     return  true
}

func (c *Clienter) Echo() {
    line := <-c.SendStr
    c.client.Write([] byte(line))
    buf := make([] byte,  1024)
    n, err := c.client.Read(buf)
     if err != nil {
        c.RecvStr <-  string( " Server close... ")
        c.client.Close()
        c.isAlive =  false
         return
    }
    time.Sleep( 1 * time.Second)
    c.RecvStr <-  string(buf[ 0:n])
}

func Work(tc *Clienter) {
     if !tc.isAlive {
         if tc.Connect() {
            tc.Echo()
        }  else {
            <-tc.SendStr
            tc.RecvStr <-  string( " Server close... ")
        }
    }  else {
        tc.Echo()
    }
}
func main() {
     // defer client.Close()
     var tc Clienter
    tc.SendStr = make(chan  string)
    tc.RecvStr = make(chan  string)
     if !tc.Connect() {
         return
    }
    r := bufio.NewReader(os.Stdin)
     for {
         switch line, ok := r.ReadString( ' \n ');  true {
         case ok != nil:
            fmt.Printf( " bye bye!\n ")
             return
         default:
            go Work(&tc)
            tc.SendStr <- line
            s := <-tc.RecvStr
            fmt.Printf( " back:%s\n ", s)
        }
    }

}