srpc/client.go

package srpc

import (
	"bytes"
	"fmt"
	"io"
	"net"
)

var REQUEST_HEADER_SIZE = int32(8)

type requestHeader struct {
	Size int64
}

type request struct {
	FuncName string
	Payload  []byte
}

type Client struct {
	ed   IEncoderDecoder
	conn net.Conn
}

func (client *Client) Call(funcName string, args ...interface{}) (ret []byte, ok bool) {
	var b bytes.Buffer
	ok = true

	enc := client.ed.NewEncoder(&b)
	for _, a := range args {
		enc.Encode(a)
	}

	payload := b.Bytes()
	req := request{funcName, payload}
	fmt.Println(req)

	enc = client.ed.NewEncoder(client.conn)
	enc.Encode(req)

	var header responseHeader
	dec := client.ed.NewDecoder(client.conn)
	dec.Decode(&header)
	fmt.Println(header.Size)
	fmt.Println(header.Status)

	respData := make([]byte, header.Size)
	dec.Decode(&respData)

	if header.Status != OK {
		ok = false
	}

	return respData, ok
}

func (client *Client) NewDecoder(r io.Reader) IDecoder {
	return client.ed.NewDecoder(r)
}

func NewClient(conn net.Conn) *Client {
	return &Client{NewEncoderDecoder(), conn}
}