Function getTableObjectSchema

backend/api/v1/document_masking.go:1339–1362

getTableObjectSchema retrieves the ObjectSchema for a table/collection/index/container by name from the database config. This is the shared implementation for all document databases (CosmosDB containers, MongoDB collections, Elasticsearch indices).

(ctx context.Context, stores *store.Store, instanceID, databaseName, tableName string)

// by name from the database config. This is the shared implementation for all document
// databases (CosmosDB containers, MongoDB collections, Elasticsearch indices).
func getTableObjectSchema(ctx context.Context, stores *store.Store, instanceID, databaseName, tableName string) (*storepb.ObjectSchema, error) {
	dbMetadata, err := stores.GetDBSchema(ctx, &store.FindDBSchemaMessage{
		Workspace:    common.GetWorkspaceIDFromContext(ctx),
		InstanceID:   instanceID,
		DatabaseName: databaseName,
	})
	if err != nil {
		return nil, errors.Wrapf(err, "failed to get database schema: %q", databaseName)
	}

	if dbMetadata == nil {
		return nil, nil
	}

	for _, schema := range dbMetadata.GetConfig().GetSchemas() {
		for _, table := range schema.GetTables() {
			if table.GetName() == tableName {
				return table.GetObjectSchema(), nil
			}
		}
	}

	return nil, nil
}

// objectSchemaHasSemanticTypes recursively checks if any node in the ObjectSchema has a semantic type set.
func objectSchemaHasSemanticTypes(os *storepb.ObjectSchema) bool {