go-micro's istio selector, for service registry.

istio.go

package istio

import (
	"fmt"
	"strings"

	"github.com/micro/go-micro/registry"
	"github.com/micro/go-micro/selector"
)

const (
	svcPort = 10088
)

var (
	serviceHostMapping = map[string]string{
		"payment": "payment.common",
	}
)

type istio struct{}

func (r *istio) Init(opts ...selector.Option) error {
	return nil
}

func (r *istio) Options() selector.Options {
	return selector.Options{}
}

func (r *istio) Select(service string, opts ...selector.SelectOption) (selector.Next, error) {
	if host, exist := serviceHostMapping[service]; exist {
		node := &registry.Node{
			Id:      service,
			Address: host,
			Port:    svcPort,
		}
		return func() (*registry.Node, error) {
			return node, nil
		}, nil
	}
	return nil, fmt.Errorf("service %s(%s) not found", service, svc)
}

func (r *istio) Mark(service string, node *registry.Node, err error) {
	return
}

func (r *istio) Reset(service string) {
	return
}

func (r *istio) Close() error {
	return nil
}

func (r *istio) String() string {
	return "istio"
}

func NewSelector(opts ...selector.Option) selector.Selector {
	return &istio{}
}