make_seeder.go 692 B

12345678910111213141516171819202122232425
  1. package make
  2. import (
  3. "github.com/runningwater/gohub/pkg/console"
  4. "github.com/spf13/cobra"
  5. )
  6. var CmdMakeSeeder = &cobra.Command{
  7. Use: "seeder",
  8. Short: "Create model's seeder file, example: make seeder user",
  9. Run: runMakeSeeder,
  10. Args: cobra.ExactArgs(1),
  11. }
  12. func runMakeSeeder(cmd *cobra.Command, args []string) {
  13. model := makeModelFromString(args[0])
  14. filePath := "database/seeders/" + model.PackageName + "_seeder.go"
  15. createFileFromTpl(filePath, "seeder", model)
  16. console.Success("Seeder file created successfully: " + filePath)
  17. // 提示用户需要在 database/seeders/init.go 中注册 Seeder
  18. console.Warning("Please register the seeder in database/seeders/init.go")
  19. }