mirror of https://github.com/rocboss/paopao-ce
Michael Li
1 year ago
parent
bc0530efc7
commit
199f77e70c
@ -0,0 +1,87 @@
|
|||||||
|
// Copyright 2023 ROC. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package dbr
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/sirupsen/logrus"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
"gorm.io/gorm/clause"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Following struct {
|
||||||
|
*Model
|
||||||
|
User *User `json:"-" gorm:"foreignKey:ID;references:FollowId"`
|
||||||
|
UserId int64 `json:"user_id"`
|
||||||
|
FollowId int64 `json:"friend_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Following) GetFollowing(db *gorm.DB, userId, followId int64) (*Following, error) {
|
||||||
|
var following Following
|
||||||
|
err := db.Omit("User").Unscoped().Where("user_id = ? AND follow_id = ?", userId, followId).First(&following).Error
|
||||||
|
if err != nil {
|
||||||
|
logrus.Debugf("Following.GetFollowing get following error:%s", err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &following, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Following) DelFollowing(db *gorm.DB, userId, followId int64) error {
|
||||||
|
return db.Omit("User").Unscoped().Where("user_id = ? AND follow_id = ?", userId, followId).Delete(f).Error
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Following) ListFollows(db *gorm.DB, userId int64, limit int, offset int) (res []*Following, total int64, err error) {
|
||||||
|
db = db.Model(f).Where("user_id=?", userId)
|
||||||
|
if err = db.Count(&total).Error; err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if offset >= 0 && limit > 0 {
|
||||||
|
db = db.Offset(offset).Limit(limit)
|
||||||
|
}
|
||||||
|
db.Joins("User").Order(clause.OrderByColumn{Column: clause.Column{Table: "User", Name: "nickname"}, Desc: false})
|
||||||
|
if err = db.Find(&res).Error; err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Following) ListFollowingIds(db *gorm.DB, userId int64, limit, offset int) (ids []int64, total int64, err error) {
|
||||||
|
db = db.Model(f).Where("follow_id=?", userId)
|
||||||
|
if err = db.Count(&total).Error; err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if offset >= 0 && limit > 0 {
|
||||||
|
db = db.Offset(offset).Limit(limit)
|
||||||
|
}
|
||||||
|
if err = db.Omit("User").Select("user_id").Find(&ids).Error; err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Following) FollowCount(db *gorm.DB, userId int64) (follows int64, followings int64, err error) {
|
||||||
|
if err = db.Model(f).Where("user_id=?", userId).Count(&follows).Error; err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err = db.Model(f).Where("follow_id=?", userId).Count(&followings).Error; err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Following) IsFollow(db *gorm.DB, userId int64, followId int64) bool {
|
||||||
|
if _, err := s.GetFollowing(db, userId, followId); err == nil {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Following) Create(db *gorm.DB) (*Following, error) {
|
||||||
|
err := db.Omit("User").Create(f).Error
|
||||||
|
return f, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Following) UpdateInUnscoped(db *gorm.DB) error {
|
||||||
|
return db.Unscoped().Omit("User").Save(c).Error
|
||||||
|
}
|
||||||
@ -0,0 +1,104 @@
|
|||||||
|
// Copyright 2023 ROC. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package jinzhu
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/rocboss/paopao-ce/internal/core"
|
||||||
|
"github.com/rocboss/paopao-ce/internal/core/ms"
|
||||||
|
"github.com/rocboss/paopao-ce/internal/dao/jinzhu/dbr"
|
||||||
|
"github.com/sirupsen/logrus"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
_ core.FollowingManageService = (*followingManageSrv)(nil)
|
||||||
|
)
|
||||||
|
|
||||||
|
type followingManageSrv struct {
|
||||||
|
db *gorm.DB
|
||||||
|
f *dbr.Following
|
||||||
|
u *dbr.User
|
||||||
|
}
|
||||||
|
|
||||||
|
func newFollowingManageService(db *gorm.DB) core.FollowingManageService {
|
||||||
|
return &followingManageSrv{
|
||||||
|
db: db,
|
||||||
|
f: &dbr.Following{},
|
||||||
|
u: &dbr.User{},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *followingManageSrv) FollowUser(userId int64, followId int64) error {
|
||||||
|
if _, err := s.f.GetFollowing(s.db, userId, followId); err != nil {
|
||||||
|
following := &dbr.Following{
|
||||||
|
UserId: userId,
|
||||||
|
FollowId: followId,
|
||||||
|
}
|
||||||
|
if _, err = following.Create(s.db); err != nil {
|
||||||
|
logrus.Errorf("contactManageSrv.fetchOrNewContact create new contact err:%s", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *followingManageSrv) UnfollowUser(userId int64, followId int64) error {
|
||||||
|
return s.f.DelFollowing(s.db, userId, followId)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *followingManageSrv) ListFollows(userId int64, limit, offset int) (*ms.ContactList, error) {
|
||||||
|
follows, totoal, err := s.f.ListFollows(s.db, userId, limit, offset)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
res := &ms.ContactList{
|
||||||
|
Total: totoal,
|
||||||
|
}
|
||||||
|
for _, f := range follows {
|
||||||
|
res.Contacts = append(res.Contacts, ms.ContactItem{
|
||||||
|
UserId: f.User.ID,
|
||||||
|
Username: f.User.Username,
|
||||||
|
Nickname: f.User.Nickname,
|
||||||
|
Avatar: f.User.Avatar,
|
||||||
|
IsFollow: true,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return res, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *followingManageSrv) ListFollowings(userId int64, limit, offset int) (*ms.ContactList, error) {
|
||||||
|
followingIds, totoal, err := s.f.ListFollowingIds(s.db, userId, limit, offset)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
followings, err := s.u.ListUserInfoById(s.db, followingIds)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
res := &ms.ContactList{
|
||||||
|
Total: totoal,
|
||||||
|
}
|
||||||
|
for _, user := range followings {
|
||||||
|
res.Contacts = append(res.Contacts, ms.ContactItem{
|
||||||
|
UserId: user.ID,
|
||||||
|
Username: user.Username,
|
||||||
|
Nickname: user.Nickname,
|
||||||
|
Avatar: user.Avatar,
|
||||||
|
IsFollow: s.IsFollow(userId, user.ID),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return res, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *followingManageSrv) GetFollowCount(userId int64) (int64, int64, error) {
|
||||||
|
return s.f.FollowCount(s.db, userId)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *followingManageSrv) IsFollow(userId int64, followId int64) bool {
|
||||||
|
if _, err := s.f.GetFollowing(s.db, userId, followId); err == nil {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
@ -0,0 +1 @@
|
|||||||
|
DROP TABLE IF EXISTS `p_following`;
|
||||||
@ -0,0 +1,11 @@
|
|||||||
|
CREATE TABLE `p_following` (
|
||||||
|
`id` bigint unsigned NOT NULL AUTO_INCREMENT,
|
||||||
|
`user_id` bigint unsigned NOT NULL,
|
||||||
|
`follow_id` bigint unsigned NOT NULL,
|
||||||
|
`is_del` tinyint NOT NULL DEFAULT 0, -- 是否删除, 0否, 1是
|
||||||
|
`created_on` bigint unsigned NOT NULL DEFAULT '0',
|
||||||
|
`modified_on` bigint unsigned NOT NULL DEFAULT '0',
|
||||||
|
`deleted_on` bigint unsigned NOT NULL DEFAULT '0',
|
||||||
|
PRIMARY KEY (`id`) USING BTREE,
|
||||||
|
KEY `idx_following_user_follow` (`user_id`,`follow_id`) USING BTREE
|
||||||
|
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;
|
||||||
@ -0,0 +1 @@
|
|||||||
|
DROP TABLE IF EXISTS p_following;
|
||||||
@ -0,0 +1,10 @@
|
|||||||
|
CREATE TABLE p_following (
|
||||||
|
id BIGSERIAL PRIMARY KEY,
|
||||||
|
user_id BIGINT NOT NULL,
|
||||||
|
follow_id BIGINT NOT NULL,
|
||||||
|
is_del SMALLINT NOT NULL DEFAULT 0, -- 是否删除, 0否, 1是
|
||||||
|
created_on BIGINT NOT NULL DEFAULT 0,
|
||||||
|
modified_on BIGINT NOT NULL DEFAULT 0,
|
||||||
|
deleted_on BIGINT NOT NULL DEFAULT 0
|
||||||
|
);
|
||||||
|
CREATE INDEX idx_following_user_follow ON p_following USING btree (user_id, follow_id);
|
||||||
@ -0,0 +1 @@
|
|||||||
|
DROP TABLE IF EXISTS "p_following";
|
||||||
@ -0,0 +1,15 @@
|
|||||||
|
CREATE TABLE "p_following" (
|
||||||
|
"id" integer NOT NULL,
|
||||||
|
"user_id" integer NOT NULL,
|
||||||
|
"follow_id" integer NOT NULL,
|
||||||
|
"is_del" integer NOT NULL,
|
||||||
|
"created_on" integer NOT NULL,
|
||||||
|
"modified_on" integer NOT NULL,
|
||||||
|
"deleted_on" integer NOT NULL,
|
||||||
|
PRIMARY KEY ("id")
|
||||||
|
);
|
||||||
|
CREATE INDEX "idx_following_user_follow"
|
||||||
|
ON "p_following" (
|
||||||
|
"user_id" ASC,
|
||||||
|
"follow_id" ASC
|
||||||
|
);
|
||||||
Loading…
Reference in new issue