Function TestQueryLoggerFlushBatchPersistsOrgID

server/querylog_test.go:449–539

(t *testing.T)

type captureQueryLogExecutor struct {
	feedbackExecutor
	entries []QueryLogEntry
}

func (e *captureQueryLogExecutor) Log(entry QueryLogEntry) {
	e.entries = append(e.entries, entry)
}

func TestLogQueryPrefersExecutorQueryLogSink(t *testing.T) {
	c, ql, cleanup := newFeedbackClientConn(t)
	defer cleanup()
	exec := &captureQueryLogExecutor{}
	c.executor = exec
	c.server.cfg.QueryLog.Enabled = true

	c.logQuery(time.Unix(1700000000, 0).UTC(), "SELECT 1", "", "SELECT", 1, 0, "", "", "simple")

	if len(exec.entries) != 1 {
		t.Fatalf("expected executor query-log sink to receive one entry, got %d", len(exec.entries))
	}
	if exec.entries[0].Query != "SELECT 1" {
		t.Fatalf("unexpected forwarded query-log entry: %#v", exec.entries[0])
	}
	select {
	case entry := <-ql.ch:
		t.Fatalf("server query logger should not receive entry when executor sink is available: %#v", entry)
	default:
	}
}

// TestLogQueryStampsScopedQueryID pins the correlation contract: every event
// logged for one statement carries that statement's ID, and a new statement
// gets a new one. Reusing the previous statement's ID would silently merge two
// statements in the query log.
func TestLogQueryStampsScopedQueryID(t *testing.T) {
	c, _, cleanup := newFeedbackClientConn(t)
	defer cleanup()
	exec := &captureQueryLogExecutor{}
	c.executor = exec
	c.server.cfg.QueryLog.Enabled = true

	first := c.beginQueryMetrics(time.Unix(1700000000, 0).UTC())
	// Two events for one statement — e.g. a per-statement log inside a batch
	// followed by the statement's terminal event.
	c.logQuery(time.Unix(1700000000, 0).UTC(), "SELECT 1", "", "SELECT", 1, 0, "", "", "simple")
	c.logQuery(time.Unix(1700000000, 0).UTC(), "SELECT 1", "", "SELECT", 1, 0, "", "", "simple")
	c.finishQueryMetrics(first)

	second := c.beginQueryMetrics(time.Unix(1700000001, 0).UTC())
	c.logQuery(time.Unix(1700000001, 0).UTC(), "SELECT 2", "", "SELECT", 1, 0, "", "", "simple")
	c.finishQueryMetrics(second)

	if len(exec.entries) != 3 {
		t.Fatalf("expected 3 entries, got %d", len(exec.entries))
	}
	for i, entry := range exec.entries {
		if entry.QueryID == "" {
			t.Fatalf("entry %d has no query_id", i)
		}