From ad7f1b2c471fedaced779a6ea89aff738eb31381 Mon Sep 17 00:00:00 2001
From: rogeecn <rogeecn@qq.com>
Date: Fri, 28 Feb 2025 10:26:54 +0800
Subject: [PATCH] feat: add db

---
 pkg/ast/model/models.gen.go.tpl | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/pkg/ast/model/models.gen.go.tpl b/pkg/ast/model/models.gen.go.tpl
index b5357fd..2a22f48 100644
--- a/pkg/ast/model/models.gen.go.tpl
+++ b/pkg/ast/model/models.gen.go.tpl
@@ -3,18 +3,26 @@
 // Code generated by the atomctl ; DO NOT EDIT.
 package models
 
+import (
+	"database/sql"
+)
+
+var db *sql.DB
 {{- range . }}
 var {{.PascalTable}} *{{.CamelTable}}Model
 {{- end }}
 
 // @provider(model)
 type models struct {
+	db *sql.DB
+
 {{- range . }}
 	{{.CamelTable}} *{{.CamelTable}}Model
 {{- end }}
 }
 
 func (m *models) Prepare() error {
+	db = m.db
 {{- range . }}
 	{{.PascalTable}} = m.{{.CamelTable}}
 {{- end }}