服务发现基础逻辑

repo/userFollow.go

@@ -0,0 +1,26 @@
+package repo
+
+import (
+	"gorm.io/gorm"
+	"work_cation/models"
+)
+
+type userFollowRepo struct{}
+
+var UserFollow = &userFollowRepo{}
+
+func (u *userFollowRepo) Follow(db *gorm.DB, user *models.UserFollows) error {
+	return db.Create(user).Error
+}
+
+func (u *userFollowRepo) GetUser(db *gorm.DB, uuid string) *models.UserFollows {
+	var user models.UserFollows
+	err := db.Model(&models.UserFollows{}).Where("id = ?", uuid).Find(&user).Error
+	if err != nil {
+		return nil
+	}
+	if user.ID == "" {
+		return nil
+	}
+	return &user
+}