Как читать n строк из файла в clojure

Question

Я хочу прочитать первые n строк из файла, используя clojure. Вот мой код:

(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)))))) 

, но когда я бегу

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

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

Я понятия не имею, почему я получил такую ​​ошибку.

Answer 1

Ваш код выдает ошибку за пределы, потому что вы вызываете line-seq несколько раз на одного и того же читателя. Если вы хотите, чтобы получить количество строк от читателя, вы должны вызвать line-seq только один раз, а затем взять нужное количество строк из этой последовательности:

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

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

Пример:

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

Если test.txt содержит менее 20 строк, это просто напечатает все строки в файле.