diff --git a/cmd/generate/cmd.go b/cmd/generate/cmd.go index f942022..394b484 100644 --- a/cmd/generate/cmd.go +++ b/cmd/generate/cmd.go @@ -1,9 +1,10 @@ package generate import ( + "log" + "github.com/spf13/afero" "github.com/spf13/cobra" - "log" ) var MakeCommand = &cobra.Command{ diff --git a/cmd/generate/crud.go b/cmd/generate/crud.go index 02fe1e1..1507cc2 100644 --- a/cmd/generate/crud.go +++ b/cmd/generate/crud.go @@ -2,13 +2,14 @@ package generate import ( "fmt" - pluralize "github.com/gertd/go-pluralize" - "github.com/spf13/afero" - "github.com/spf13/cobra" "log" "path" "runtime" "strings" + + pluralize "github.com/gertd/go-pluralize" + "github.com/spf13/afero" + "github.com/spf13/cobra" ) var PackageName string diff --git a/cmd/generate/stubs/model.stub b/cmd/generate/stubs/model.stub index 424911a..95df8fb 100644 --- a/cmd/generate/stubs/model.stub +++ b/cmd/generate/stubs/model.stub @@ -12,7 +12,9 @@ type {{TitleName}} struct { Id primitive.ObjectID `json:"id,omitempty" bson:"_id"` Task string `json:"task" bson:"task"` Status string `json:"status" bson:"status"` + CreatedBy string `json:"created_by,omitempty" bson:"created_by"` CreatedAt time.Time `json:"created_at,omitempty" bson:"created_at"` + UpdatedBy string `json:"updated_by,omitempty" bson:"updated_by"` UpdatedAt time.Time `json:"updated_at,omitempty" bson:"updated_at"` } diff --git a/cmd/main.go b/cmd/main.go index aa5804e..4e921a2 100644 --- a/cmd/main.go +++ b/cmd/main.go @@ -1,9 +1,10 @@ package cmd import ( + "log" + "github.com/shipu/artifact/cmd/generate" "github.com/spf13/cobra" - "log" ) var RootCmd = &cobra.Command{