Function TestMemoryLeaks_SelectStatementPool

pkg/sql/ast/pool_test.go:312–372

(t *testing.T)

}

func TestMemoryLeaks_SelectStatementPool(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 SelectStatement pool operations...", iterations)

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

		stmt.TableName = "users"
		stmt.Columns = append(stmt.Columns, &Identifier{Name: "id"}, &Identifier{Name: "name"})
		stmt.Where = &BinaryExpression{
			Left:     &Identifier{Name: "active"},
			Operator: "=",
			Right:    &LiteralValue{Value: "true"},
		}

		PutSelectStatement(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("SelectStatement 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 SelectStatement pool: %d bytes (threshold: %d)", allocDiff, maxAllocIncrease)
	}

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

	if allocDiff <= maxAllocIncrease && bytesPerOp <= maxBytesPerOp {