Function ScannerFunction

internal/pipe/scanner.go:25–67

ScannerFunction creates a function-based `Stage`. The function will be passed input, one line at a time, and may emit output. See the definition of `LinewiseStageFunc` for more information.

(
	name string, newScanner NewScannerFunc, f LinewiseStageFunc,
)

// be passed input, one line at a time, and may emit output. See the
// definition of `LinewiseStageFunc` for more information.
func ScannerFunction(
	name string, newScanner NewScannerFunc, f LinewiseStageFunc,
) Stage {
	return Function(
		name,
		func(ctx context.Context, env Env, stdin io.Reader, stdout io.Writer) (theErr error) {
			scanner, err := newScanner(stdin)
			if err != nil {
				return err
			}

			var out *bufio.Writer
			if stdout != nil {
				out = bufio.NewWriter(stdout)
				defer func() {
					err := out.Flush()
					if err != nil && theErr == nil {
						// Note: this sets the named return value,
						// thereby causing the whole stage to report
						// the error.
						theErr = err
					}
				}()
			}

			for scanner.Scan() {
				if ctx.Err() != nil {
					return ctx.Err()
				}
				err := f(ctx, env, scanner.Bytes(), out)
				if err != nil {
					return err
				}
			}
			if err := scanner.Err(); err != nil {
				return err
			}

			return nil
			// `p.AddFunction()` arranges for `stdout` to be closed.
		},
	)
}