Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| f681f0bb17 | |||
| 44319d03b9 |
@@ -15,8 +15,6 @@ import (
|
|||||||
var (
|
var (
|
||||||
// MigrateTables holds the tables that need to be auto-migrated on database initialization
|
// MigrateTables holds the tables that need to be auto-migrated on database initialization
|
||||||
MigrateTables []any
|
MigrateTables []any
|
||||||
// Init is an optional initialization function that can be executed after database connection is established
|
|
||||||
InitFunc *func() error = nil
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// NewDatabase creates a new database connection based on the provided driver type
|
// NewDatabase creates a new database connection based on the provided driver type
|
||||||
|
|||||||
@@ -108,6 +108,14 @@ func (s *Service) Gateway(grpcAddr string, httpAddr string) {
|
|||||||
http.ListenAndServe(httpAddr, s.Opts.GatewayMux)
|
http.ListenAndServe(httpAddr, s.Opts.GatewayMux)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (s *Service) Use(initFunc func() error) {
|
||||||
|
err := (initFunc)()
|
||||||
|
if err != nil {
|
||||||
|
print.Error(err.Error())
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func (s *Service) Stop() {
|
func (s *Service) Stop() {
|
||||||
s.GrpcSrv.GracefulStop()
|
s.GrpcSrv.GracefulStop()
|
||||||
}
|
}
|
||||||
|
|||||||
18
with/init.go
18
with/init.go
@@ -1,18 +0,0 @@
|
|||||||
package with
|
|
||||||
|
|
||||||
import (
|
|
||||||
"git.apinb.com/bsm-sdk/core/database"
|
|
||||||
"git.apinb.com/bsm-sdk/core/errcode"
|
|
||||||
"git.apinb.com/bsm-sdk/core/print"
|
|
||||||
)
|
|
||||||
|
|
||||||
func InitData() {
|
|
||||||
// Execute the Init function if it's not nil
|
|
||||||
if database.InitFunc != nil {
|
|
||||||
err := (*database.InitFunc)()
|
|
||||||
if err != nil {
|
|
||||||
print.Error(errcode.ErrEtcd.Error())
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Reference in New Issue
Block a user