Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 44319d03b9 | |||
| 2f57edd277 | |||
| cf0ee224f7 |
@@ -15,8 +15,6 @@ import (
|
||||
var (
|
||||
// MigrateTables holds the tables that need to be auto-migrated on database initialization
|
||||
MigrateTables []any
|
||||
// Init is an optional initialization function that can be executed after database connection is established
|
||||
Init *func() = nil
|
||||
)
|
||||
|
||||
// NewDatabase creates a new database connection based on the provided driver type
|
||||
@@ -36,11 +34,6 @@ func NewDatabase(driver string, dsn []string, options *types.SqlOptions) (db *go
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// Execute the Init function if it's not nil
|
||||
if Init != nil {
|
||||
(*Init)()
|
||||
}
|
||||
|
||||
return db, nil
|
||||
}
|
||||
|
||||
|
||||
@@ -108,6 +108,17 @@ func (s *Service) Gateway(grpcAddr string, httpAddr string) {
|
||||
http.ListenAndServe(httpAddr, s.Opts.GatewayMux)
|
||||
}
|
||||
|
||||
func (s *Service) Use(initFunc *func() error) {
|
||||
// Execute the Init function if it's not nil
|
||||
if initFunc != nil {
|
||||
err := (*initFunc)()
|
||||
if err != nil {
|
||||
print.Error(err.Error())
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Service) Stop() {
|
||||
s.GrpcSrv.GracefulStop()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user