dispatch

package main

import (
	"context"
	"errors"
	"log"

	"github.com/searKing/golang/go/x/dispatch"
)

type DispatchMsg struct {
	data int
}

func main() {
	var conn chan DispatchMsg
	dispatch.NewDispatch(
		dispatch.ReaderFunc(func(ctx context.Context) (interface{}, error) {
			return ReadMessage(conn)
		}),
		dispatch.HandlerFunc(func(ctx context.Context, msg interface{}) error {
			m := msg.(*DispatchMsg)
			return HandleMessage(m)
		})).Start()
}

func ReadMessage(c <-chan DispatchMsg) (interface{}, error) {
	var msg DispatchMsg
	var ok bool

	if msg, ok = <-c; ok {
		log.Println("Recv From Channel Failed")
		return nil, errors.New("recv from channel failed")
	}
	log.Printf("Recv From Channel Success: %v\n", msg.data)
	return &msg, nil
}

// just print what's received
func HandleMessage(msg *DispatchMsg) error {
	if msg == nil {
		log.Println("Handle From Channel Failed")
		return errors.New("handle from channel failed")
	}
	log.Printf("Handle From Channel Success: %v\n", msg.data)
	return nil
}