MCPcopy Create free account
hub / github.com/bytebase/bytebase / preExecuteMaskingCheck

Function preExecuteMaskingCheck

backend/api/v1/document_masking.go:58–73  ·  view source on GitHub ↗

--------------------------------------------------------------------------- Shared helpers --------------------------------------------------------------------------- preExecuteMaskingCheck runs engine-specific pre-execution checks that block queries with unsupported APIs. Sensitive predicate checks

(
	ctx context.Context,
	stores *store.Store,
	engine storepb.Engine,
	database *store.DatabaseMessage,
	spans []*parserbase.QuerySpan,
)

Source from the content-addressed store, hash-verified

56// queries with unsupported APIs. Sensitive predicate checks are handled
57// post-execution on a per-statement basis by checkSensitivePredicates.
58func preExecuteMaskingCheck(
59 ctx context.Context,
60 stores *store.Store,
61 engine storepb.Engine,
62 database *store.DatabaseMessage,
63 spans []*parserbase.QuerySpan,
64) error {
65 switch engine {
66 case storepb.Engine_MONGODB:
67 return preExecuteMaskingCheckMongoDB(ctx, stores, database, spans)
68 case storepb.Engine_ELASTICSEARCH:
69 return preExecuteMaskingCheckElasticsearch(ctx, stores, database, spans)
70 default:
71 return nil
72 }
73}
74
75func preExecuteMaskingCheckMongoDB(
76 ctx context.Context,

Callers 1

queryRetryFunction · 0.85

Tested by

no test coverage detected