Function TestMemoryLeaks_InsertStatementPool

pkg/sql/ast/pool_test.go:374–431

(t *testing.T)

}

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

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

		stmt.TableName = "users"
		stmt.Columns = append(stmt.Columns, &Identifier{Name: "name"}, &Identifier{Name: "email"})
		// Values is now [][]Expression for multi-row support
		stmt.Values = append(stmt.Values, []Expression{&LiteralValue{Value: "John"}, &LiteralValue{Value: "john@test.com"}})

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

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

	if allocDiff <= maxAllocIncrease && bytesPerOp <= maxBytesPerOp {
		t.Logf("✅ InsertStatement pool memory leak test PASSED")
	}
}