net/rpc
在网上找grpc学习资料的时候,发先go语言自带一个rpc框架,不过只支持go语言,赶紧过来看看。
net/rpc
这里简要记了,详细的内容在go源码的注释里面。
注意事项:
- Go的RPC只支持go写的系统
- Go RPC的函数有特殊要求
- 首字母必须大写
- 必须有两个参数,一个入参,一个是返回的参数,第二个参数必须是指针类型的,函数还要有一个error。
服务端
package main
import (
"fmt"
"net"
"net/http"
"net/rpc"
)
type Server struct {
}
//请求
type Req struct {
NumOne int
NumTwo int
}
//返回
type Res struct {
Num int
}
func (s *Server) Add(req Req, res *Res) error {
res.Num = req.NumOne + req.NumTwo
return nil
}
func main() {
rpc.Register(new(Server))
rpc.HandleHTTP()
l, err := net.Listen("tcp", ":8888")
if err != nil {
fmt.Println(err.Error())
}
http.Serve(l, nil)
}
我们将server注册起来,然后跑在8888端口上。server有一个函数Add,就是将入参的两个值加到一起赋给出参。
客户端
package main
import (
"fmt"
"net/rpc"
)
type Req struct {
NumOne int
NumTwo int
}
type Res struct {
Num int
}
func main() {
req := Req{NumOne: 1, NumTwo: 2}
var res Res
client, err := rpc.DialHTTP("tcp", "localhost:8888")
if err != nil {
fmt.Println(err.Error())
}
client.Call("Server.Add", req, &res)
fmt.Println(res)
}
客户端调用服务端的这个函数,结果如下:
使用Call()方法是同步方式的调用,使用Go方法可以是实现异步调用
修改服务端为:
func (s *Server) Add(req Req, res *Res) error {
time.Sleep(5 * time.Second)
res.Num = req.NumOne + req.NumTwo
return nil
}
我们让add函数等5秒再返回
客户端修改为这样:
func main() {
req := Req{NumOne: 1, NumTwo: 2}
var res Res
client, err := rpc.DialHTTP("tcp", "localhost:8888")
if err != nil {
fmt.Println(err.Error())
}
ca := client.Go("Server.Add",req,&res,nil)
for {
select {
case <-ca.Done:
fmt.Println(res)
return
default:
time.Sleep(time.Second)
fmt.Println("哈哈啊哈")
}
}
}
然后结果是