dubbogo-cli-v2 新增hessian生成器模式

Author: ChangedenCZD

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

@@ -0,0 +1,54 @@
+package cmd
+
+import (
+	"runtime"
+)
+
+import (
+	"github.com/spf13/cobra"
+)
+
+import (
+	"github.com/dubbogo/tools/cmd/dubbogo-cli-v2/generator/sample"
+	"github.com/dubbogo/tools/cmd/dubbogo-cli-v2/generator/sample/hessian"
+)
+
+// hessianCmd represents the hessian-register-generator command
+var hessianCmd = &cobra.Command{
+	Use:   "hessian",
+	Short: "automatic generate hessian pojo register statement",
+	Run:   generateHessianRegistry,
+}
+
+func init() {
+	rootCmd.AddCommand(hessianCmd)
+	hessianCmd.Flags().StringP(hessian.CmdFlagInclude, "i", "./", "file scan directory path, default `./`")
+	hessianCmd.Flags().IntP(hessian.CmdFlagThread, "t", runtime.NumCPU()*2, "worker thread limit, default (cpu core) * 2")
+	hessianCmd.Flags().BoolP(hessian.CmdFlagOnlyError, "e", false, "only print error message, default false")
+}
+
+func generateHessianRegistry(cmd *cobra.Command, _ []string) {
+	var include string
+	var thread int
+	var onlyError bool
+	var err error
+
+	include, err = cmd.Flags().GetString(hessian.CmdFlagInclude)
+	if err != nil {
+		panic(err)
+	}
+
+	thread, err = cmd.Flags().GetInt(hessian.CmdFlagThread)
+	if err != nil {
+		panic(err)
+	}
+
+	onlyError, err = cmd.Flags().GetBool(hessian.CmdFlagOnlyError)
+	if err != nil {
+		panic(err)
+	}
+
+	if err = sample.HessianRegistryGenerate(include, thread, onlyError); err != nil {
+		panic(err)
+	}
+}

cmd/dubbogo-cli-v2/generator/sample/hessian/constant.go

@@ -0,0 +1,46 @@
+/*
+ * 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 hessian
+
+const (
+	CmdFlagInclude   = "include"
+	CmdFlagThread    = "thread"
+	CmdFlagOnlyError = "error"
+)
+
+const (
+	PackageRegexp = `^package\s[a-zA-Z_][0-9a-zA-Z_]*$`
+
+	LineCommentRegexp         = `\/\/`
+	MutLineCommentStartRegexp = `\/\*`
+	MutLineCommentEndRegexp   = `\*\/`
+
+	InitFunctionRegexp = `^func\sinit\(\)\s\{$`
+
+	HessianImportRegexp = `"github.com/apache/dubbo-go-hessian2"`
+
+	HessianPOJORegexp     = `\*[0-9a-zA-Z_]+\)\sJavaClassName\(\)\sstring\s\{$`
+	HessianPOJONameRegexp = `\*[0-9a-zA-Z_]+\)`
+)
+
+const (
+	newLine byte = '\n'
+	funcEnd byte = '}'
+
+	TargetFileSuffix = ".go"
+)

cmd/dubbogo-cli-v2/generator/sample/hessian/file_scanner.go

@@ -0,0 +1,46 @@
+/*
+ * 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 hessian
+
+import (
+	"fmt"
+	"io/fs"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+// ListFiles 获取目标目录下所有go文件
+func ListFiles(dirPath string, suffix string) (fileList []string, err error) {
+	suffix = strings.ToUpper(suffix)
+	fileList = make([]string, 0)
+	_, err = os.Lstat(dirPath)
+	if err != nil {
+		return nil, fmt.Errorf("找不到该目录[%s]", dirPath)
+	}
+	err = filepath.WalkDir(dirPath, func(path string, d fs.DirEntry, err error) error { // 递归获取目录下所有go文件
+		if d == nil || d.IsDir() {
+			return nil
+		}
+		if strings.HasSuffix(strings.ToUpper(d.Name()), suffix) {
+			fileList = append(fileList, path)
+		}
+		return nil
+	})
+	return
+}