Skip to content

Commit

Permalink
cli: fix debug zip with new columns for cluster settings
Browse files Browse the repository at this point in the history
The addition of new columns on cluster_settings view done
on cockroachdb#104449 were causing debug zip fails to add the
information from that table, since the query used to
gather the information was doing a join and not considering
the new columns.

This commit updates the query to use the explicit columns,
so even if new columns are added it won't be a problem in the future.
It also adds tests for all custom querys used to generate the
debug zip, so this type of issue would have been caught.

Fixes cockroachdb#107103

Release note (bug fix): Debug zip now are properly showing the
information from cluster_settings.
  • Loading branch information
maryliag committed Jul 18, 2023
1 parent 25855ac commit 68daa3b
Show file tree
Hide file tree
Showing 2 changed files with 49 additions and 9 deletions.
18 changes: 9 additions & 9 deletions pkg/cli/zip_table_registry.go
Original file line number Diff line number Diff line change
Expand Up @@ -217,15 +217,15 @@ var zipInternalTablesPerCluster = DebugZipTableRegistry{
},
},
"crdb_internal.cluster_settings": {
customQueryRedacted: `SELECT * FROM (
SELECT *
FROM crdb_internal.cluster_settings
WHERE type <> 's'
) UNION (
SELECT variable, '<redacted>' as value, type, public, description
FROM crdb_internal.cluster_settings g
WHERE type = 's'
)`,
customQueryRedacted: `SELECT
variable,
CASE WHEN type = 's' THEN '<redacted>' ELSE value END value,
type,
public,
description,
default_value,
origin
FROM crdb_internal.cluster_settings`,
},
"crdb_internal.cluster_transactions": {
// `last_auto_retry_reason` contains error text that may contain
Expand Down
40 changes: 40 additions & 0 deletions pkg/cli/zip_table_registry_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,11 +11,18 @@
package cli

import (
"context"
"fmt"
"testing"

"github.com/cockroachdb/cockroach/pkg/base"
"github.com/cockroachdb/cockroach/pkg/sql/tests"
"github.com/cockroachdb/cockroach/pkg/testutils/serverutils"
"github.com/cockroachdb/cockroach/pkg/testutils/sqlutils"

"github.com/cockroachdb/cockroach/pkg/util/leaktest"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)

func TestQueryForTable(t *testing.T) {
Expand Down Expand Up @@ -134,3 +141,36 @@ you must redact them at the SQL level.`
}
}
}

func executeAllCustomQuerys(
t *testing.T, sqlDB *sqlutils.SQLRunner, tableRegistry DebugZipTableRegistry,
) {
for table, regConfig := range tableRegistry {
if regConfig.customQueryRedacted != "" {
rows := sqlDB.Query(t, regConfig.customQueryRedacted)
require.NoError(t, rows.Err(), "failed to select for table %s redacted", table)
}

if regConfig.customQueryUnredacted != "" {
rows := sqlDB.Query(t, regConfig.customQueryUnredacted)
require.NoError(t, rows.Err(), "failed to select for table %s unredacted", table)
}
}
}

func TestCustomQuery(t *testing.T) {
defer leaktest.AfterTest(t)()
ctx := context.Background()

params, _ := tests.CreateTestServerParams()
cluster := serverutils.StartNewTestCluster(t, 3 /* numNodes */, base.TestClusterArgs{
ServerArgs: params,
})
testConn := cluster.ServerConn(0 /* idx */)
sqlDB := sqlutils.MakeSQLRunner(testConn)
defer cluster.Stopper().Stop(ctx)

executeAllCustomQuerys(t, sqlDB, zipInternalTablesPerCluster)
executeAllCustomQuerys(t, sqlDB, zipInternalTablesPerNode)
executeAllCustomQuerys(t, sqlDB, zipSystemTables)
}

0 comments on commit 68daa3b

Please sign in to comment.