Browse Source

修改为使用内部事务

WhiteBatman 9 năm trước cách đây
mục cha
commit
70b830ae90
1 tập tin đã thay đổi với 9 bổ sung9 xóa
  1. 9 9
      engine.go

+ 9 - 9
engine.go

@@ -1425,16 +1425,16 @@ func (engine *Engine) dropAll() error {
 	session := engine.NewSession()
 	session := engine.NewSession()
 	defer session.Close()
 	defer session.Close()
 
 
-	tx, err := session.Begin()
+	err := session.begin()
 	if err != nil {
 	if err != nil {
 		return err
 		return err
 	}
 	}
 	err = session.dropAll()
 	err = session.dropAll()
 	if err != nil {
 	if err != nil {
-		tx.Rollback()
+		session.rollback()
 		return err
 		return err
 	}
 	}
-	return tx.Commit()
+	return session.commit()
 }
 }
 
 
 // CreateTables create tabls according bean
 // CreateTables create tabls according bean
@@ -1442,7 +1442,7 @@ func (engine *Engine) CreateTables(beans ...interface{}) error {
 	session := engine.NewSession()
 	session := engine.NewSession()
 	defer session.Close()
 	defer session.Close()
 
 
-	tx, err := session.Begin()
+	err := session.begin()
 	if err != nil {
 	if err != nil {
 		return err
 		return err
 	}
 	}
@@ -1450,11 +1450,11 @@ func (engine *Engine) CreateTables(beans ...interface{}) error {
 	for _, bean := range beans {
 	for _, bean := range beans {
 		err = session.CreateTable(bean)
 		err = session.CreateTable(bean)
 		if err != nil {
 		if err != nil {
-			tx.Rollback()
+			session.rollback()
 			return err
 			return err
 		}
 		}
 	}
 	}
-	return tx.Commit()
+	return session.commit()
 }
 }
 
 
 // DropTables drop specify tables
 // DropTables drop specify tables
@@ -1462,7 +1462,7 @@ func (engine *Engine) DropTables(beans ...interface{}) error {
 	session := engine.NewSession()
 	session := engine.NewSession()
 	defer session.Close()
 	defer session.Close()
 
 
-	tx, err := session.Begin()
+	err := session.begin()
 	if err != nil {
 	if err != nil {
 		return err
 		return err
 	}
 	}
@@ -1470,11 +1470,11 @@ func (engine *Engine) DropTables(beans ...interface{}) error {
 	for _, bean := range beans {
 	for _, bean := range beans {
 		err = session.DropTable(bean)
 		err = session.DropTable(bean)
 		if err != nil {
 		if err != nil {
-			tx.Rollback()
+			session.rollback()
 			return err
 			return err
 		}
 		}
 	}
 	}
-	return tx.Commit()
+	return session.commit()
 }
 }
 
 
 func (engine *Engine) createAll() error {
 func (engine *Engine) createAll() error {