Function TestPreparedStmtLruFromTransaction

tests/prepared_stmt_test.go:94–151

(t *testing.T)

}

func TestPreparedStmtLruFromTransaction(t *testing.T) {
	db, _ := OpenTestConnection(&gorm.Config{PrepareStmt: true, PrepareStmtMaxSize: 10, PrepareStmtTTL: 20 * time.Second})

	tx := db.Begin()
	defer func() {
		if r := recover(); r != nil {
			tx.Rollback()
		}
	}()
	if err := tx.Error; err != nil {
		t.Errorf("Failed to start transaction, got error %v\n", err)
	}

	if err := tx.Where("name=?", "zzjin").Delete(&User{}).Error; err != nil {
		tx.Rollback()
		t.Errorf("Failed to run one transaction, got error %v\n", err)
	}

	if err := tx.Create(&User{Name: "zzjin"}).Error; err != nil {
		tx.Rollback()
		t.Errorf("Failed to run one transaction, got error %v\n", err)
	}

	if err := tx.Commit().Error; err != nil {
		t.Errorf("Failed to commit transaction, got error %v\n", err)
	}

	if result := db.Where("name=?", "zzjin").Delete(&User{}); result.Error != nil || result.RowsAffected != 1 {
		t.Fatalf("Failed, got error: %v, rows affected: %v", result.Error, result.RowsAffected)
	}

	tx2 := db.Begin()
	if result := tx2.Where("name=?", "zzjin").Delete(&User{}); result.Error != nil || result.RowsAffected != 0 {
		t.Fatalf("Failed, got error: %v, rows affected: %v", result.Error, result.RowsAffected)
	}

	tx2.Commit()
	// Attempt to convert the connection pool of tx to the *gorm.PreparedStmtDB type.
	// If the conversion is successful, ok will be true and conn will be the converted object;
	// otherwise, ok will be false and conn will be nil.
	conn, ok := tx.ConnPool.(*gorm.PreparedStmtDB)
	// Get the number of statement keys stored in the PreparedStmtDB.
	lens := len(conn.Stmts.Keys())
	// Check if the number of stored statement keys is 0.
	if lens == 0 {
		// If the number is 0, it means there are no statements stored in the LRU cache.
		// The test fails and an error message is output.
		t.Fatalf("lru should not be empty")
	}
	// Wait for 40 seconds to give the statements in the cache enough time to expire.
	time.Sleep(time.Second * 40)
	// Assert whether the connection pool of tx is successfully converted to the *gorm.PreparedStmtDB type.
	AssertEqual(t, ok, true)
	// Assert whether the number of statement keys stored in the PreparedStmtDB is 0 after 40 seconds.
	// If it is not 0, it means the statements in the cache have not expired as expected.
	AssertEqual(t, len(conn.Stmts.Keys()), 0)

}