upper case collected commands

DataDog · Dec 17, 2024 · c6d44ad · c6d44ad
1 parent 90812ab
commit c6d44ad
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/normalizer.go b/normalizer.go
@@ -176,7 +176,7 @@ func (n *Normalizer) collectMetadata(token *Token, lastToken *Token, statementMe
 		}
 		if n.config.CollectCommands && isCommand(tokenVal) {
 			// Collect commands
-			statementMetadata.Commands = append(statementMetadata.Commands, tokenVal)
+			statementMetadata.Commands = append(statementMetadata.Commands, strings.ToUpper(tokenVal))
 		} else if isWith(lastToken.Value) && token.Type == IDENT {
 			// Collect CTEs so we can skip them later in table collection
 			ctes[tokenVal] = true

diff --git a/normalizer_test.go b/normalizer_test.go
@@ -163,7 +163,7 @@ multiline comment */
 			statementMetadata: StatementMetadata{
 				Tables:     []string{"users", "test_users", "user?"},
 				Comments:   []string{},
-				Commands:   []string{"CREATE", "BEGIN", "SELECT", "Update", "Delete"},
+				Commands:   []string{"CREATE", "BEGIN", "SELECT", "UPDATE", "DELETE"},
 				Procedures: []string{},
 				Size:       49,
 			},

diff --git a/obfuscate_and_normalize_test.go b/obfuscate_and_normalize_test.go
@@ -127,7 +127,7 @@ multiline comment */
 			statementMetadata: StatementMetadata{
 				Tables:     []string{"v$sql"},
 				Comments:   []string{},
-				Commands:   []string{"select"},
+				Commands:   []string{"SELECT"},
 				Procedures: []string{},
 				Size:       11,
 			},
@@ -163,7 +163,7 @@ multiline comment */
 			statementMetadata: StatementMetadata{
 				Tables:     []string{},
 				Comments:   []string{},
-				Commands:   []string{"begin", "execute"},
+				Commands:   []string{"BEGIN", "EXECUTE"},
 				Procedures: []string{},
 				Size:       12,
 			},
@@ -234,7 +234,7 @@ multiline comment */
 			statementMetadata: StatementMetadata{
 				Tables:     []string{`public.users`},
 				Comments:   []string{},
-				Commands:   []string{"select"},
+				Commands:   []string{"SELECT"},
 				Procedures: []string{},
 				Size:       18,
 			},