Go语言 RPC通信示例

最新推荐文章于 2023-04-16 23:36:39 发布

mengni123321

最新推荐文章于 2023-04-16 23:36:39 发布

阅读量642

点赞数 1

分类专栏： Go语言文章标签： Go语言

本文链接：https://blog.csdn.net/mengni123321/article/details/79120320

版权

Go语言专栏收录该内容

1 篇文章 0 订阅

订阅专栏

服务端程序

package main

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

type Args struct {
    A, B int
}

type Quotient struct {
    Quo, Rem int
}

type Arith int

func (t *Arith) Multiply(args *Args, reply *int) error {
    *reply = args.A * args.B

    fmt.Println("Multiply", args.A, args.B)
    return nil
}

func (t *Arith) Divide(args *Args, quo *Quotient) error {
    fmt.Println("Divide", Args.A, Args.B)

    if args.B == 0 {
        return errors.New("divide by zero")
    }
    quo.Quo = args.A / args.B
    quo.Rem = args.A % args.B

    fmt.Println("Divide", quo.Quo, quo.Rem)

    return nil
}

func main() {
    arith := new(Arith)
    rpc.Register(arith)
    rpc.HandleHTTP()
    l, e := net.Listen("tcp", ":1234")
    if e != nil {
        log.Fatal("listen error:", e)
    }
    http.Serve(l, nil)
}

客户端程序

package main

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

type Args struct {
    A, B int
}

type Quotient struct {
    Quo, Rem int
}

func main() {

    if len(os.Args) != 3 {
        fmt.Println("please input two number!")
        os.Exit(1)
    }
    strnum1 := os.Args[1]
    strnum2 := os.Args[2]

    num1, error := strconv.Atoi(strnum1)
    if error != nil {
        fmt.Println("error number!")
        os.Exit(1)
    }

    num2, error2 := strconv.Atoi(strnum2)
    if error2 != nil {
        fmt.Println("error number!")
        os.Exit(1)
    }

    client, err := rpc.DialHTTP("tcp", "127.0.0.1:1234")
    if err != nil {
        log.Fatal("dialing:", err)
    }

    // Synchronous call
    args := &Args{num1, num2}

    fmt.Println("Multiply", args.A, args.B)

    var reply int
    err = client.Call("Arith.Multiply", args, &reply)
    if err != nil {
        log.Fatal("arith error:", err)
    }
    fmt.Printf("Arith: %d*%d=%d\n", args.A, args.B, reply)

    // Asynchronous call
    quotient := new(Quotient)

    fmt.Println("Divide", args.A, args.B)

    divCall := client.Go("Arith.Divide", args, quotient, nil)
    replyCall := <-divCall.Done // will be equal to divCall
    if replyCall.Error != nil {
        log.Fatal("arith error:", replyCall.Error)
    }
    fmt.Printf("Arith: %d/%d=%d...%d", args.A, args.B, quotient.Quo, quotient.Rem)
    // check errors, print, etc.
}