util/log: dedup some code by providing FromLogEntry

Epic: none
Release note: None

Author: stevendanna

pkg/jobs/jobs_test.go

@@ -8,7 +8,6 @@ package jobs_test
 import (
 	"context"
 	gosql "database/sql"
-	"encoding/json"
 	"fmt"
 	"path/filepath"
 	"reflect"
@@ -232,14 +231,7 @@ func (rts *registryTestSuite) setUp(t *testing.T) func() {
 		t,
 		[]logpb.Channel{logpb.Channel_OPS},
 		[]string{"status_change"},
-		func(entry logpb.Entry) (eventpb.StatusChange, error) {
-			var structuredPayload eventpb.StatusChange
-			err := json.Unmarshal([]byte(entry.Message[entry.StructuredStart:entry.StructuredEnd]), &structuredPayload)
-			if err != nil {
-				return structuredPayload, err
-			}
-			return structuredPayload, nil
-		},
+		logtestutils.FromLogEntry[eventpb.StatusChange],
 	)
 
 	rts.statusChangeLogSpy = spy

pkg/sql/event_log_test.go

@@ -125,13 +125,7 @@ func TestStructuredEventLogging_txnReadTimestamp(t *testing.T) {
 		t,
 		[]logpb.Channel{logpb.Channel_SQL_SCHEMA},
 		[]string{"create_table"},
-		func(entry logpb.Entry) (eventpb.CreateTable, error) {
-			var cte eventpb.CreateTable
-			if err := json.Unmarshal([]byte(entry.Message[entry.StructuredStart:entry.StructuredEnd]), &cte); err != nil {
-				return cte, err
-			}
-			return cte, nil
-		},
+		logtestutils.FromLogEntry[eventpb.CreateTable],
 	)
 
 	cleanup := log.InterceptWith(ctx, appLogsSpy)

pkg/sql/sql_exec_log_test.go

@@ -7,7 +7,6 @@ package sql
 
 import (
 	"context"
-	"encoding/json"
 	"testing"
 
 	"github.com/cockroachdb/cockroach/pkg/base"
@@ -38,13 +37,7 @@ func TestSqlExecLog(t *testing.T) {
 		t,
 		[]logpb.Channel{logpb.Channel_SQL_EXEC},
 		[]string{"query_execute"},
-		func(entry logpb.Entry) (eventpb.QueryExecute, error) {
-			var qe eventpb.QueryExecute
-			if err := json.Unmarshal([]byte(entry.Message[entry.StructuredStart:entry.StructuredEnd]), &qe); err != nil {
-				return qe, err
-			}
-			return qe, nil
-		},
+		logtestutils.FromLogEntry[eventpb.QueryExecute],
 		func(entry logpb.Entry, qe eventpb.QueryExecute) bool {
 			// Filter out internal queries.
 			return qe.ExecMode != executorTypeInternal.logLabel()

pkg/util/log/event_log_test.go

@@ -7,7 +7,6 @@ package log_test
 
 import (
 	"context"
-	"encoding/json"
 	"testing"
 	"time"
 
@@ -73,12 +72,8 @@ func TestEventLog(t *testing.T) {
 		[]logpb.Channel{logpb.Channel_DEV},
 		[]string{logtestutils.TestEventType},
 		func(entry logpb.Entry) (TestEventSev, error) {
-			var structuredPayload logtestutils.TestEvent
-			err := json.Unmarshal([]byte(entry.Message[entry.StructuredStart:entry.StructuredEnd]), &structuredPayload)
-			if err != nil {
-				return TestEventSev{event: structuredPayload, severity: entry.Severity}, err
-			}
-			return TestEventSev{event: structuredPayload, severity: entry.Severity}, nil
+			te, err := logtestutils.FromLogEntry[logtestutils.TestEvent](entry)
+			return TestEventSev{event: te, severity: entry.Severity}, err
 		},
 	)
 

pkg/util/log/logtestutils/structured_log_spy.go

@@ -104,6 +104,12 @@ func FormatEntryAsJSON(entry logpb.Entry) (string, error) {
 	return printJSONMap(jsonMap)
 }
 
+func FromLogEntry[T any](entry logpb.Entry) (T, error) {
+	var payload T
+	err := json.Unmarshal([]byte(entry.Message[entry.StructuredStart:entry.StructuredEnd]), &payload)
+	return payload, err
+}
+
 // StructuredLogSpy is a test utility that intercepts structured log entries
 // and stores them in memory. It can be used to verify the contents of log
 // entries in tests.