20-golang简单并发服务器

最新推荐文章于 2023-02-20 20:13:22 发布

Lich Howger

最新推荐文章于 2023-02-20 20:13:22 发布

阅读量214

点赞数

分类专栏： golang

本文链接：https://blog.csdn.net/qq_33781658/article/details/83821576

版权

golang 专栏收录该内容

68 篇文章 0 订阅

订阅专栏

func main() {
   listener, err := net.Listen("tcp", "127.0.0.1:8080")

   if err != nil {
      fmt.Println("err是", err)
      return
   }

   defer listener.Close()

   for {
      conn, err := listener.Accept()
      if err != nil {
         fmt.Println("err是", err)
         return
      }

      go handleConnection(conn)
   }

}

func handleConnection(conn net.Conn) {
   defer conn.Close()

   addr := conn.RemoteAddr().String()
   fmt.Println(addr, "连接成功")

   bytes := make([]byte, 2048)
   num, err := conn.Read(bytes)
   if err != nil {
      fmt.Println("err是", err)
      return
   }

   conn.Write(bytes[:num])

}

如果简化一下,就是

func main() {
   listener, _ := net.Listen("tcp", "127.0.0.1:8080")

   defer listener.Close()

   for {
      conn, _ := listener.Accept()

      go handleConnection(conn)
   }

}

func handleConnection(conn net.Conn) {
   defer conn.Close()

   addr := conn.RemoteAddr().String()
   fmt.Println(addr, "连接成功")

   bytes := make([]byte, 2048)
   num, _ := conn.Read(bytes)

   conn.Write(bytes[:num])

}

然后我们再增加一个退出

func main() {

   listener, _ := net.Listen("tcp", "127.0.0.1:8080")

   defer listener.Close()

   for {

      conn, _ := listener.Accept()

      go handleConn(conn)
   }

}
func handleConn(conn net.Conn) {
   defer conn.Close()

   addr := conn.RemoteAddr()
   fmt.Println(addr, "连接成功")

   bytes := make([]byte, 2048)
   for {
      num, _ := conn.Read(bytes)

      if string(bytes[:num]) == "exit" {
         fmt.Println("退出")
         return     
      }

      conn.Write(bytes[:num])
   }

}