|
4 | 4 | [clojure.string :as str] |
5 | 5 | [clojure.core.reducers :as r])) |
6 | 6 |
|
| 7 | +(def log-regex |
| 8 | + #"^([\d+.]+) (\S+) (\S+) (\[[\w+/]+:[\w+:]+ \+\d+\]) \"(.+?)\" (\d{3}) (\d+) \"([^\"]+)\" \"(.+?)\"") |
| 9 | + |
7 | 10 | (defn parse-log |
8 | 11 | [^String log] |
9 | | - (let [regex #"^([\d+.]+) (\S+) (\S+) (\[[\w+/]+:[\w+:]+ \+\d+\]) \"(.+?)\" (\d{3}) (\d+) \"([^\"]+)\" \"(.+?)\"" |
10 | | - [_ ip user-name date-time request response size referer user-agent] (re-find regex log)] |
| 12 | + (let [[_ ip _ user-name date-time request response size referer user-agent] (re-find log-regex log)] |
11 | 13 | {:ip ip |
12 | 14 | :user-name user-name |
13 | 15 | :date-time date-time |
|
19 | 21 |
|
20 | 22 | (defn get-logs-paths |
21 | 23 | [dir-path] |
22 | | - (->> (io/as-file dir-path) |
23 | | - (.listFiles) |
24 | | - (filter #(.isFile %)) |
25 | | - (map #(.getPath %)))) |
| 24 | + (->> (io/as-file dir-path) |
| 25 | + (file-seq) |
| 26 | + (filter #(.isFile %)))) |
26 | 27 |
|
27 | 28 | (defn read-logs |
28 | 29 | [logs] |
29 | 30 | (map #(io/reader %) logs)) |
30 | 31 |
|
| 32 | +(defn ->url [log] |
| 33 | + {:pre [(contains? log :request)]} |
| 34 | + (let [request (-> log :request)] |
| 35 | + (when request |
| 36 | + (-> request |
| 37 | + (str/split #" ") |
| 38 | + second)))) |
| 39 | + |
31 | 40 | (defn filter-by-url |
32 | 41 | [url logs] |
33 | 42 | (if (= url :all) |
34 | 43 | logs |
35 | | - (letfn [(->url [log] (-> log :request (str/split #" ") second))] |
36 | | - (r/filter #(= url (->url %)) logs)))) |
| 44 | + (r/filter #(= url (->url %)) logs))) |
| 45 | + |
| 46 | +(comment |
| 47 | + (r/fold str (filter-by-url "test" [{:request "test"} {:request "test2"} {:request "test"}]))) |
| 48 | + |
| 49 | +(defn ->referer [log] |
| 50 | + {:pre [(contains? log :referer)]} |
| 51 | + (-> log :referer)) |
37 | 52 |
|
38 | 53 | (defn filter-by-referer |
39 | 54 | [referer logs] |
40 | 55 | (if (= referer :all) |
41 | 56 | logs |
42 | | - (letfn [(->referer [log] (-> log :referer))] |
43 | | - (r/filter #(= referer (->referer %)) logs)))) |
| 57 | + (r/filter #(= referer (->referer %)) logs))) |
44 | 58 |
|
45 | 59 | (defn close-logs |
46 | 60 | [logs] |
|
59 | 73 | (r/map #(parse-int-safe (:size %))) |
60 | 74 | (r/fold +))) |
61 | 75 |
|
| 76 | +(defn process-partition |
| 77 | + [logs {:keys [url referrer]}] |
| 78 | + (->> logs |
| 79 | + (r/map parse-log) |
| 80 | + (filter-by-url url) |
| 81 | + (filter-by-referer referrer))) |
| 82 | + |
62 | 83 | (defn sum-partition |
63 | 84 | [partition] |
64 | 85 | (reduce #(+ %1 (sum-size %2)) |
65 | 86 | 0 |
66 | 87 | partition)) |
67 | 88 |
|
68 | 89 | (defn process-log |
69 | | - [log-file & |
70 | | - {:keys [url referrer]}] |
71 | | - (->> log-file |
72 | | - (line-seq) |
73 | | - (partition-all 5000) |
74 | | - (pmap #(map parse-log %)) |
75 | | - (filter-by-url url) |
76 | | - (filter-by-referer referrer) |
77 | | - (sum-partition))) |
78 | | - |
79 | | -(comment |
80 | | - (time |
81 | | - (let [file (io/reader "./logs/access.log.2")] |
82 | | - (-> file |
83 | | - (process-log {:url :all :referrer :all}) |
84 | | - (println)) |
85 | | - (.close file)))) |
86 | | - |
| 90 | + [log-file filter-params] |
| 91 | + (->> log-file |
| 92 | + (line-seq) |
| 93 | + (partition-all 5000) |
| 94 | + (pmap #(process-partition % filter-params)) |
| 95 | + (sum-partition))) |
87 | 96 |
|
88 | 97 | (defn solution |
89 | 98 | [& {:keys [url referrer] |
|
92 | 101 | get-logs-paths |
93 | 102 | read-logs)] |
94 | 103 | (->> logs |
95 | | - (pmap #(process-log % :url url :referrer referrer)) |
| 104 | + (pmap #(process-log % {:url url :referrer referrer})) |
96 | 105 | (reduce +) |
97 | 106 | (println "Bytes:")) |
98 | 107 | (close-logs logs))) |
99 | 108 |
|
100 | 109 | (comment |
101 | 110 | ;; возможные вызовы функции |
102 | | - (time (solution)) |
103 | | - (solution :url "some-url") |
104 | | - (solution :referrer "some-referrer") |
105 | | - (solution :url "some-url" :referrer "some-referrer")) |
| 111 | + (time (solution)) |
| 112 | + (time (solution :url "/rss/")) |
| 113 | + (solution :referrer "some-referrer") |
| 114 | + (solution :url "some-url" :referrer "some-referrer")) |
0 commit comments