From e839e7ba2a1036d4edb79ba7d08e9212849d6b27 Mon Sep 17 00:00:00 2001 From: SoXX Date: Sun, 11 Aug 2024 00:26:20 +0200 Subject: [PATCH] feat(database): return gorm itself --- pkg/database/client.go | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/pkg/database/client.go b/pkg/database/client.go index 89ff2ac..57d59e2 100644 --- a/pkg/database/client.go +++ b/pkg/database/client.go @@ -4,6 +4,7 @@ import ( "context" "embed" "fmt" + otterError "git.anthrove.art/Anthrove/otter-space-sdk/v2/pkg/error" "git.anthrove.art/Anthrove/otter-space-sdk/v2/pkg/models" migrate "github.com/rubenv/sql-migrate" log "github.com/sirupsen/logrus" @@ -66,3 +67,11 @@ func migrateDatabase(dbPool *gorm.DB, config models.DatabaseConfig) error { return nil } + +func GetGorm(ctx context.Context) (*gorm.DB, error) { + if client == nil { + return nil, &otterError.Database{Reason: otterError.DatabaseIsNotConnected} + } + + return client, nil +}