Heavily optimize CSV reader.

This commit is contained in:
Dominik Pantůček 2023-03-31 19:20:49 +02:00
parent a216b27825
commit 306cca2ae3

View file

@ -41,49 +41,52 @@
testing
progress)
;; Curry version of line parser with configurable cell separator and
;; Curry version of line parser with configurable cell separator and
;; string delimiter. Returns a list of lists of strings.
(define ((make-csv-line-parser separator string-delimiter) line)
(let loop ((tokens (irregex-extract (irregex "." 'u) line))
(let loop ((tokens (string->list line))
(res '())
(state 1))
(if (null? tokens)
(reverse res)
(reverse
(map
(lambda (cell)
(list->string (reverse cell)))
res))
(let ((token (car tokens)))
(case state
((0) ; Parsing regular unquoted cell data - separator creates new cell
(if (equal? token separator)
(if (eq? token separator)
(loop (cdr tokens)
res
1) ; Start a new cell
(loop (cdr tokens)
(cons (string-append (car res) token) (cdr res))
(cons (cons token (car res)) (cdr res))
0)))
((1) ; Starting a new cell - check for string delimiter
(if (equal? token string-delimiter)
(if (eq? token string-delimiter)
(loop (cdr tokens)
(cons "" res) ; If it is quoted, keep even empty strings there
(cons '() res) ; If it is quoted, keep even empty strings there
2)
(if (equal? token separator)
(if (eq? token separator)
(loop (cdr tokens)
(cons "" res) ; This was an empty cell
(cons '() res) ; This was an empty cell
1) ; Another new cell awaiting
(loop (cdr tokens)
(cons token res) ; first token of regular new cell
(cons (list token) res) ; first token of regular new cell
0))))
((2) ; Parsing quoted cell data - no support for escaping string delimiter!
(if (equal? token string-delimiter)
(if (eq? token string-delimiter)
(loop (cdr tokens)
res
0) ; There shouldn't be anything more, but it is safe to append the rest as normal unquoted data
(loop (cdr tokens)
(cons (string-append (car res) token) (cdr res))
(cons (cons token (car res)) (cdr res))
2)))))))) ; Continue inside quoted data
;; Parses given CSV lines list
(define (csv-parse-lines lines . args)
(let* ((separator (get-keyword #:separator args (lambda () ";")))
(string-delimiter (get-keyword #:string-delimiter args (lambda () "\"")))
(let* ((separator (get-keyword #:separator args (lambda () #\;)))
(string-delimiter (get-keyword #:string-delimiter args (lambda () #\")))
(csv-parse-line (make-csv-line-parser separator string-delimiter))
(total (max (sub1 (length lines)) 1)))
(let loop ((lines lines)