Come leggere n righe da un file in clojure

Question

Voglio leggere le prime n righe da un file usando clojure. Qui è il mio codice:

(defn read-nth-line [file] 
    (with-open [rdr (reader file)] 
    (loop [line-number 0] 
     (when (< line-number 20) 
      (nth (line-seq rdr) line-number) 
      (recur (inc line-number)))))) 

ma quando corro

user=> (read-nth-line "test.txt") 

IndexOutOfBoundsException clojure.lang.RT.nthFrom (RT.java:871) 

non ho idea del perché ho ottenuto un errore del genere.

Answer 1

Il codice genera un errore di limite perché si chiama line-seq più volte sullo stesso lettore. Se si desidera ottenere un certo numero di linee da un lettore, si dovrebbe chiamare line-seq solo una volta, poi prendere il numero desiderato di linee da quella sequenza:

(require '[clojure.java.io :as io]) 

(defn lines [n filename] 
    (with-open [rdr (io/reader filename)] 
    (doall (take n (line-seq rdr))))) 

Esempio:

(run! println (lines 20 "test.txt")) 

Se test.txt contiene meno di 20 righe, questo semplicemente stamperà tutte le linee nel file.