Function extractStatement

backend/plugin/parser/pg/restore.go:255–287

(statement string, backupItem *storepb.PriorBackupDetail_Item)

}

func extractStatement(statement string, backupItem *storepb.PriorBackupDetail_Item) (string, error) {
	if backupItem == nil {
		return "", errors.Errorf("backup item is nil")
	}

	stmts, err := ParsePg(statement)
	if err != nil {
		return "", errors.Wrapf(err, "failed to parse statement")
	}

	var sqls []string
	posMapper := base.NewByteOffsetPositionMapper(statement)
	for _, stmt := range stmts {
		if stmt.Empty() {
			continue
		}
		switch stmt.AST.(type) {
		case *ast.UpdateStmt, *ast.DeleteStmt:
			nodeLoc := ast.NodeLoc(stmt.AST)
			startPos := posMapper.Position(nodeLoc.Start)
			startPos.Column-- // 0-based
			endPos := posMapper.Position(nodeLoc.End)
			if inRange(startPos, endPos, backupItem.StartPosition, backupItem.EndPosition) {
				// Extract statement text without trailing semicolon/whitespace.
				text := strings.TrimRight(strings.TrimSpace(stmt.Text), ";")
				sqls = append(sqls, text)
			}
		default:
		}
	}

	return strings.Join(sqls, ";\n"), nil
}

func inRange(start, end, targetStart, targetEnd *storepb.Position) bool {
	if start.Line < targetStart.Line || (start.Line == targetStart.Line && start.Column < targetStart.Column) {