同时使用HTTP和RPC

时间:2019-01-24 11:17:20

标签: http go rpc

我正在尝试在系统中实现rpchttp服务器。 因此,要同时监听服务器,我需要运行2 goroutines 这是代码

func main() {
    // Recovering all errors during the process
    defer errorHandler()

    wg.Add(2)

    go RPCConnect()

    fmt.Println("Listening for RPC   127.0.0.1:" + config.rpcPort)

    go HTTPConnect()

    fmt.Println("Listening for HTTP  127.0.0.1:" + config.httpPort)

    wg.Wait()
}


func RPCConnect() {
    err := rpc.Register(pool)

    if err != nil {
        panic(err)
    }

    rpc.HandleHTTP()

    listener, e := net.Listen("tcp", ":"+config.rpcPort)
    if e != nil {
        panic(e)
    }
    err = http.Serve(listener, nil)
    if err != nil {
        panic(err)
    }
}

func HTTPConnect() {

    var httpPool HTTPPool

    r := mux.NewRouter()
    r.HandleFunc("/create", httpPool.Create).Methods("POST")
    r.HandleFunc("/generate", httpPool.Generate).Methods("POST")
    r.HandleFunc("/list", httpPool.List).Methods("GET")
    r.HandleFunc("/delete", httpPool.Delete).Methods("POST")

    err := http.ListenAndServe("localhost:"+config.httpPort, r)

    if err != nil {
        panic(err)
    }

}

我不知道这是不是最好的方法。有人可以告诉我更简单灵活的方法吗?

如果问题与我无关,我先向您道歉

1 个答案:

答案 0 :(得分:1)

建议的方法很好。

如果您只想使用一个端口而不是两个,可以使用多种第三方工具来实现:

1)https://github.com/soheilhy/cmux

2)https://github.com/grpc-ecosystem/grpc-gateway