go实现RPC httprpc、tcprpc、jsonrpc等

一、HTTP RPC

package main

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

type Args struct {
	A, B int
}
type Math int

func (m *Math) Multiply(args *Args, reply *int) error{
	*reply = args.A * args.B
	return nil
}

//
type Quotient struct {
	Quo, Rem int    //商、余数
}

func (m *Math) Divide(args *Args, quo *Quotient) error{
	if args.B == 0 {
		return errors.New("divide by zero")
	}
	quo.Quo = args.A / args.B
	quo.Rem = args.A % args.B
	return nil
}
func main()  {
	math := new(Math)
	rpc.Register(math)  //注册
	rpc.HandleHTTP()

	err := http.ListenAndServe(":1234", nil)
	if err != nil {
		fmt.Println(err.Error())
	}
}

package  main

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

//需要注意的是 两边均需要定义的类型存在
type Args struct {
	A, B int
}
type Quotient struct {
	Quo, Rem int
}

//请求调用rpc
func main()  {
	if len(os.Args) != 2 {  //获取命令行参数 参数个数若不为2
		fmt.Println("Usage: ", os.Args[0], "server")
		os.Exit(1)   //退出
	}
	serverAddr := os.Args[1]
	//走的是tcp  http也是基于tcp的
	client, err := rpc.DialHTTP("tcp", serverAddr+":1234")
	if err != nil {
		log.Fatal("dialing:", err)
	}

	args := Args{17, 8}   //入参
	var reply int  // 返回值
	err = client.Call("Math.Multiply", args,&reply)

	if err != nil {
		log.Fatal("Math error:", err)
	}
	fmt.Printf("Math: %d*%d=%d\n",args.A, args.B, reply)

	var quo Quotient
	err = client.Call("Math.Divide", args, &quo)
	if err != nil {
		log.Fatal("Math error:", err)
	}
	fmt.Printf("Math: %d/%d=%d remainder %d\n", args.A, args.B, quo.Quo, quo.Rem)
}

先启动servergo, 然后再启动client.go（传递参数localhost）发现如下输出：

下面将http rpc改为tcp rpc 实际上属于降级操作，因为http是基于tcp的

二、TCP RPC

type Args struct {
	A, B int
}
type Math int

//提供乘法运算
func (m *Math) Multiply(args *Args, reply *int) error{
	*reply = args.A * args.B
	return nil
}

//
type Quotient struct {
	Quo, Rem int    //商、余数
}

//提供除法运算
func (m *Math) Divide(args *Args, quo *Quotient) error{
	if args.B == 0 {
		return errors.New("divide by zero")
	}
	quo.Quo = args.A / args.B
	quo.Rem = args.A % args.B
	return nil
}
func main()  {
	math := new(Math)
	rpc.Register(math)  //注册

	tcpAddr, err := net.ResolveTCPAddr("tcp",":1234") //tcp网络 tcp地址

	if err != nil {
		fmt.Println("Fatal error:", err)
		os.Exit(2)
	}

	listener, err := net.ListenTCP("tcp", tcpAddr)  //在指定地址 监听tcp网络请求
	if err != nil {
		fmt.Println("Fatal error:", err)
		os.Exit(2)
	}
	for {
		// 发现一个请求 就会产生一个连接对象
		conn, err := listener.Accept()
		if err != nil {
			fmt.Println("conn error")
			continue
		}
		rpc.ServeConn(conn)  //只需要把上面建立连接返回的对象 传进去即可 使用了
		//这就是使用Go 标准库 rpc的好处了
	}
}

下面client.go只改一行：

先运行go run server.go
然后运行 go run client.go localhost，如下图：

三、JSON RPC的实现
server.go文件中只改了下面两处即可：

然后client.go文件也只是改了两行代码：