blob: 833913145af19653b21b4c332bc3788df9d44bbb [file] [log] [blame]
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package polaris
import (
"bytes"
"context"
"fmt"
"strconv"
"strings"
)
import (
"dubbo.apache.org/dubbo-go/v3/common"
"dubbo.apache.org/dubbo-go/v3/common/constant"
"dubbo.apache.org/dubbo-go/v3/registry"
)
var (
protocolForDubboGO string = "dubbo"
)
type PolarisHeartbeat struct {
cancel context.CancelFunc
instance registry.ServiceInstance
url *common.URL
}
func getInstanceKey(namespace string, instance registry.ServiceInstance) string {
return fmt.Sprintf("%s-%s-%s-%d", namespace, instance.GetServiceName(), instance.GetHost(), instance.GetPort())
}
// just copy from dubbo-go for nacos
func getCategory(url *common.URL) string {
role, _ := strconv.Atoi(url.GetParam(constant.RegistryRoleKey, strconv.Itoa(constant.PolarisDefaultRoleType)))
category := common.DubboNodes[role]
return category
}
// just copy from dubbo-go for nacos
func getServiceName(url *common.URL) string {
var buffer bytes.Buffer
buffer.Write([]byte(getCategory(url)))
appendParam(&buffer, url, constant.InterfaceKey)
return buffer.String()
}
// just copy from dubbo-go for nacos
func appendParam(target *bytes.Buffer, url *common.URL, key string) {
value := url.GetParam(key, "")
target.Write([]byte(constant.PolarisServiceNameSeparator))
if strings.TrimSpace(value) != "" {
target.Write([]byte(value))
}
}