tmrgo1/.svn/pristine/02/02ab502de54198625802f1a33950c11aaa681e93.svn-base

package service

import (
	"context"
	"errors"
	"github.com/hashicorp/consul/api"
	"github.com/longjoy/micro-go-course/section14/register/discovery"
	"log"
)

type Service interface {

	HealthCheck() string

	DiscoveryService(ctx context.Context, serviceName string) ([]*api.AgentService, error)

}


var ErrNotServiceInstances = errors.New("instances are not existed")

type RegisterServiceImpl struct {
	discoveryClient *discovery.DiscoveryClient
}

func NewRegisterServiceImpl(discoveryClient *discovery.DiscoveryClient) Service  {
	return &RegisterServiceImpl{
		discoveryClient:discoveryClient,
	}
}


func (service *RegisterServiceImpl) DiscoveryService(ctx context.Context, serviceName string) ([]*api.AgentService, error)  {

	instances, err := service.discoveryClient.DiscoverServices(ctx, serviceName)

	if err != nil {
		log.Printf("get service info err: %s", err)
	}
	if instances == nil || len(instances) == 0 {
		return nil, ErrNotServiceInstances
	}
	return instances, nil
}



func (*RegisterServiceImpl) HealthCheck() string {
	return "OK"
}