feat: get interfaces and methods form remote

Author: cachen

cmd/dubbogo-cli-v2/README.md

@@ -0,0 +1,35 @@
+# dubbogo-cli-v2
+> dubbo-go integration tool
+
+## How to use
+1. 安装
+```bash
+go get -u github.com/dubbogo/tools/cmd/dubbogo-cli-v2
+```
+## The main function
+
+### Get a list of interfaces and methods
+```bash
+./dubbogo-cli-v2 show --r zookeeper --h 127.0.0.1:2181
+```
+The output is as follows
+
+```bash
+interface: org.apache.dubbo.game.basketballService
+methods: []
+interface: com.apache.dubbo.sample.basic.IGreeter
+methods: []
+interface: com.dubbogo.pixiu.UserService
+methods: [CreateUser,GetUserByCode,GetUserByName,GetUserByNameAndAge,GetUserTimeout,UpdateUser,UpdateUserByName]
+interface: org.apache.dubbo.gate.basketballService
+methods: []
+interface: org.apache.dubbo.game.basketballService
+methods: []
+interface: com.apache.dubbo.sample.basic.IGreeter
+methods: []
+interface: com.dubbogo.pixiu.UserService
+methods: [CreateUser,GetUserByCode,GetUserByName,GetUserByNameAndAge,GetUserTimeout,UpdateUser,UpdateUserByName]
+interface: org.apache.dubbo.gate.basketballService
+methods: []
+
+```

cmd/dubbogo-cli-v2/README_CN.md

@@ -0,0 +1,35 @@
+# dubbogo-cli-v2
+> dubbo-go 集成工具
+
+## 使用方式
+1. 安装
+```bash
+go get -u github.com/dubbogo/tools/cmd/dubbogo-cli-v2
+```
+## 主要功能
+
+### 获取接口及方法列表
+```bash
+./dubbogo-cli-v2 show --r zookeeper --h 127.0.0.1:2181
+```
+输出如下
+
+```bash
+interface: org.apache.dubbo.game.basketballService
+methods: []
+interface: com.apache.dubbo.sample.basic.IGreeter
+methods: []
+interface: com.dubbogo.pixiu.UserService
+methods: [CreateUser,GetUserByCode,GetUserByName,GetUserByNameAndAge,GetUserTimeout,UpdateUser,UpdateUserByName]
+interface: org.apache.dubbo.gate.basketballService
+methods: []
+interface: org.apache.dubbo.game.basketballService
+methods: []
+interface: com.apache.dubbo.sample.basic.IGreeter
+methods: []
+interface: com.dubbogo.pixiu.UserService
+methods: [CreateUser,GetUserByCode,GetUserByName,GetUserByNameAndAge,GetUserTimeout,UpdateUser,UpdateUserByName]
+interface: org.apache.dubbo.gate.basketballService
+methods: []
+
+```

cmd/dubbogo-cli-v2/cmd/root.go

