Function getEffectiveQueryDataPolicy

backend/api/v1/sql_service.go:470–496

(
	ctx context.Context,
	stores *store.Store,
	licenseService *enterprise.LicenseService,
	limit int32,
	projectID string,
)

}

func getEffectiveQueryDataPolicy(
	ctx context.Context,
	stores *store.Store,
	licenseService *enterprise.LicenseService,
	limit int32,
	projectID string,
) *store.EffectiveQueryDataPolicy {
	value := &store.EffectiveQueryDataPolicy{
		MaximumResultSize: common.DefaultMaximumSQLResultSize,
		MaximumResultRows: math.MaxInt32,
	}
	if err := licenseService.IsFeatureEnabled(ctx, common.GetWorkspaceIDFromContext(ctx), v1pb.PlanFeature_FEATURE_QUERY_POLICY); err == nil {
		policy, err := stores.GetEffectiveQueryDataPolicy(ctx, common.GetWorkspaceIDFromContext(ctx), common.FormatProject(projectID))
		if err != nil {
			slog.Error("failed to get the query data policy", log.BBError(err))
			return value
		}
		value = policy
	}
	if limit > 0 {
		value.MaximumResultRows = min(limit, value.MaximumResultRows)
	}
	if value.MaximumResultRows == math.MaxInt32 {
		value.MaximumResultRows = 0
	}
	return value
}

type accessCheckFunc func(context.Context, *store.InstanceMessage, *store.DatabaseMessage, *store.UserMessage, []*parserbase.QuerySpan, bool /* isExplain */, []parserbase.Statement, string /* requestSchema */, bool /* multiStatement */) error