聊聊Golang实现长连接的方案

package main

import (
    "fmt"
    "net"
)

func main() {
    conn, err := net.Dial("tcp", "localhost:8000")
    if err != nil {
        fmt.Println(err)
        return
    }
    defer conn.Close()

    for {
        // 从服务端读取数据
        buf := make([]byte, 1024)
        n, err := conn.Read(buf)
        if err != nil {
            fmt.Println(err)
            return
        }

        fmt.Println(string(buf[:n]))
    }
}

package main

import (
    "fmt"
    "log"
    "net/Http"

    "GitHub.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{}

func websocketHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }

    defer conn.Close()

    for {
        // 从客户端读取数据
        _, message, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            return
        }

        // 向客户端发送数据
        err = conn.WriteMessage(websocket.TextMessage, message)
        if err != nil {
            log.Println(err)
            return
        }
    }
}

func main() {
    http.HandleFunc("/websocket", websocketHandler)
    err := http.ListenAndServe(":8000", nil)
    if err != nil {
        log.Fatal(err)
    }
}

package main

import (
    "context"
    "fmt"
    "log"
    "net"

    "google.golang.org/grpc"
)

type HelloServer struct{}

func (h *HelloServer) SayHello(ctx context.Context, req *HelloRequest) (*HelloResponse, error) {
    return &HelloResponse{Message: fmt.Sprintf("Hello %s!", req.Name)}, nil
}

func main() {
    lis, err := net.Listen("tcp", ":8000")
    if err != nil {
        log.Fatal(err)
    }

    grpcServer := grpc.NewServer()

    ReGISterHelloServiceServer(grpcServer, &HelloServer{})

    err = grpcServer.Serve(lis)
    if err != nil {
        log.Fatal(err)
    }
}