-
Notifications
You must be signed in to change notification settings - Fork 0
/
share.go
92 lines (86 loc) · 2 KB
/
share.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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
package dao
import (
"context"
"go-common/app/service/main/share/model"
"go-common/library/ecode"
"go-common/library/log"
"github.com/pkg/errors"
)
// Shares get shares
func (d *Dao) Shares(ctx context.Context, oids []int64, tp int) (shares map[int64]int64, err error) {
shares, err = d.SharesCache(ctx, oids, tp)
if err != nil {
log.Error("d.SharesCache(%v) tp(%d) error(%v)", oids, tp, err)
err = nil
shares = make(map[int64]int64, len(oids))
}
var missed []int64
for _, oid := range oids {
if _, ok := shares[oid]; !ok {
missed = append(missed, oid)
}
}
if len(missed) == 0 {
return
}
// 最大30个id,并且分了100张表,用in的优化空间也不大,暂时循环单个查
for _, oid := range missed {
cnt, err := d.ShareCount(ctx, oid, tp)
if err != nil {
continue
}
shares[oid] = cnt
}
return
}
// ShareCount get share from cache/db
func (d *Dao) ShareCount(ctx context.Context, oid int64, tp int) (count int64, err error) {
count, err = d.ShareCache(ctx, oid, tp)
if count != -1 && err == nil {
return
}
var share *model.Share
if share, err = d.Share(ctx, oid, tp); err != nil {
err = errors.WithStack(err)
return
}
count = 0
if share != nil {
count = share.Count
}
d.asyncCache.Save(func() {
if err = d.SetShareCache(context.Background(), oid, tp, count); err != nil {
log.Error("%+v", err)
return
}
})
return
}
// Add add share
func (d *Dao) Add(ctx context.Context, p *model.ShareParams) (shared int64, err error) {
var ok bool
if ok, err = d.AddShareMember(ctx, p); err != nil {
return
}
if !ok {
err = ecode.ShareAlreadyAdd
return
}
if err = d.AddShare(ctx, p.OID, p.TP); err != nil {
err = errors.WithStack(err)
return
}
var share *model.Share
if share, err = d.Share(ctx, p.OID, p.TP); err != nil {
err = errors.WithStack(err)
return
}
shared = share.Count
d.asyncCache.Save(func() {
if err = d.SetShareCache(context.Background(), p.OID, p.TP, shared); err != nil {
log.Error("%+v", err)
return
}
})
return
}