databse provider

templates/project/database/-transform.yaml.raw

@@ -1,5 +1,11 @@
 ignores:
 - migrations
+- river_client
+- river_client_queue
+- river_job
+- river_leader
+- river_migration
+- river_queue
 imports:
 - go.ipao.vip/gen
 field_type:

templates/project/database/database.go.tpl

@@ -5,6 +5,14 @@ import (
 	"database/sql"
 	"embed"
 	"fmt"
+
+	"{{.ModuleName}}/database/models"
+
+	"github.com/rogeecn/atom"
+	"go.ipao.vip/atom/container/container"
+	"go.ipao.vip/atom/contracts/contracts"
+	"go.ipao.vip/atom/opt/opt"
+	"gorm.io/gorm"
 )
 
 //go:embed migrations/*
@@ -31,3 +39,17 @@ func WrapLikeLeft(v string) string {
 func WrapLikeRight(v string) string {
 	return "%" + v
 }
+
+func Provide(...opt.Option) error {
+	return container.Container.Provide(func(db *gorm.DB) contracts.Initial {
+		models.SetDefault(db)
+		return models.Q
+	}, atom.GroupInitial)
+}
+
+func DefaultProvider() container.ProviderContainer {
+	return container.ProviderContainer{
+		Provider: Provide,
+		Options:  []opt.Option{},
+	}
+}