You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
1.2 KiB
67 lines
1.2 KiB
2 years ago
|
package selector
|
||
|
|
||
|
import (
|
||
|
"strconv"
|
||
|
|
||
|
"github.com/go-kratos/kratos/v2/registry"
|
||
|
)
|
||
|
|
||
|
// DefaultNode is selector node
|
||
|
type DefaultNode struct {
|
||
|
scheme string
|
||
|
addr string
|
||
|
weight *int64
|
||
|
version string
|
||
|
name string
|
||
|
metadata map[string]string
|
||
|
}
|
||
|
|
||
|
// Scheme is node scheme
|
||
|
func (n *DefaultNode) Scheme() string {
|
||
|
return n.scheme
|
||
|
}
|
||
|
|
||
|
// Address is node address
|
||
|
func (n *DefaultNode) Address() string {
|
||
|
return n.addr
|
||
|
}
|
||
|
|
||
|
// ServiceName is node serviceName
|
||
|
func (n *DefaultNode) ServiceName() string {
|
||
|
return n.name
|
||
|
}
|
||
|
|
||
|
// InitialWeight is node initialWeight
|
||
|
func (n *DefaultNode) InitialWeight() *int64 {
|
||
|
return n.weight
|
||
|
}
|
||
|
|
||
|
// Version is node version
|
||
|
func (n *DefaultNode) Version() string {
|
||
|
return n.version
|
||
|
}
|
||
|
|
||
|
// Metadata is node metadata
|
||
|
func (n *DefaultNode) Metadata() map[string]string {
|
||
|
return n.metadata
|
||
|
}
|
||
|
|
||
|
// NewNode new node
|
||
|
func NewNode(scheme, addr string, ins *registry.ServiceInstance) Node {
|
||
|
n := &DefaultNode{
|
||
|
scheme: scheme,
|
||
|
addr: addr,
|
||
|
}
|
||
|
if ins != nil {
|
||
|
n.name = ins.Name
|
||
|
n.version = ins.Version
|
||
|
n.metadata = ins.Metadata
|
||
|
if str, ok := ins.Metadata["weight"]; ok {
|
||
|
if weight, err := strconv.ParseInt(str, 10, 64); err == nil {
|
||
|
n.weight = &weight
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
return n
|
||
|
}
|