package services

import (
	"context"

	"gitee.com/mixerj/pitaya"
	"gitee.com/mixerj/pitaya/component"
	"gitee.com/mixerj/pitaya/examples/demo/worker/protos"
	"github.com/golang/protobuf/proto"
)

// Worker server
type Worker struct {
	component.Base
}

// Configure starts workers and register rpc job
func (w *Worker) Configure() error {
	err := pitaya.StartWorker(pitaya.GetConfig())
	if err != nil {
		return err
	}

	pitaya.RegisterRPCJob(&RPCJob{})
	return nil
}

// RPCJob implements worker.RPCJob
type RPCJob struct{}

// ServerDiscovery returns a serverID="", meaning any server
// is ok
func (r *RPCJob) ServerDiscovery(
	route string,
	rpcMetadata map[string]interface{},
) (serverID string, err error) {
	return "", nil
}

// RPC calls pitaya's rpc
func (r *RPCJob) RPC(
	ctx context.Context,
	serverID, routeStr string,
	reply, arg proto.Message,
) error {
	return pitaya.RPCTo(ctx, serverID, routeStr, reply, arg)
}

// GetArgReply returns reply and arg of LogRemote,
// since we have no other methods in this example
func (r *RPCJob) GetArgReply(
	route string,
) (arg, reply proto.Message, err error) {
	return &protos.Arg{}, &protos.Response{}, nil
}