Add global DefaultClient (readonly), WithUserAgent option

Signed-off-by: Vartan Benohanian <vartanbeno@gmail.com>
This commit is contained in:
Vartan Benohanian
2020-08-29 02:48:22 -04:00
parent aac3b48d6e
commit 37e712b334
11 changed files with 260 additions and 142 deletions
+29 -37
View File
@@ -3,8 +3,9 @@ package main
import (
"context"
"fmt"
"log"
"os"
"os/signal"
"syscall"
"time"
"github.com/vartanbeno/go-reddit/reddit"
@@ -13,44 +14,35 @@ import (
var ctx = context.Background()
func main() {
if err := run(); err != nil {
log.Fatal(err)
}
}
sig := make(chan os.Signal, 1)
defer close(sig)
signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
func run() (err error) {
credentials := &reddit.Credentials{
ID: "id",
Secret: "secret",
Username: "username",
Password: "password",
}
client, err := reddit.NewClient(credentials)
if err != nil {
return
}
posts, errs, stop := client.Stream.Posts("AskReddit", reddit.StreamInterval(time.Second*3), reddit.StreamDiscardInitial)
posts, errs, stop := reddit.DefaultClient.Stream.Posts("AskReddit", reddit.StreamInterval(time.Second*3), reddit.StreamDiscardInitial)
defer stop()
go func() {
for {
select {
case post, ok := <-posts:
if !ok {
return
}
fmt.Printf("Received post: %s\n", post.Title)
case err, ok := <-errs:
if !ok {
return
}
fmt.Fprintf(os.Stderr, "Error! %v\n", err)
}
}
}()
timer := time.NewTimer(time.Minute)
defer timer.Stop()
<-time.After(time.Minute)
return
for {
select {
case post, ok := <-posts:
if !ok {
return
}
fmt.Printf("Received post: %s\n", post.Title)
case err, ok := <-errs:
if !ok {
return
}
fmt.Fprintf(os.Stderr, "Error! %v\n", err)
case rcvSig, ok := <-sig:
if !ok {
return
}
fmt.Printf("Stopping due to %s signal\n", rcvSig)
case <-timer.C:
return
}
}
}