@@ -0,0 +1,74 @@
+/*
+ * 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 cmd
+
+import (
+	"fmt"
+	"os"
+)
+
+import (
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+var cfgFile string
+
+// rootCmd represents the base command when called without any subcommands
+var rootCmd = &cobra.Command{
+	Use:   "dubbogo-cli-v2",
+	Short: "",
+	Long:  ``,
+	Run: func(cmd *cobra.Command, args []string) {
+		fmt.Println("hello")
+	},
+}
+
+// Execute adds all child commands to the root command and sets flags appropriately.
+// This is called by main.main(). It only needs to happen once to the rootCmd.
+func Execute() {
+	cobra.CheckErr(rootCmd.Execute())
+}
+
+func init() {
+	cobra.OnInitialize(initConfig)
+	rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.dubbogo-cli-v2.yaml)")
+	rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
+}
+
+// initConfig reads in config file and ENV variables if set.
+func initConfig() {
+	if cfgFile != "" {
+		// Use config file from the flag.
+		viper.SetConfigFile(cfgFile)
+	} else {
+		home, err := os.UserHomeDir()
+		cobra.CheckErr(err)
+
+		viper.AddConfigPath(home)
+		viper.SetConfigType("yaml")
+		viper.SetConfigName(".dubbogo-cli-v2")
+	}
+
+	viper.AutomaticEnv() // read in environment variables that match
+
+	// If a config file is found, read it in.
+	if err := viper.ReadInConfig(); err == nil {
+		fmt.Fprintln(os.Stderr, "Using config file:", viper.ConfigFileUsed())
+	}
+}

cmd/dubbogo-cli-v2/cmd/show.go

@@ -0,0 +1,68 @@
+/*
+ * 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 cmd
+
+import (
+	"fmt"
+	"github.com/dubbogo/tools/cmd/dubbogo-cli-v2/metadata"
+	"log"
+
+	"github.com/spf13/cobra"
+)
+
+import (
+	_ "github.com/dubbogo/tools/cmd/dubbogo-cli-v2/metadata/zookeeper"
+)
+
+// showCmd represents the show command
+var showCmd = &cobra.Command{
+	Use:   "show",
+	Short: "show interfaces and methods",
+	Long:  ``,
+	Run:   show,
+}
+
+func init() {
+	rootCmd.AddCommand(showCmd)
+	showCmd.Flags().String("r", "r", "")
+	showCmd.Flags().String("h", "h", "")
+}
+
+func show(cmd *cobra.Command, args []string) {
+	registry, err := cmd.Flags().GetString("r")
+	if err != nil {
+		panic(err)
+	}
+	host, err := cmd.Flags().GetString("h")
+	if err != nil {
+		panic(err)
+	}
+	fact, ok := metadata.GetFactory(registry)
+	if !ok {
+		log.Print("registry not support")
+		return
+	}
+	methodsMap, err := fact("dubbogo-cli", []string{host}).ShowChildren()
+	if err != nil {
+		panic(err)
+	}
+	for k, v := range methodsMap {
+		fmt.Printf("interface: %s\n", k)
+		fmt.Printf("methods: %v\n", v)
+	}
+}

cmd/dubbogo-cli-v2/main.go

@@ -0,0 +1,24 @@
+/*
+ * 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 main
+
+import "github.com/dubbogo/tools/cmd/dubbogo-cli-v2/cmd"
+
+func main() {
+	cmd.Execute()
+}

cmd/dubbogo-cli-v2/metadata/metadata.go

@@ -0,0 +1,42 @@
+/*
+ * 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 metadata
+
+import (
+	"log"
+)
+
+type MetaData interface {
+	ShowChildren() (map[string][]string, error)
+}
+
+type Factory func(name string, zkAddrs []string) MetaData
+
+var factMap = make(map[string]Factory)
+
+func Register(name string, fact Factory) {
+	if _, ok := factMap[name]; ok {
+		log.Printf("duplicate name: %s", name)
+	}
+	factMap[name] = fact
+}
+
+func GetFactory(name string) (fact Factory, ok bool) {
+	fact, ok = factMap[name]
+	return
+}

cmd/dubbogo-cli-v2/metadata/zookeeper/zookeeper.go

@@ -0,0 +1,107 @@
+/*
+ * 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 zookeeper
+
+import (
+	"fmt"
+	"log"
+	"time"
+)
+
+import (
+	"dubbo.apache.org/dubbo-go/v3/common"
+	gxzookeeper "github.com/dubbogo/gost/database/kv/zk"
+)
+
+import (
+	"github.com/dubbogo/tools/cmd/dubbogo-cli-v2/metadata"
+)
+
+func init() {
+	metadata.Register("zookeeper", NewZk)
+}
+
+func NewZk(name string, zkAddrs []string) metadata.MetaData {
+	return NewZookeeperMetadataReport(name, zkAddrs)
+}
+
+// ZookeeperMetadataReport is the implementation of
+// MetadataReport based on zookeeper.
+type ZookeeperMetadataReport struct {
+	client  *gxzookeeper.ZookeeperClient
+	rootDir string
+}
+
+func NewZookeeperMetadataReport(name string, zkAddrs []string) *ZookeeperMetadataReport {
+	client, err := gxzookeeper.NewZookeeperClient(
+		name,
+		zkAddrs,
+		false,
+		gxzookeeper.WithZkTimeOut(15*time.Second))
+	if err != nil {
+		panic(err)
+	}
+	return &ZookeeperMetadataReport{
+		client:  client,
+		rootDir: "/dubbo",
+	}
+}
+
+func (z *ZookeeperMetadataReport) GetChildren(path string) ([]string, error) {
+	delimiter := "/"
+	if path == "" {
+		delimiter = path
+	}
+	return z.client.GetChildren(z.rootDir + delimiter + path)
+}
+
+func (z *ZookeeperMetadataReport) ShowChildren() (map[string][]string, error) {
+	methodsMap := map[string][]string{}
+	inters, err := z.GetChildren("")
+	if err != nil {
+		return nil, err
+	}
+	for _, inter := range inters {
+		if _, ok := methodsMap[inter]; !ok {
+			methodsMap[inter] = make([]string, 0)
+		}
+
+		interChildren, err := z.GetChildren(inter)
+		if err != nil {
+			return nil, err
+		}
+		for _, interChild := range interChildren {
+			interURLs, err := z.GetChildren(inter + "/" + interChild)
+			if err != nil {
+				log.Println(err)
+			}
+			for _, interURL := range interURLs {
+				url, err := common.NewURL(interURL)
+				if err != nil {
+					return nil, err
+				}
+				methodsMap[inter] = append(methodsMap[inter], url.GetParam("methods", ""))
+			}
+		}
+	}
+	for k, v := range methodsMap {
+		fmt.Printf("interface: %s\n", k)
+		fmt.Printf("methods: %v\n", v)
+	}
+	return methodsMap, nil
+}