diff --git a/database/new.go b/database/new.go
index 1d0fd19..4baf740 100644
--- a/database/new.go
+++ b/database/new.go
@@ -15,11 +15,13 @@ 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
+	InitFunc *func() error = nil
 )
 
 // NewDatabase creates a new database connection based on the provided driver type
 // It supports both MySQL and PostgreSQL databases
-func NewDatabase(driver string, dsn []string, options *types.SqlOptions, Init *func()) (db *gorm.DB, err error) {
+func NewDatabase(driver string, dsn []string, options *types.SqlOptions) (db *gorm.DB, err error) {
 	driver = strings.ToLower(driver)
 	switch driver {
 	case "mysql":
@@ -34,11 +36,6 @@ func NewDatabase(driver string, dsn []string, options *types.SqlOptions, Init *f
 		return nil, err
 	}
 
-	// Execute the Init function if it's not nil
-	if Init != nil {
-		(*Init)()
-	}
-
 	return db, nil
 }
 
diff --git a/with/init.go b/with/init.go
new file mode 100644
index 0000000..1b60b68
--- /dev/null
+++ b/with/init.go
@@ -0,0 +1,18 @@
+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)
+		}
+	}
+}