Support nacos discovery (#108)

pkg/common/protocol.go

@@ -0,0 +1,80 @@
+// Copyright (c) 2022 Alibaba Group Holding Ltd.
+//
+// Licensed 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 common
+
+import "strings"
+
+type Protocol string
+
+const (
+	TCP         Protocol = "TCP"
+	HTTP        Protocol = "HTTP"
+	GRPC        Protocol = "GRPC"
+	Dubbo       Protocol = "Dubbo"
+	Unsupported Protocol = "UnsupportedProtocol"
+)
+
+func ParseProtocol(s string) Protocol {
+	switch strings.ToLower(s) {
+	case "tcp":
+		return TCP
+	case "http":
+		return HTTP
+	case "grpc":
+		return GRPC
+	case "dubbo":
+		return Dubbo
+	}
+	return Unsupported
+}
+
+func (p Protocol) IsTCP() bool {
+	switch p {
+	case TCP:
+		return true
+	default:
+		return false
+	}
+}
+
+func (p Protocol) IsHTTP() bool {
+	switch p {
+	case HTTP, GRPC:
+		return true
+	default:
+		return false
+	}
+}
+
+func (p Protocol) IsGRPC() bool {
+	switch p {
+	case GRPC:
+		return true
+	default:
+		return false
+	}
+}
+
+func (p Protocol) IsDubbo() bool {
+	return p == Dubbo
+}
+
+func (p Protocol) IsUnsupported() bool {
+	return p == Unsupported
+}
+
+func (p Protocol) String() string {
+	return string(p)
+}