Go RPC HTTP服务正确实现与常见陷阱解析

// 服务器端（错误示例）
func main() {
    // ... 其他初始化代码 ...
    chat := new(Chat)
    rpc.Register(chat) // 注册RPC服务
    l, e := net.Listen("tcp", *server)
    if e != nil {
        log.Fatal("listen error:", e)
    }
    log.Println(l.Addr().String())
    go rpc.Accept(l) // 尝试接受RPC连接
    http.Serve(l, nil) // 启动HTTP服务器
    // ...
}

// 客户端
func main() {
    // ... 其他初始化代码 ...
    client, err := rpc.DialHTTP("tcp", *server) // 客户端通过HTTP拨号
    if err != nil {
        log.Fatal("dialing: ", err) // 这里会报404错误
    }
    // ...
}

package main

import (
    "flag"
    "log"
    "net"
    "net/http"
    "net/rpc"
)

// Chat 是一个RPC服务类型
type Chat string

// Msg 是Chat服务的一个方法，接收一个字符串消息，返回其长度
func (t *Chat) Msg(msg string, bytes *int) error {
    *bytes = len(msg)
    log.Printf("Received message: '%s', length: %d\n", msg, *bytes)
    return nil
}

func main() {
    serverAddr := flag.String("server", "localhost:0", "Server address and port (e.g., localhost:8080)")
    isClient := flag.Bool("c", false, "Run as client (default is server)")
    flag.Parse()

    if !(*isClient) {
        // 服务器模式
        chat := new(Chat)
        rpc.Register(chat)     // 注册RPC服务
        rpc.HandleHTTP()       // 关键：注册RPC的HTTP处理程序

        l, e := net.Listen("tcp", *serverAddr)
        if e != nil {
            log.Fatalf("listen error: %v", e)
        }
        defer l.Close() // 确保监听器关闭

        actualAddr := l.Addr().String()
        log.Printf("RPC server listening on %s", actualAddr)

        // 启动HTTP服务器，http.Serve会使用http.DefaultServeMux处理请求
        // rpc.HandleHTTP()已经将RPC路由注册到DefaultServeMux
        err := http.Serve(l, nil)
        if err != nil && err != http.ErrServerClosed {
            log.Fatalf("HTTP server error: %v", err)
        }
    } else {
        // 客户端模式 (将在下一节详细介绍)
        log.Println("Running as client, please provide server address.")
    }
}

package main

import (
    "flag"
    "log"
    "net/rpc"
    "time" // 引入time包用于等待服务器启动
)

// Chat 是一个RPC服务类型 (客户端无需完整实现，只需知道方法签名)
type Chat string

// Msg 是Chat服务的一个方法，接收一个字符串消息，返回其长度
// 客户端通常只关注调用，无需实现此方法
// func (t *Chat) Msg(msg string, bytes *int) error { return nil } // 客户端不需要这个实现

func main() {
    serverAddr := flag.String("server", "localhost:8080", "Server address and port (e.g., localhost:8080)")
    isClient := flag.Bool("c", false, "Run as client (default is server)")
    flag.Parse()

    if *isClient {
        // 客户端模式
        log.Printf("Client connecting to %s", *serverAddr)

        // 尝试连接，可以添加重试逻辑以应对服务器启动延迟
        var client *rpc.Client
        var err error
        for i := 0; i < 5; i++ { // 尝试5次
            client, err = rpc.DialHTTP("tcp", *serverAddr)
            if err == nil {
                break
            }
            log.Printf("Dialing failed: %v. Retrying in 1 second...", err)
            time.Sleep(1 * time.Second)
        }

        if err != nil {
            log.Fatalf("Failed to dial RPC server after multiple attempts: %v", err)
        }
        defer client.Close() // 确保客户端连接关闭

        var reply int
        message := "Make it so!"
        err = client.Call("Chat.Msg", message, &reply)
        if err != nil {
            log.Fatalf("Chat.Msg call error: %v", err)
        }
        log.Printf("Msg: returned %d (length of '%s')", reply, message)
    } else {
        // 服务器模式 (将在上一节详细介绍)
        log.Println("Running as server, waiting for client connections.")
    }
}