瀏覽代碼

feat(goctl): Add with session for model tpl (#3272)

anqiansong 1 年之前
父節點
當前提交
3ca40001b4
共有 2 個文件被更改,包括 8 次插入1 次删除
  1. 1 1
      tools/goctl/internal/version/version.go
  2. 7 0
      tools/goctl/model/sql/template/tpl/model-new.tpl

+ 1 - 1
tools/goctl/internal/version/version.go

@@ -6,7 +6,7 @@ import (
 )
 
 // BuildVersion is the version of goctl.
-const BuildVersion = "1.5.2"
+const BuildVersion = "1.5.3"
 
 var tag = map[string]int{"pre-alpha": 0, "alpha": 1, "pre-bata": 2, "beta": 3, "released": 4, "": 5}
 

+ 7 - 0
tools/goctl/model/sql/template/tpl/model-new.tpl

@@ -4,3 +4,10 @@ func new{{.upperStartCamelObject}}Model(conn sqlx.SqlConn{{if .withCache}}, c ca
 		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}},
+	}
+}