feat: gen model export WithSession

zeromicro · Oct 18, 2023 · 1f58fce · 1f58fce
1 parent 423597a
commit 1f58fce
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/tools/goctl/model/sql/template/tpl/model-new.tpl b/tools/goctl/model/sql/template/tpl/model-new.tpl
@@ -11,3 +11,10 @@ func (m *default{{.upperStartCamelObject}}Model) withSession(session sqlx.Sessio
 		table:      {{.table}},
 	}
 }
+
+func (m *default{{.upperStartCamelObject}}Model) WithSession(session sqlx.Session) *default{{.upperStartCamelObject}}Model {
+	return &default{{.upperStartCamelObject}}Model{
+		{{if .withCache}}CachedConn:m.CachedConn.WithSession(session){{else}}conn:sqlx.NewSqlConnFromSession(session){{end}},
+		table:      {{.table}},
+	}
+}
diff --git a/tools/goctl/model/sql/template/tpl/types.tpl b/tools/goctl/model/sql/template/tpl/types.tpl
@@ -1,6 +1,7 @@
 type (
 	{{.lowerStartCamelObject}}Model interface{
 		{{.method}}
+        WithSession(session sqlx.Session) *default{{.upperStartCamelObject}}Model
 	}
 
 	default{{.upperStartCamelObject}}Model struct {