forked from ShaneMaglangit/ronin-axie-alert-demo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
firebase.go
71 lines (63 loc) · 1.59 KB
/
firebase.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
package firebase
import (
"context"
firebase "firebase.google.com/go"
"firebase.google.com/go/db"
"google.golang.org/api/option"
"log"
)
type Firebase struct {
db *db.Client
}
const RowCap = 1000000
func New() *Firebase {
conf := firebase.Config{DatabaseURL: "..."}
opt := option.WithCredentialsFile("key.json")
app, err := firebase.NewApp(context.Background(), &conf, opt)
if err != nil {
log.Fatal(err)
}
db, err := app.Database(context.Background())
if err != nil {
log.Fatal(err)
}
return &Firebase{db: db}
}
func (fb *Firebase) SaveSalesMultiple(sales []Sale) {
salesRef := fb.db.NewRef("sales")
for _, sale := range sales {
err := salesRef.Child(sale.TxHash).Set(context.Background(), sale)
if err != nil {
log.Fatal(err)
}
}
//fb.LimitSalesRecord()
}
func (fb *Firebase) LimitSalesRecord() {
var hashes map[string]bool
salesRef := fb.db.NewRef("sales")
_ = salesRef.GetShallow(context.Background(), &hashes)
if len(hashes) > RowCap {
fb.ReduceExcessSalesRecord(len(hashes) - RowCap)
}
}
func (fb *Firebase) ReduceExcessSalesRecord(n int) {
salesRef := fb.db.NewRef("sales")
res, err := salesRef.OrderByChild("timestamp").GetOrdered(context.Background())
if err != nil {
log.Fatal(err)
}
for i := 0; i < n; i++ {
var sale Sale
if err := res[i].Unmarshal(&sale); err != nil {
log.Fatalln("Error unmarshalling result:", err)
}
fb.DeleteSaleRecord(sale.TxHash)
}
}
func (fb *Firebase) DeleteSaleRecord(id string) {
err := fb.db.NewRef("sales").Child(id).Delete(context.Background())
if err != nil {
log.Fatalln("Error while deleting excess", err)
}
}