Prechádzať zdrojové kódy

增加研究员关注状态

kobe6258 4 mesiacov pred
rodič
commit
c28ec4e448

+ 11 - 8
models/user/user_analyst_follow_list.go

@@ -178,14 +178,17 @@ func GetFollowed(userId int, analystId int) string {
 	return string(record.Followed)
 }
 func GetFollowing(userId int, analystId int) string {
-	db := models.Main()
-	var dbFollow UserAnalystFollowList
-	err := db.Model(&UserAnalystFollowList{}).Where("user_id = ? and financial_analyst_id = ?", userId, analystId).First(&dbFollow).Error
-	if err != nil {
-		if !errors.Is(err, gorm.ErrRecordNotFound) {
-			logger.Error("查询用户关注状态失败:%v", err)
+	if userId > 0 {
+		db := models.Main()
+		var dbFollow UserAnalystFollowList
+		err := db.Model(&UserAnalystFollowList{}).Where("user_id = ? and financial_analyst_id = ?", userId, analystId).First(&dbFollow).Error
+		if err != nil {
+			if !errors.Is(err, gorm.ErrRecordNotFound) {
+				logger.Error("查询用户关注状态失败:%v", err)
+			}
+			return string(Unfollowed)
 		}
-		return string(Unfollowed)
+		return string(dbFollow.Followed)
 	}
-	return string(dbFollow.Followed)
+	return string(Unfollowed)
 }

+ 2 - 2
service/product/product_service.go

@@ -78,7 +78,7 @@ func GetProductRiskLevel(product merchantService.MerchantProductDTO) (riskLevel,
 			riskLevel = fmt.Sprintf("R%d", riskNum)
 		}
 	case "report":
-		report, reportErr := reportDomain.GetReportById(product.SourceId)
+		report, reportErr := reportDomain.GetReportById(product.SourceId, 0)
 		var coverSrcId int
 		sourceAbsract, sourceTitle, coverSrcId = report.Abstract, report.Title, report.CoverSrc
 		if coverSrcId > 0 {
@@ -202,7 +202,7 @@ func GetRelatePackage(info ProductDTO) (prodList []ProductDTO, err error) {
 		}
 		return
 	case "report":
-		reportInfo, reportErr := reportDomain.GetReportById(info.SourceId)
+		reportInfo, reportErr := reportDomain.GetReportById(info.SourceId, 0)
 		if reportErr != nil {
 			logger.Error("获取报告信息失败:%v", err)
 			return

+ 2 - 2
service/report/report_service.go

@@ -153,7 +153,7 @@ func matchRiskLevel(userId int, report reportService.ReportDTO) (riskLevelMatch
 
 func GetReportById(reportId int, login bool, userId int) (report *reportService.ReportDTO, err error) {
 	var reportInfo reportService.ReportDTO
-	reportInfo, err = reportService.GetReportById(reportId)
+	reportInfo, err = reportService.GetReportById(reportId, userId)
 	if err != nil {
 		logger.Error("获取研报失败:%v,研报ID:%d", err, report.ReportID)
 		err = exception.NewWithException(exception.GetReportFailed, err.Error())
@@ -427,7 +427,7 @@ func GetReportPageByAnalyst(pageInfo page.PageInfo, analyst string, reportIds []
 	return
 }
 func CountReport(count RecordCount) (traceId string, err error) {
-	report, err := reportService.GetReportById(count.ReportId)
+	report, err := reportService.GetReportById(count.ReportId, 0)
 	if err != nil {
 		err = exception.New(exception.GetReportFailed)
 		return