Function TestMemoryLeaks_UpdateStatementPool

pkg/sql/ast/pool_test.go:433–496

(t *testing.T)

}

func TestMemoryLeaks_UpdateStatementPool(t *testing.T) {
	runtime.GC()
	runtime.GC()
	time.Sleep(10 * time.Millisecond)

	var m1 runtime.MemStats
	runtime.ReadMemStats(&m1)

	const iterations = 10000

	t.Logf("Running %d iterations of UpdateStatement pool operations...", iterations)

	for i := 0; i < iterations; i++ {
		stmt := GetUpdateStatement()

		stmt.TableName = "users"
		stmt.Assignments = append(stmt.Assignments, UpdateExpression{
			Column: &Identifier{Name: "status"},
			Value:  &LiteralValue{Value: "active"},
		})
		stmt.Where = &BinaryExpression{
			Left:     &Identifier{Name: "id"},
			Operator: "=",
			Right:    &LiteralValue{Value: "123"},
		}

		PutUpdateStatement(stmt)

		if i%1000 == 0 && i > 0 {
			runtime.GC()
		}
	}

	runtime.GC()
	runtime.GC()
	time.Sleep(10 * time.Millisecond)

	var m2 runtime.MemStats
	runtime.ReadMemStats(&m2)

	allocDiff := int64(m2.Alloc) - int64(m1.Alloc)
	totalAllocDiff := int64(m2.TotalAlloc) - int64(m1.TotalAlloc)
	bytesPerOp := float64(totalAllocDiff) / float64(iterations)

	t.Logf("UpdateStatement pool memory stats:")
	t.Logf("  Alloc diff: %d bytes", allocDiff)
	t.Logf("  TotalAlloc diff: %d bytes", totalAllocDiff)
	t.Logf("  Bytes per operation: %.2f", bytesPerOp)

	const maxAllocIncrease = 1024 * 1024 // 1MB
	const maxBytesPerOp = 6000           // 6KB

	if allocDiff > maxAllocIncrease {
		t.Errorf("Memory leak in UpdateStatement pool: %d bytes (threshold: %d)", allocDiff, maxAllocIncrease)
	}

	if bytesPerOp > maxBytesPerOp {
		t.Errorf("High memory usage in UpdateStatement pool: %.2f bytes/op (threshold: %d)", bytesPerOp, maxBytesPerOp)