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.
79 lines
1.7 KiB
79 lines
1.7 KiB
package archive |
|
|
|
import ( |
|
"context" |
|
|
|
"go-common/app/interface/main/app-show/conf" |
|
"go-common/app/service/main/archive/api" |
|
arcrpc "go-common/app/service/main/archive/api/gorpc" |
|
"go-common/app/service/main/archive/model/archive" |
|
"go-common/library/log" |
|
) |
|
|
|
// Dao is archive dao. |
|
type Dao struct { |
|
c *conf.Config |
|
// rpc |
|
arcRpc *arcrpc.Service2 |
|
} |
|
|
|
// New new a archive dao. |
|
func New(c *conf.Config) (d *Dao) { |
|
d = &Dao{ |
|
c: c, |
|
// rpc |
|
arcRpc: arcrpc.New2(c.ArchiveRPC), |
|
} |
|
return |
|
} |
|
|
|
// Archive get archive by aid. |
|
func (d *Dao) Archive(ctx context.Context, aid int64) (a *api.Arc, err error) { |
|
arg := &archive.ArgAid2{Aid: aid} |
|
if a, err = d.arcRpc.Archive3(ctx, arg); err != nil { |
|
log.Error("d.arcRpc.Archive3(%v) error(%v)", arg, err) |
|
return |
|
} |
|
return |
|
} |
|
|
|
// ArchivesPB multi get archives. |
|
func (d *Dao) ArchivesPB(ctx context.Context, aids []int64) (as map[int64]*api.Arc, err error) { |
|
arg := &archive.ArgAids2{Aids: aids} |
|
return d.arcRpc.Archives3(ctx, arg) |
|
} |
|
|
|
// RanksArcs |
|
func (d *Dao) RanksArcs(ctx context.Context, rid, pn, ps int) (res []*api.Arc, aids []int64, err error) { |
|
arg := &archive.ArgRank2{ |
|
Rid: int16(rid), |
|
Pn: pn, |
|
Ps: ps, |
|
} |
|
var as *archive.RankArchives3 |
|
if as, err = d.arcRpc.RankArcs3(ctx, arg); err != nil { |
|
log.Error("d.arcRpc.RankArcs3(%v) error(%v)", arg, err) |
|
return |
|
} |
|
if as != nil { |
|
res = as.Archives |
|
for _, a := range res { |
|
aids = append(aids, a.Aid) |
|
} |
|
} |
|
return |
|
} |
|
|
|
// RankTopArcs |
|
func (d *Dao) RankTopArcs(ctx context.Context, rid, pn, ps int) (res []*api.Arc, err error) { |
|
arg := &archive.ArgRankTop2{ |
|
ReID: int16(rid), |
|
Pn: pn, |
|
Ps: ps, |
|
} |
|
if res, err = d.arcRpc.RankTopArcs3(ctx, arg); err != nil { |
|
log.Error("d.arcRpc.RankTopArcs3(%v) error(%v)", arg, err) |
|
return |
|
} |
|
return |
|
}
|
|
|