Method WithFlags

pkg/sql/sem/tree/format.go:448–457

WithFlags changes the flags in the FmtCtx, runs the given function, then restores the old flags.

(flags FmtFlags, fn func())

// WithFlags changes the flags in the FmtCtx, runs the given function, then
// restores the old flags.
func (ctx *FmtCtx) WithFlags(flags FmtFlags, fn func()) {
	if ctx.ann == nil && flags&flagsRequiringAnnotations != 0 {
		panic(errors.AssertionFailedf("no Annotations provided"))
	}
	oldFlags := ctx.flags
	ctx.flags = flags
	defer func() { ctx.flags = oldFlags }()

	fn()
}

// WithoutConstantRedaction modifies FmtCtx to ensure that constants are
// displayed rather than being replaced by '_', calls fn, then restores the