tmrgo/.svn/pristine/07/0734fea0fe4fdfb4a256c83cccd65684807a256a.svn-base

package endpoint

import (
	"context"
	"github.com/go-kit/kit/endpoint"
	"github.com/hashicorp/consul/api"
	"github.com/longjoy/micro-go-course/section14/register/service"
)


type RegisterEndpoints struct {
	DiscoveryEndpoint   endpoint.Endpoint
	HealthCheckEndpoint endpoint.Endpoint
}

// 服务发现请求结构体
type DiscoveryRequest struct {
	ServiceName string
}
// 服务发现响应结构体
type DiscoveryResponse struct {
	Instances []*api.AgentService `json:"instances"`
	Error string `json:"error"`
}
// 创建服务发现的 Endpoint
func MakeDiscoveryEndpoint(svc service.Service) endpoint.Endpoint {
	return func(ctx context.Context, request interface{}) (response interface{}, err error) {

		req := request.(DiscoveryRequest)
		instances, err := svc.DiscoveryService(ctx, req.ServiceName)
		var errString = ""

		if err != nil{
			errString = err.Error()
		}
		return &DiscoveryResponse{
			Instances:instances,
			Error:errString,
		}, nil
	}
}

// HealthRequest 健康检查请求结构
type HealthRequest struct{}
// HealthResponse 健康检查响应结构
type HealthResponse struct {
	Status string `json:"status"`
}

// MakeHealthCheckEndpoint 创建健康检查Endpoint
func MakeHealthCheckEndpoint(svc service.Service) endpoint.Endpoint {
	return func(ctx context.Context, request interface{}) (response interface{}, err error) {
		status := svc.HealthCheck()
		return HealthResponse{
			Status:status,
		}, nil
	}
}