GoZeroPlus/zrpc/resolver/internal/kube/targetparser.go

package kube

import (
	"strconv"
	"strings"

	"github.com/wuntsong-org/go-zero-plus/zrpc/resolver/internal/targets"
	"google.golang.org/grpc/resolver"
)

const (
	colon            = ":"
	defaultNamespace = "default"
)

var emptyService Service

// Service represents a service with namespace, name and port.
type Service struct {
	Namespace string
	Name      string
	Port      int
}

// ParseTarget parses the resolver.Target.
func ParseTarget(target resolver.Target) (Service, error) {
	var service Service
	service.Namespace = targets.GetAuthority(target)
	if len(service.Namespace) == 0 {
		service.Namespace = defaultNamespace
	}

	endpoints := targets.GetEndpoints(target)
	if strings.Contains(endpoints, colon) {
		segs := strings.SplitN(endpoints, colon, 2)
		service.Name = segs[0]
		port, err := strconv.Atoi(segs[1])
		if err != nil {
			return emptyService, err
		}

		service.Port = port
	} else {
		service.Name = endpoints
	}

	return service, nil
}