Browse Source

Update database.go

Dominic Renner 6 năm trước cách đây
mục cha
commit
aaa57b353f
1 tập tin đã thay đổi với 5 bổ sung5 xóa
  1. 5 5
      database/database.go

+ 5 - 5
database/database.go

@@ -63,20 +63,20 @@ func New(dbType string, uri string) (*Database, error) {
 	return db, nil
 }
 
-func (db *Database) CreateTables() error {
-	err := db.User.CreateTable()
+func (db *Database) CreateTables(dbType string) error {
+	err := db.User.CreateTable(dbType)
 	if err != nil {
 		return err
 	}
-	err = db.Portal.CreateTable()
+	err = db.Portal.CreateTable(dbType)
 	if err != nil {
 		return err
 	}
-	err = db.Puppet.CreateTable()
+	err = db.Puppet.CreateTable(dbType)
 	if err != nil {
 		return err
 	}
-	err = db.Message.CreateTable()
+	err = db.Message.CreateTable(dbType)
 	if err != nil {
 		return err
 	}