首先实现基于net/rpc的RPC服务,再通过http接口构建注册中心,服务启动时注册并定期发送心跳,注册中心定时清理超时节点,客户端通过查询中心获取地址并调用远程方法。

在go语言中实现RPC服务注册中心,核心是让服务提供者注册自己,服务调用方能发现并调用远程方法。golang标准库提供了net/rpc包支持RPC通信,但不包含服务注册与发现功能。要实现注册中心,需自行设计或集成第三方组件。以下是基于标准库和常见实践的实现思路。
1. 使用 net/rpc 搭建基础RPC服务
先定义一个简单的服务结构体和可导出方法(首字母大写):
type Arith int <p>func (t <em>Arith) Multiply(args </em>Args, reply <em>int) error { </em>reply = args.A * args.B return nil }</p>
启动RPC服务器:
arith := new(Arith) rpc.Register(arith) listener, _ := net.Listen("tcp", ":1234") for { conn, _ := listener.Accept() go rpc.ServeConn(conn) }
这样就有一个监听在:1234端口的RPC服务。
立即学习“go语言免费学习笔记(深入)”;
2. 设计服务注册中心
注册中心本质是一个中间服务,记录哪些服务运行在哪些地址上。可以使用HTTP+内存存储快速实现:
定义服务信息结构:
type ServiceInfo struct { Name string Addr string Time time.Time // 用于健康检查 } </font>
注册中心维护一个map保存服务列表:
var services = make(map[string]ServiceInfo)
提供HTTP接口供服务注册:
http.HandleFunc("/register", func(w http.ResponseWriter, r *http.Request) { name := r.FormValue("name") addr := r.FormValue("addr") services[name] = ServiceInfo{Name: name, Addr: addr, Time: time.Now()} fmt.Fprintf(w, "Registered: %s at %s", name, addr) }) http.ListenAndServe(":8500", nil)
服务启动后主动向注册中心发送注册请求:
http.Get("http://localhost:8500/register?name=Arith&addr=localhost:1234")
3. 实现服务发现与调用
客户端需要从注册中心获取服务地址,再建立RPC连接:
查询服务地址:
func getServiceAddr(name string) (string, error) { resp, err := http.Get("http://localhost:8500/services") if err != nil { return "", err } defer resp.Body.Close() // 解析jsON响应,查找对应服务的Addr var services map[string]ServiceInfo json.NewDecoder(resp.Body).Decode(&services) if info, ok := services[name]; ok { return info.Addr, nil } return "", errors.New("service not found") }
调用远程方法:
addr, _ := getServiceAddr("Arith") conn, _ := net.Dial("tcp", addr) client := rpc.NewClient(conn) defer client.Close() <p>args := &Args{A: 7, B: 8} var reply int client.Call("Arith.Multiply", args, &reply) fmt.Println("Result:", reply)</p>
4. 增强功能:心跳与自动注销
真实场景中需防止失效节点。服务定期发送心跳:
go func() { for { time.Sleep(10 * time.Second) http.Get("http://localhost:8500/heartbeat?name=Arith") } }()
注册中心定时清理超时服务:
go func() { for { time.Sleep(30 * time.Second) now := time.Now() for name, info := range services { if now.Sub(info.Time) > 60*time.Second { delete(services, name) } } } }()
基本上就这些。通过组合net/rpc、HTTP注册接口和定时任务,就能构建一个简易但可用的服务注册与发现系统。生产环境建议使用consul、etcd等成熟方案,它们提供更完善的健康检查、分布式一致性与安全机制。不过自研有助于理解底层逻辑,适合特定场景或学习用途。


