handler.go

package netchan

import (
	"bytes"
	"encoding/gob"
	"reflect"
)

type Handler struct {
	sr SendRecver
}

func NewHandler(sr SendRecver) *Handler {
	return &Handler{sr: sr}
}

func (s *Handler) HandleRecv(name string, ch chan<- interface{}, t reflect.Type) error {
	for {
		body := s.sr.Recv(name)
		v := reflect.New(t)
		dec := gob.NewDecoder(bytes.NewReader(body))
		err := dec.DecodeValue(v)
		if err != nil {
			return err
		}

		ch <- v.Elem().Interface()
	}
}

func (s *Handler) HandleSend(network, address, name string, ch <-chan interface{}) error {
	for v := range ch {
		var buf bytes.Buffer
		enc := gob.NewEncoder(&buf)
		err := enc.Encode(v)
		if err != nil {
			return err
		}

		err = s.sr.Send(network, address, name, buf.Bytes())
		if err != nil {
			return err
		}
	}

	return nil
}