feat: add job creation

cmd/static/cmd.go

@@ -24,6 +24,7 @@ import (
 	"github.com/cloudwego/cwgo/pkg/consts"
 	"github.com/cloudwego/cwgo/pkg/curd/doc"
 	"github.com/cloudwego/cwgo/pkg/fallback"
+	"github.com/cloudwego/cwgo/pkg/job"
 	"github.com/cloudwego/cwgo/pkg/model"
 	"github.com/cloudwego/cwgo/pkg/server"
 	"github.com/urfave/cli/v2"
@@ -95,6 +96,17 @@ func Init() *cli.App {
 				return doc.Doc(globalArgs.DocArgument)
 			},
 		},
+		{
+			Name:  JobName,
+			Usage: JobUsage,
+			Flags: jobFlags(),
+			Action: func(c *cli.Context) error {
+				if err := globalArgs.JobArgument.ParseCli(c); err != nil {
+					return err
+				}
+				return job.Job(globalArgs.JobArgument)
+			},
+		},
 		{
 			Name:  ApiListName,
 			Usage: ApiUsage,
@@ -197,7 +209,12 @@ Examples:
 Examples:
   cwgo api --project_path ./
 `
+	JobName  = "job"
+	JobUsage = `generate job code
 
+Examples:
+	cwgo job --job_name jobOne --job_name jobTwo --module my_job
+`
 	FallbackName  = "fallback"
 	FallbackUsage = "fallback to hz or kitex"
 

cmd/static/job_flags.go

@@ -0,0 +1,14 @@
+package static
+
+import (
+	"github.com/cloudwego/cwgo/pkg/consts"
+	"github.com/urfave/cli/v2"
+)
+
+func jobFlags() []cli.Flag {
+	return []cli.Flag{
+		&cli.StringSliceFlag{Name: consts.JobName, Usage: "Specify the job name."},
+		&cli.StringFlag{Name: consts.Module, Aliases: []string{"mod"}, Usage: "Specify the Go module name to generate go.mod."},
+		&cli.StringFlag{Name: consts.OutDir, Usage: "Specify output directory, default is current dir."},
+	}
+}

config/argument.go

@@ -38,6 +38,7 @@ type Argument struct {
 	*ClientArgument
 	*ModelArgument
 	*DocArgument
+	*JobArgument
 	*ApiArgument
 	*FallbackArgument
 }
@@ -48,6 +49,7 @@ func NewArgument() *Argument {
 		ClientArgument:   NewClientArgument(),
 		ModelArgument:    NewModelArgument(),
 		DocArgument:      NewDocArgument(),
+		JobArgument:      NewJobArgument(),
 		ApiArgument:      NewApiArgument(),
 		FallbackArgument: NewFallbackArgument(),
 	}

config/job.go

@@ -0,0 +1,40 @@
+/*
+ * Copyright 2024 CloudWeGo Authors
+ *
+ * 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 config
+
+import (
+	"github.com/cloudwego/cwgo/pkg/consts"
+	"github.com/urfave/cli/v2"
+)
+
+type JobArgument struct {
+	GoMod         string
+	PackagePrefix string
+	JobName       []string
+	OutDir        string
+}
+
+func NewJobArgument() *JobArgument {
+	return &JobArgument{}
+}
+
+func (j *JobArgument) ParseCli(ctx *cli.Context) error {
+	j.JobName = ctx.StringSlice(consts.JobName)
+	j.GoMod = ctx.String(consts.Module)
+	j.OutDir = ctx.String(consts.OutDir)
+	return nil
+}

pkg/consts/const.go

@@ -165,6 +165,10 @@ const (
 	MongoDb = "mongodb"
 )
 
+const (
+	JobName = "job_name"
+)
+
 const (
 	BashAutocomplete = `#! /bin/bash