Method MakeFetchOrMoveStmt

pkg/sql/plpgsql/parser/lexer.go:248–307

(isMove bool)

}

func (l *lexer) MakeFetchOrMoveStmt(isMove bool) (plpgsqltree.Statement, error) {
	if l.parser.Lookahead() != -1 {
		// Push back the lookahead token so that it can be included.
		l.PushBack(1)
	}
	prefix := "FETCH "
	if isMove {
		prefix = "MOVE "
	}
	sqlStr, terminator, err := l.ReadSqlStatement(INTO, ';')
	if err != nil {
		return nil, err
	}
	sqlStr = prefix + sqlStr
	sqlStmt, err := parser.ParseOne(sqlStr)
	if err != nil {
		return nil, err
	}
	var cursor tree.CursorStmt
	switch t := sqlStmt.AST.(type) {
	case *tree.FetchCursor:
		cursor = t.CursorStmt
	case *tree.MoveCursor:
		cursor = t.CursorStmt
	default:
		return nil, errors.Newf("invalid FETCH or MOVE syntax")
	}
	var target []plpgsqltree.Variable
	if !isMove {
		if terminator != INTO {
			return nil, errors.Newf("invalid syntax for FETCH")
		}
		// Read past the INTO.
		l.lastPos++
		startPos, endPos, _, err := l.readSQLConstruct(true /* isExpr */, false /* allowEmpty */, ';')
		if err != nil {
			return nil, err
		}
		var targetEnd int
		target, targetEnd, err = l.readTarget(startPos, endPos)
		if err != nil {
			return nil, err
		}
		if targetEnd != endPos {
			return nil, errors.Newf("expected INTO target to be a comma-separated list")
		}
		if len(target) == 0 {
			return nil, errors.Newf("expected INTO target")
		}
	}
	// Move past the semicolon.
	l.lastPos++
	ann := tree.MakeAnnotations(sqlStmt.NumAnnotations)
	return &plpgsqltree.Fetch{
		Cursor:      cursor,
		Target:      target,
		IsMove:      isMove,
		Annotations: &ann,