You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
92 lines
2.0 KiB
92 lines
2.0 KiB
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 |
|
}
|
|
|