fix: template

This commit is contained in:
Rogee
2024-12-21 00:25:12 +08:00
parent 58450dd63d
commit 6f04023a42
7 changed files with 26 additions and 20 deletions

View File

@@ -17,6 +17,7 @@ import (
func CommandGenEnum(root *cobra.Command) {
cmd := &cobra.Command{
Use: "enum",
Aliases: []string{"e"},
Short: "Generate enums",
RunE: commandGenEnumE,
}

View File

@@ -20,6 +20,7 @@ import (
func CommandGenModel(root *cobra.Command) {
cmd := &cobra.Command{
Use: "model",
Aliases: []string{"m"},
Short: "Generate jet models",
RunE: commandGenModelE,
}

View File

@@ -29,6 +29,7 @@ func getTypePkgName(typ string) string {
func CommandGenProvider(root *cobra.Command) {
cmd := &cobra.Command{
Use: "provider",
Aliases: []string{"p"},
Short: "Generate providers",
Long: `
// @provider

View File

@@ -15,6 +15,7 @@ import (
func CommandMigrate(root *cobra.Command) {
cmd := &cobra.Command{
Use: "migrate [up|up-by-one|up-to|create|down|down-to|fix|redo|reset|status|version]",
Aliases: []string{"m"},
RunE: commandMigrate,
}
cmd.Flags().StringP("config", "c", "config.toml", "database config file")

View File

@@ -19,6 +19,7 @@ import (
func CommandNewModule(root *cobra.Command) {
cmd := &cobra.Command{
Use: "module",
Aliases: []string{"m"},
Short: "new module",
Args: cobra.ExactArgs(1),
RunE: commandNewModuleE,

View File

@@ -25,6 +25,7 @@ func isValidGoPackageName(name string) bool {
func CommandNewProject(root *cobra.Command) {
cmd := &cobra.Command{
Use: "project",
Aliases: []string{"p"},
Short: "new project",
RunE: commandNewProjectE,
}

View File

@@ -5,13 +5,13 @@ import (
"database/sql"
"fmt"
"{{.ModuleName}}/pkg/consts"
"github.com/go-jet/jet/v2/qrm"
)
const CtxDB = "__db__tx:"
func FromContext(ctx context.Context, db *sql.DB) qrm.DB {
if tx, ok := ctx.Value(consts.CtxKeyTx).(*sql.Tx); ok {
if tx, ok := ctx.Value(CtxDB).(*sql.Tx); ok {
return tx
}
return db