template.go 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
  1. package template
  2. import (
  3. _ "embed"
  4. "fmt"
  5. "github.com/zeromicro/go-zero/tools/goctl/util"
  6. )
  7. // Vars defines a template for var block in model
  8. //
  9. //go:embed tpl/var.tpl
  10. var Vars string
  11. // Types defines a template for types in model.
  12. //
  13. //go:embed tpl/types.tpl
  14. var Types string
  15. // Tag defines a tag template text
  16. //
  17. //go:embed tpl/tag.tpl
  18. var Tag string
  19. // TableName defines a template that generate the tableName method.
  20. //
  21. //go:embed tpl/table-name.tpl
  22. var TableName string
  23. // New defines the template for creating model instance.
  24. //
  25. //go:embed tpl/model-new.tpl
  26. var New string
  27. // ModelCustom defines a template for extension
  28. //
  29. //go:embed tpl/model.tpl
  30. var ModelCustom string
  31. // ModelGen defines a template for model
  32. var ModelGen = fmt.Sprintf(`%s
  33. package {{.pkg}}
  34. {{.imports}}
  35. {{.vars}}
  36. {{.types}}
  37. {{.new}}
  38. {{.delete}}
  39. {{.find}}
  40. {{.insert}}
  41. {{.update}}
  42. {{.extraMethod}}
  43. {{.tableName}}
  44. `, util.DoNotEditHead)
  45. // Insert defines a template for insert code in model
  46. //
  47. //go:embed tpl/insert.tpl
  48. var Insert string
  49. // InsertMethod defines an interface method template for insert code in model
  50. //
  51. //go:embed tpl/interface-insert.tpl
  52. var InsertMethod string
  53. // Update defines a template for generating update codes
  54. //
  55. //go:embed tpl/update.tpl
  56. var Update string
  57. // UpdateMethod defines an interface method template for generating update codes
  58. //
  59. //go:embed tpl/interface-update.tpl
  60. var UpdateMethod string
  61. // Imports defines a import template for model in cache case
  62. //
  63. //go:embed tpl/import.tpl
  64. var Imports string
  65. // ImportsNoCache defines a import template for model in normal case
  66. //
  67. //go:embed tpl/import-no-cache.tpl
  68. var ImportsNoCache string
  69. // FindOne defines find row by id.
  70. //
  71. //go:embed tpl/find-one.tpl
  72. var FindOne string
  73. // FindOneByField defines find row by field.
  74. //
  75. //go:embed tpl/find-one-by-field.tpl
  76. var FindOneByField string
  77. // FindOneByFieldExtraMethod defines find row by field with extras.
  78. //
  79. //go:embed tpl/find-one-by-field-extra-method.tpl
  80. var FindOneByFieldExtraMethod string
  81. // FindOneMethod defines find row method.
  82. //
  83. //go:embed tpl/interface-find-one.tpl
  84. var FindOneMethod string
  85. // FindOneByFieldMethod defines find row by field method.
  86. //
  87. //go:embed tpl/interface-find-one-by-field.tpl
  88. var FindOneByFieldMethod string
  89. // Field defines a filed template for types
  90. //
  91. //go:embed tpl/field.tpl
  92. var Field string
  93. // Error defines an error template
  94. //
  95. //go:embed tpl/err.tpl
  96. var Error string
  97. // Delete defines a delete template
  98. //
  99. //go:embed tpl/delete.tpl
  100. var Delete string
  101. // DeleteMethod defines a delete template for interface method
  102. //
  103. //go:embed tpl/interface-delete.tpl
  104. var DeleteMethod string