diff --git a/internal/app/api/idl/bilibili_picture.go b/internal/app/api/idl/bilibili_picture.go index 6f591b3..759713a 100644 --- a/internal/app/api/idl/bilibili_picture.go +++ b/internal/app/api/idl/bilibili_picture.go @@ -130,6 +130,7 @@ type BilibiliRandomPicture struct { ImgSrc string `json:"img_src"` DynamicID uint64 `json:"dynamic_id"` Tags []BilibiliRandomPictureTag `json:"tags"` + Verify bool `json:"verify"` } type BilibiliRandomPictureTag struct { TagID uint64 `json:"tag_id"` diff --git a/internal/app/api/service/bilbil_picture.go b/internal/app/api/service/bilbil_picture.go index a1c7955..8f93da8 100644 --- a/internal/app/api/service/bilbil_picture.go +++ b/internal/app/api/service/bilbil_picture.go @@ -97,6 +97,7 @@ func (c *RandomPicsCache) flush() { ImgSrc: list[i].Pictures[j].ImgSrc, DynamicID: list[i].DynamicID, Tags: tags, + Verify: list[i].Verify, }) } } @@ -202,6 +203,7 @@ func (service *BilbilPicture) Random(ctx context.Context) (*idl.BilibiliPictureR ImgSrc: list[i].Pictures[j].ImgSrc, DynamicID: list[i].DynamicID, Tags: tags, + Verify: list[i].Verify, }) } } diff --git a/internal/repository/bilibili_picture.go b/internal/repository/bilibili_picture.go index 3525a50..e8fc036 100644 --- a/internal/repository/bilibili_picture.go +++ b/internal/repository/bilibili_picture.go @@ -104,7 +104,7 @@ func (impl *BilibiliPictureMysqlImpl) Random(rand float64) (list []*idl.Bilibili return nil, err } offset := math.Floor((pair.MaxId - pair.MinId) * rand) - err = conn.Where("id >= ?", uint64(pair.MinId+offset)).Select("dynamic_id,pictures,topic_details").Limit(5).Find(&list).Error + err = conn.Where("id >= ?", uint64(pair.MinId+offset)).Select("dynamic_id,pictures,topic_details,verify").Limit(5).Find(&list).Error if err != nil { return nil, err }