rpc: draft of a RPC service

rpc/client.go

@@ -0,0 +1,79 @@
+package rpc
+
+import (
+	"encoding/gob"
+	"image/color"
+	"net/rpc"
+
+	"github.com/mcuadros/go-rpi-rgb-led-matrix"
+)
+
+func init() {
+	gob.Register(color.RGBA{})
+}
+
+// RGBLedMatrix matrix representation for ws281x
+type Client struct {
+	network string
+	addr    string
+	client  *rpc.Client
+	leds    []color.Color
+}
+
+// NewRGBLedMatrix returns a new matrix using the given size and config
+func NewClient(network, addr string) (rgbmatrix.Matrix, error) {
+	client, err := rpc.DialHTTP(network, addr)
+	if err != nil {
+		return nil, err
+	}
+
+	return &Client{
+		network: network,
+		addr:    addr,
+		client:  client,
+		leds:    make([]color.Color, 2048),
+	}, nil
+}
+
+// Geometry returns the width and the height of the matrix
+func (c *Client) Geometry() (width, height int) {
+	var reply *GeometryReply
+	err := c.client.Call("RPCMatrix.Geometry", &GeometryArgs{}, &reply)
+	if err != nil {
+		panic(err)
+	}
+
+	return reply.Width, reply.Height
+}
+
+func (c *Client) Apply(leds []color.Color) error {
+	defer func() { c.leds = make([]color.Color, 2048) }()
+
+	var reply *ApplyReply
+	return c.client.Call("RPCMatrix.Apply", &ApplyArgs{Colors: leds}, &reply)
+}
+
+// Render update the display with the data from the LED buffer
+func (c *Client) Render() error {
+	return c.Apply(c.leds)
+}
+
+// At return an Color which allows access to the LED display data as
+// if it were a sequence of 24-bit RGB values.
+func (c *Client) At(position int) color.Color {
+	if c.leds[position] == nil {
+		return color.Black
+	}
+
+	return c.leds[position]
+}
+
+// Set set LED at position x,y to the provided 24-bit color value.
+func (m *Client) Set(position int, c color.Color) {
+	m.leds[position] = color.RGBAModel.Convert(c)
+}
+
+// Close finalizes the ws281x interface
+func (c *Client) Close() error {
+	return c.Apply(make([]color.Color, 2048))
+}

rpc/server.go

@@ -0,0 +1,54 @@
+package rpc
+
+import (
+	"fmt"
+	"image/color"
+	"log"
+	"net"
+	"net/http"
+	"net/rpc"
+
+	"github.com/mcuadros/go-rpi-rgb-led-matrix"
+)
+
+type RPCMatrix struct {
+	m rgbmatrix.Matrix
+}
+
+type GeometryArgs struct{}
+type GeometryReply struct{ Width, Height int }
+
+func (m *RPCMatrix) Geometry(_ *GeometryArgs, reply *GeometryReply) error {
+	w, h := m.m.Geometry()
+	reply.Width = w
+	reply.Height = h
+
+	return nil
+}
+
+type ApplyArgs struct{ Colors []color.Color }
+type ApplyReply struct{}
+
+func (m *RPCMatrix) Apply(args *ApplyArgs, reply *ApplyReply) error {
+	return m.m.Apply(args.Colors)
+}
+
+type CloseArgs struct{}
+type CloseReply struct{}
+
+func (m *RPCMatrix) Close(_ *CloseArgs, _ *CloseReply) error {
+	return m.m.Close()
+}
+
+func Serve(m rgbmatrix.Matrix) {
+	rpc.Register(&RPCMatrix{m})
+
+	rpc.HandleHTTP()
+	l, e := net.Listen("tcp", ":1234")
+	if e != nil {
+		log.Fatal("listen error:", e)
+	}
+
+	fmt.Println(l)
+	http.Serve(l, nil)
+}