v/examples/news_fetcher.v

78 lines
1.5 KiB
V
Raw Normal View History

2020-01-23 21:04:46 +01:00
// Copyright (c) 2019-2020 Alexander Medvednikov. All rights reserved.
2019-06-23 04:21:30 +02:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
2019-12-30 05:42:23 +01:00
import net.http
2019-06-22 21:51:07 +02:00
import json
import sync
const (
2019-12-30 10:45:56 +01:00
nr_threads = 4
2019-06-22 21:51:07 +02:00
)
struct Story {
title string
2019-12-21 23:41:42 +01:00
url string
2019-06-22 21:51:07 +02:00
}
struct Fetcher {
mut:
2019-12-21 23:41:42 +01:00
mu &sync.Mutex
ids []int
cursor int
wg &sync.WaitGroup
2019-06-22 21:51:07 +02:00
}
fn (f mut Fetcher) fetch() {
for {
if f.cursor >= f.ids.len {
return
}
id := f.ids[f.cursor]
f.mu.lock()
2019-06-22 21:51:07 +02:00
f.cursor++
f.mu.unlock()
cursor := f.cursor
2019-07-29 19:18:26 +02:00
resp := http.get('https://hacker-news.firebaseio.com/v0/item/${id}.json') or {
println('failed to fetch data from /v0/item/${id}.json')
exit(1)
}
2019-12-21 23:41:42 +01:00
story := json.decode(Story,resp.text) or {
2019-06-23 10:41:42 +02:00
println('failed to decode a story')
exit(1)
2019-06-22 21:51:07 +02:00
}
println('#$cursor) $story.title | $story.url')
f.wg.done()
2019-06-22 21:51:07 +02:00
}
}
2019-07-30 15:06:16 +02:00
// Fetches top HN stories in 4 coroutines
2019-06-22 21:51:07 +02:00
fn main() {
2019-07-29 19:18:26 +02:00
resp := http.get('https://hacker-news.firebaseio.com/v0/topstories.json') or {
println('failed to fetch data from /v0/topstories.json')
return
}
2019-12-21 23:41:42 +01:00
mut ids := json.decode([]int,resp.text) or {
2019-07-29 19:18:26 +02:00
println('failed to decode topstories.json')
2019-06-22 21:51:07 +02:00
return
}
2019-07-30 15:06:16 +02:00
if ids.len > 10 {
// ids = ids[:10]
mut tmp := [0].repeat(10)
2019-12-21 23:41:42 +01:00
for i := 0; i < 10; i++ {
2019-07-30 15:06:16 +02:00
tmp[i] = ids[i]
}
ids = tmp
}
2019-12-21 23:41:42 +01:00
mut fetcher := &Fetcher{
ids: ids
mu: sync.new_mutex()
wg: sync.new_waitgroup()
2019-12-21 23:41:42 +01:00
}
2019-10-25 16:24:40 +02:00
fetcher.wg.add(ids.len)
2019-12-30 10:45:56 +01:00
for i := 0; i < nr_threads; i++ {
2019-06-22 21:51:07 +02:00
go fetcher.fetch()
}
2019-10-25 16:24:40 +02:00
fetcher.wg.wait()
2019-06-22 21:51:07 +02:00
}