bilibili-backup/app/service/main/ugcpay/dao/grpc.go
2019-04-22 02:59:20 +00:00

33 lines
576 B
Go

package dao
import (
"context"
arc "go-common/app/service/main/archive/api"
"go-common/library/ecode"
)
// ArchiveUGCPay get archive ugcpay flag.
func (d *Dao) ArchiveUGCPay(ctx context.Context, aid int64) (pay bool, err error) {
var (
req = &arc.ArcRequest{
Aid: aid,
}
reply *arc.ArcReply
)
if reply, err = d.archiveAPI.Arc(ctx, req); err != nil {
if err == ecode.NothingFound {
err = nil
pay = false
return
}
return
}
if reply != nil && reply.Arc != nil && reply.Arc.Rights.UGCPay == 1 {
pay = true
} else {
pay = false
}
return
}