go語言net包rpc遠程調用的使用

一、基於http的RPC

服務端：

package main;

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

//go對RPC的支持，支持三個級別：TCP、HTTP、JSONRPC
//go的RPC只支持GO開發的服務器與客戶端之間的交互，因為采用了gob編碼

//注意字段必須是導出
type Params struct {
	Width, Height int;
}

type Rect struct{}

//函數必須是導出的
//必須有兩個導出類型參數
//第一個參數是接收參數
//第二個參數是返回給客戶端參數，必須是指針類型
//函數還要有一個返回值error
func (r *Rect) Area(p Params, ret *int) error {
	*ret = p.Width * p.Height;
	return nil;
}

func (r *Rect) Perimeter(p Params, ret *int) error {
	*ret = (p.Width + p.Height) * 2;
	return nil;
}

func main() {
	rect := new(Rect);
	//注冊一個rect服務
	rpc.Register(rect);
	//把服務處理綁定到http協議上
	rpc.HandleHTTP();
	err := http.ListenAndServe(":8080", nil);
	if err != nil {
		log.Fatal(err);
	}
}

客戶端：

package main;

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

type Params struct {
	Width, Height int;
}

func main() {
	//連接遠程rpc服務
	rpc, err := rpc.DialHTTP("tcp", "127.0.0.1:8080");
	if err != nil {
		log.Fatal(err);
	}
	ret := 0;
	//調用遠程方法
	//注意第三個參數是指針類型
	err2 := rpc.Call("Rect.Area", Params{50, 100}, &ret);
	if err2 != nil {
		log.Fatal(err2);
	}
	fmt.Println(ret);
	err3 := rpc.Call("Rect.Perimeter", Params{50, 100}, &ret);
	if err3 != nil {
		log.Fatal(err3);
	}
	fmt.Println(ret);
}

二、基於tcp的RPC

服務端：

package main;

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

//注意字段必須是導出
type Params struct {
	Width, Height int;
}

type Rect struct{}

func (r *Rect) Area(p Params, ret *int) error {
	*ret = p.Width * p.Height;
	return nil;
}

func (r *Rect) Perimeter(p Params, ret *int) error {
	*ret = (p.Width + p.Height) * 2;
	return nil;
}

func chkError(err error) {
	if err != nil {
		log.Fatal(err);
	}
}

func main() {
	rect := new(Rect);
	//注冊rpc服務
	rpc.Register(rect);
	//獲取tcpaddr
	tcpaddr, err := net.ResolveTCPAddr("tcp4", "127.0.0.1:8080");
	chkError(err);
	//監聽端口
	tcplisten, err2 := net.ListenTCP("tcp", tcpaddr);
	chkError(err2);
	//死循環處理連接請求
	for {
		conn, err3 := tcplisten.Accept();
		if err3 != nil {
			continue;
		}
		//使用goroutine單獨處理rpc連接請求
		go rpc.ServeConn(conn);
	}
}

客戶端：

package main;

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

type Params struct {
	Width, Height int;
}

func main() {
	//連接遠程rpc服務
	//這里使用Dial，http方式使用DialHTTP，其他代碼都一樣
	rpc, err := rpc.Dial("tcp", "127.0.0.1:8080");
	if err != nil {
		log.Fatal(err);
	}
	ret := 0;
	//調用遠程方法
	//注意第三個參數是指針類型
	err2 := rpc.Call("Rect.Area", Params{50, 100}, &ret);
	if err2 != nil {
		log.Fatal(err2);
	}
	fmt.Println(ret);
	err3 := rpc.Call("Rect.Perimeter", Params{50, 100}, &ret);
	if err3 != nil {
		log.Fatal(err3);
	}
	fmt.Println(ret);
}

三、JSON RPC 方式

jsonrpc方式是數據編碼采用了json，而不是gob編碼。

服務端：

package main;

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

//注意字段必須是導出
type Params struct {
	Width, Height int;
}

type Rect struct{}

func (r *Rect) Area(p Params, ret *int) error {
	*ret = p.Width * p.Height;
	return nil;
}

func (r *Rect) Perimeter(p Params, ret *int) error {
	*ret = (p.Width + p.Height) * 2;
	return nil;
}

func chkError(err error) {
	if err != nil {
		log.Fatal(err);
	}
}

func main() {
	rect := new(Rect);
	//注冊rpc服務
	rpc.Register(rect);
	//獲取tcpaddr
	tcpaddr, err := net.ResolveTCPAddr("tcp4", "127.0.0.1:8080");
	chkError(err);
	//監聽端口
	tcplisten, err2 := net.ListenTCP("tcp", tcpaddr);
	chkError(err2);
	for {
		conn, err3 := tcplisten.Accept();
		if err3 != nil {
			continue;
		}
		//使用goroutine單獨處理rpc連接請求
		//這里使用jsonrpc進行處理
		go jsonrpc.ServeConn(conn);
	}
}

客戶端：

package main;

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

type Params struct {
	Width, Height int;
}

func main() {
	//連接遠程rpc服務
	//這里使用jsonrpc.Dial
	rpc, err := jsonrpc.Dial("tcp", "127.0.0.1:8080");
	if err != nil {
		log.Fatal(err);
	}
	ret := 0;
	//調用遠程方法
	//注意第三個參數是指針類型
	err2 := rpc.Call("Rect.Area", Params{50, 100}, &ret);
	if err2 != nil {
		log.Fatal(err2);
	}
	fmt.Println(ret);
	err3 := rpc.Call("Rect.Perimeter", Params{50, 100}, &ret);
	if err3 != nil {
		log.Fatal(err3);
	}
	fmt.Println(ret);
}