通过 RPC 传递匿名函数的 GobEncoder 实现

package main

import (
    "fmt"
    "log"
    "net"
    "net/rpc"
)

// 定义可执行的函数类型
type FuncType func(int) int

// 定义函数映射
var functionMap = map[string]FuncType{
    "square": func(x int) int { return x * x },
    "double": func(x int) int { return x * 2 },
}

// RPC 服务
type Arith struct{}

// 定义 RPC 方法
func (a *Arith) ExecuteFunction(args struct {
    FuncName string
    Value    int
}, reply *int) error {
    // 根据函数名查找函数
    fn, ok := functionMap[args.FuncName]
    if !ok {
        return fmt.Errorf("function %s not found", args.FuncName)
    }

    // 执行函数并返回结果
    *reply = fn(args.Value)
    return nil
}

func main() {
    // 注册 RPC 服务
    arith := new(Arith)
    rpc.Register(arith)
    rpc.HandleHTTP()

    // 监听端口
    l, e := net.Listen("tcp", ":1234")
    if e != nil {
        log.Fatal("listen error:", e)
    }

    // 启动服务
    go func() {
        err := http.Serve(l, nil) // 修改为 http.Serve
        if err != nil {
            log.Fatal("serve error:", err)
        }
    }()

    // 客户端调用示例 (需要在另一个程序中运行)
    client, err := rpc.DialHTTP("tcp", "localhost:1234")
    if err != nil {
        log.Fatal("dialing:", err)
    }

    args := struct {
        FuncName string
        Value    int
    }{
        FuncName: "square",
        Value:    5,
    }
    var reply int
    err = client.Call("Arith.ExecuteFunction", args, &reply)
    if err != nil {
        log.Fatal("arith error:", err)
    }

    fmt.Printf("Result: %d\n", reply)
}