/
blaze.go
43 lines (38 loc) · 1.09 KB
/
blaze.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
package main
import (
"context"
"database/sql"
"log"
"time"
bot "github.com/MixinNetwork/bot-api-go-client"
"github.com/crossle/hacker-news-mixin-bot/config"
"github.com/crossle/hacker-news-mixin-bot/durable"
"github.com/crossle/hacker-news-mixin-bot/session"
)
func StartBlaze(db *sql.DB) error {
log.Println("start blaze")
logger, err := durable.NewLoggerClient("", true)
if err != nil {
return err
}
defer logger.Close()
ctx, cancel := newBlazeContext(db, logger)
defer cancel()
for {
blazeClient := bot.NewBlazeClient(config.MixinClientId, config.MixinSessionId, config.MixinPrivateKey)
r := &ResponseMessage{
client: blazeClient,
}
if err := blazeClient.Loop(ctx, r); err != nil {
session.Logger(ctx).Error(err)
}
session.Logger(ctx).Info("connection loop end")
time.Sleep(time.Second)
}
return nil
}
func newBlazeContext(db *sql.DB, client *durable.LoggerClient) (context.Context, context.CancelFunc) {
ctx := session.WithLogger(context.Background(), durable.BuildLogger(client, "blaze", nil))
ctx = session.WithDatabase(ctx, db)
return context.WithCancel(ctx)
}