go tcp server

weiwenwang · 2018-04-23 21:43:16 · 1853 次点击
更多评论

问题还是有点意思的,可以考虑用 chan 来实现这个,下面是我写的代码,可以参考着写个更完善的。

package main

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

func main() {
    l, _ := net.Listen("tcp", ":8000")
    for {
        conn, _ := l.Accept()
        ch := make(chan string)
        go talk(conn, ch)
        go input(ch)
    }
}

func talk(conn io.ReadWriteCloser, ch chan string) {
    for c := range ch {
        fmt.Fprintf(conn, c)
    }
    fmt.Fprintf(conn, "closed, byebye.")
    conn.Close()
}

func input(ch chan string) {
    reader := bufio.NewReader(os.Stdin)
    text, _ := reader.ReadString('\n')
    for text != "#\n" {
        ch <- text
        text, _ = reader.ReadString('\n')
    }
    close(ch)
}
#2

为什么要存数据库或者redis?

用户的连接是实时的,一旦断开就失效了,下次连接就是新的,你就存在内存的一个map里就可以。

存数据库或者redis的话,只可能存 conn 的内存地址,或者更底层的 socket fd, 但不管是哪个,在用户断开连接之后都会失效。

使用失效的内存地址作为指针获取conn会导致野指针访问,使用失效的 socket fd 如果这个fd号没被重用还好,被重用了你就把消息发给错误的人了

#3