This repository has been archived by the owner on Dec 17, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
72 lines (67 loc) · 1.49 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package main
import (
"encoding/gob"
"log"
"github.com/bgpat/tweet-picker/client"
"github.com/bgpat/tweet-picker/database"
"github.com/bgpat/tweet-picker/models"
"github.com/bgpat/tweet-picker/server"
)
func main() {
gob.Register(map[string]interface{}{})
gob.Register([]interface{}{})
if _, err := database.New(); err != nil {
log.Fatalf("exit process: %s", err.Error())
}
go func() {
err := startServer()
log.Fatalf("exit process: %s", err.Error())
}()
err := startClient()
log.Fatalf("exit process: %s", err.Error())
}
func startClient() error {
c, err := client.New()
if err != nil {
return err
}
if err := c.Open(); err != nil {
return err
}
c.DeletedTweet = make(chan *client.Tweet)
c.StreamingError = make(chan error)
db := database.Default()
for {
select {
case tweet := <-c.DeletedTweet:
model, err := tweet.Model()
if err != nil {
log.Printf("error: %+v\n", err)
continue
}
db.Create(model)
if tweet.Tweet == nil {
log.Printf("tweet from %d\n", tweet.UserID)
continue
}
user := models.User{}
userModel, err := tweet.UserModel()
if err != nil {
log.Printf("error: %+v\n", err)
continue
}
db.Where(models.User{
ID: tweet.UserID,
}).Assign(userModel).FirstOrCreate(&user)
db.Save(&user)
log.Printf("tweet from @%s: %s\n", tweet.User.ScreenName, tweet.Text)
case err := <-c.StreamingError:
log.Printf("error: %+v\n", err)
}
}
return nil
}
func startServer() error {
s := server.New()
return s.Run()
}