eta_mini_ht_api/controllers/report/report_controller.go

package report

import (
	"encoding/json"
	logger "eta/eta_mini_ht_api/common/component/log"
	"eta/eta_mini_ht_api/common/exception"
	"eta/eta_mini_ht_api/common/utils/page"
	"eta/eta_mini_ht_api/controllers"
	reportService "eta/eta_mini_ht_api/domian/report"
	"eta/eta_mini_ht_api/service/report"
	"eta/eta_mini_ht_api/service/user"
)

type ReportController struct {
	controllers.ListController
}

// Search 搜索报告列表
// @Description 搜索报告列表
// @Success 200 {object}
// @router /search [get]
func (r *ReportController) Search(key string) {
	controllers.Wrap(&r.BaseController, func() (result *controllers.WrapData, err error) {
		result = r.InitWrapData("分页搜索报告列表失败")
		if key == "" {
			err = exception.New(exception.SearchKeyEmptyError)
			r.FailedResult("分页搜索报告列表失败", result)
			return
		}
		detailType := r.Data["detailType"].(string)
		userInfo := r.Data["user"].(user.User)
		pageRes := page.Page{
			Current:  r.PageInfo.Current,
			PageSize: r.PageInfo.PageSize,
		}
		//获取当前可以被搜索的报告原始ID
		var orgIds map[string][]int
		//先要限制查询的id范围
		pageRes.Total, pageRes.LatestId, orgIds = report.RangeSearch(isLogin(detailType), userInfo.Id)
		reportIds, err := report.GetReportByIdListByOrgIds(orgIds)
		if err != nil {
			logger.Error("获取报告原始ID列表失败:%v", err)
			r.FailedResult("分页搜索报告列表失败", result)
			return
		}
		if len(reportIds) == 0 {
			reports := new(page.PageResult)
			reports.Data = []reportService.ReportDTO{}
			reports.Page = pageRes
			logger.Info("没有可以查询的报告列表")
			r.SuccessResult("分页搜索报告列表成功", reports, result)
			return
		}
		if r.PageInfo.LatestId == 0 {
			//pageRes.Total, pageRes.LatestId = report.SearchMaxReportId(key)
			r.PageInfo.LatestId = pageRes.LatestId
			r.PageInfo.Total = pageRes.Total
		} else {
			pageRes.LatestId = r.PageInfo.LatestId
			pageRes.Total = r.PageInfo.Total
		}
		pageRes.TotalPage = page.TotalPages(pageRes.Total, pageRes.PageSize)
		list := make([]reportService.ReportDTO, 0)
		if pageRes.LatestId > 0 {
			//订阅 TODO
			list, err = report.SearchReportList(key, reportIds, r.PageInfo, isLogin(detailType), userInfo.Id)
			if err != nil {
				r.FailedResult("分页搜索报告列表失败", result)
				return
			}
		}
		reports := new(page.PageResult)
		reports.Data = list
		reports.Page = pageRes
		r.SuccessResult("分页搜索报告列表成功", reports, result)
		return
	})
}

// List 获取报告列表
// @Description 获取报告列表
// @Success 200 {object}
// @router /list [get]
func (r *ReportController) List(permissionIds string) {
	controllers.Wrap(&r.BaseController, func() (result *controllers.WrapData, err error) {
		result = r.InitWrapData("分页查询报告列表失败")
		pageRes := page.Page{
			Current:  r.PageInfo.Current,
			PageSize: r.PageInfo.PageSize,
		}
		detailType := r.Data["detailType"].(string)
		userInfo := r.Data["user"].(user.User)
		permissionIdList, err := r.TransPermissionIds(permissionIds)
		if err != nil {
			logger.Error("品种列表解析错误:%v", err)
			r.FailedResult("分页查询报告列表失败", result)
			err = exception.New(exception.QueryReportPageFailed)
			return
		}
		var reportOrgIds map[string][]int
		pageRes.Total, pageRes.LatestId, reportOrgIds = report.GetTotalPageCountByPermissionIds(permissionIdList, isLogin(detailType), userInfo.Id)
		if err != nil {
			logger.Error("分页查询报告列表失败:%v", err)
			r.FailedResult("分页查询报告列表失败", result)
			return
		}
		if r.PageInfo.LatestId == 0 {
			r.PageInfo.LatestId = pageRes.LatestId
		} else {
			pageRes.LatestId = r.PageInfo.LatestId
			pageRes.Total = r.PageInfo.Total
		}
		pageRes.TotalPage = page.TotalPages(pageRes.Total, pageRes.PageSize)
		if len(reportOrgIds) == 0 {
			logger.Info("没有搜索到相关品种的报告，返回空列表")
			reports := new(page.PageResult)
			reports.Data = []reportService.ReportDTO{}
			reports.Page = pageRes
			r.SuccessResult("查询报告列表成功", reports, result)
			return
		}
		list, err := report.GetReportPage(r.PageInfo, reportOrgIds, false, isLogin(detailType), userInfo.Id)
		if err != nil {
			r.FailedResult("分页查询报告列表失败", result)
			return
		}
		reports := new(page.PageResult)
		reports.Data = list
		reports.Page = pageRes
		r.SuccessResult("查询报告列表成功", reports, result)
		return
	})
}
func isLogin(listType string) bool {
	if listType == "logout" {
		logger.Info("当前用户未登录,展示部分详情")
		return false
	}
	if listType == "login" {
		return true
	}
	return false
}

// HotRanked @Title 获取本周最热的报告列表
// @Description 获取本周最热的报告列表
// @Success 200 {object}
// @router /hotRankedList [get]
func (r *ReportController) HotRanked(permissionIds string, limit int) {
	controllers.Wrap(&r.BaseController, func() (result *controllers.WrapData, err error) {
		if limit <= 0 {
			limit = 3
		}
		result = r.InitWrapData("获取本周最热报告列表失败")
		detailType := r.Data["detailType"].(string)
		userInfo := r.Data["user"].(user.User)
		//
		permissionIdsWithRisk, pdRiskLevel, err := report.RangePermissionIds(isLogin(detailType), userInfo.Id)
		if err != nil {
			logger.Error("获取带有风险等级的品种列表错误:%v", err)
			r.FailedResult("获取最热报告列表失败", result)
			err = exception.New(exception.GetHotRandListFailed)
			return
		}
		if len(permissionIdsWithRisk) == 0 {
			r.SuccessResult("获取本周最热报告列表成功", []report.HotRankedReport{}, result)
			return
		}
		list, err := report.GetRandedReportByWeeklyHot(limit, isLogin(detailType), userInfo.Id, pdRiskLevel)
		//二级品种
		permissionIdList, err := r.TransPermissionIds(permissionIds)
		if err != nil {
			logger.Error("品种列表解析错误:%v", err)
			r.FailedResult("获取最热报告列表失败", result)
			err = exception.New(exception.GetHotRandListFailed)
			return
		}
		filterList := make([]report.HotRankedReport, 0)
		if len(permissionIdList) > 0 {
			//通过有权限的品种过滤查询品种
			var filterPermissionIds []int
			for _, permissionId := range permissionIdList {
				for _, permissionIdWIthRisk := range permissionIdsWithRisk {
					if permissionId == permissionIdWIthRisk {
						filterPermissionIds = append(filterPermissionIds, permissionId)
					}
				}
			}
			for _, item := range list {
				for _, permissionId := range filterPermissionIds {
					if _, ok := item.SecondPermissions[permissionId]; ok {
						filterList = append(filterList, item)
					}
				}
			}
		} else {
			for _, item := range list {
				for _, permissionId := range permissionIdsWithRisk {
					if _, ok := item.SecondPermissions[permissionId]; ok {
						filterList = append(filterList, item)
					}
				}
			}
		}
		if err != nil {
			r.FailedResult("获取本周最热报告列表成功", result)
			return
		}
		r.SuccessResult("获取本周最热报告列表成功", filterList, result)
		return
	})
}

// PublishRanked @Title 获取最新发布的报告列表
// @Description 获取最新发布的报告列表
// @Success 200 {object}
// @router /publishRankedList [get]
func (r *ReportController) PublishRanked(permissionIds string, limit int, week bool) {
	controllers.Wrap(&r.BaseController, func() (result *controllers.WrapData, err error) {
		result = r.InitWrapData("获取最新发布报告列表失败")
		if limit <= 0 {
			limit = 3
		}
		detailType := r.Data["detailType"].(string)
		userInfo := r.Data["user"].(user.User)
		//
		permissionIdsWithRisk, pdRiskLevel, err := report.RangePermissionIds(isLogin(detailType), userInfo.Id)
		if err != nil {
			logger.Error("获取带有风险等级的品种列表错误:%v", err)
			r.FailedResult("获取最新发布报告列表失败", result)
			err = exception.New(exception.GetPublishedRandListFailed)
			return
		}
		if len(permissionIdsWithRisk) == 0 {
			r.SuccessResult("获取最新发布报告列表成功", []report.HotRankedReport{}, result)
			return
		}
		//获取最新的报告列表
		list, err := report.GetRandedReportByPublishTimeWeekly(limit, week, isLogin(detailType), userInfo.Id, pdRiskLevel)
		if err != nil {
			r.FailedResult("获取最新发布报告列表失败", result)
			return
		}
		//二级品种
		permissionIdList, err := r.TransPermissionIds(permissionIds)
		if err != nil {
			logger.Error("品种列表解析错误:%v", err)
			r.FailedResult("获取最新发布报告列表失败", result)
			err = exception.New(exception.GetPublishedRandListFailed)
			return
		}

		filterList := make([]report.PublishRankedReport, 0)
		if len(permissionIdList) > 0 {
			//通过有权限的品种过滤查询品种
			var filterPermissionIds []int
			for _, permissionId := range permissionIdList {
				for _, permissionIdWIthRisk := range permissionIdsWithRisk {
					if permissionId == permissionIdWIthRisk {
						filterPermissionIds = append(filterPermissionIds, permissionId)
					}
				}
			}
			for _, item := range list {
				for _, permissionId := range filterPermissionIds {
					if _, ok := item.SecondPermissions[permissionId]; ok {
						filterList = append(filterList, item)
					}
				}
			}
		} else {
			for _, item := range list {
				for _, permissionId := range permissionIdsWithRisk {
					if _, ok := item.SecondPermissions[permissionId]; ok {
						filterList = append(filterList, item)
					}
				}
			}
		}
		r.SuccessResult("获取最新发布报告列表成功", filterList, result)
		return
	})
}

// GetPermissions  @Title 获取品种列表
// @Description获取品种列表
// @Success 200 {object}
// @router /permissionList [get]
func (r *ReportController) GetPermissions() {
	controllers.Wrap(&r.BaseController, func() (result *controllers.WrapData, err error) {
		result = r.InitWrapData("获取品种列表失败")
		//获取最新的报告列表
		json, err := report.GetPermissionList()
		if err != nil {
			r.FailedResult("获取品种列表失败", result)
			return
		}
		r.SuccessResult("获取最新发布报告列表成功", json.Children, result)
		return
	})
}

type RecordCountReq struct {
	ReportId   int         `json:"reportId"`
	TraceId    string      `json:"traceId"`
	IpAddress  string      `json:"ipAddress"`
	Location   string      `json:"location"`
	Referer    string      `json:"referer"`
	Additional interface{} `json:"additional"`
}

// Count 获取品种列表
// @Description 获取最新发布的报告列表
// @Success 200 {object}
// @router /count [post]
func (r *ReportController) Count() {
	controllers.Wrap(&r.BaseController, func() (result *controllers.WrapData, err error) {
		result = r.InitWrapData("研报点击记录失败")
		var userInfo user.User
		userInfo = r.Data["user"].(user.User)
		recordReq := new(RecordCountReq)
		r.GetPostParams(recordReq)
		if recordReq.ReportId <= 0 {
			logger.Error("研报点击记录失败,reportId不能为空")
			r.FailedResult("研报点击记录失败", result)
			err = exception.New(exception.ReportRecordClickCountFailed)
			return
		}
		record := convertToRecordCount(recordReq)
		record.UserId = userInfo.Id
		record.Mobile = userInfo.Mobile
		traceId, err := report.CountReport(record)
		if err != nil {
			r.FailedResult("研报点击记录失败", result)
			err = exception.New(exception.ReportRecordClickCountFailed)
			return
		}
		countResp := struct {
			TraceId string `json:"traceId"`
		}{TraceId: traceId}
		r.SuccessResult("研报点击记录成功", countResp, result)
		return
	})
}

// GetReport 获取研报详情
// @Description 获取研报详情
// @Success 200 {object}
// @router /report [get]
func (r *ReportController) GetReport(reportId int) {
	controllers.Wrap(&r.BaseController, func() (result *controllers.WrapData, err error) {
		result = r.InitWrapData("获取研报详情失败")
		userInfo := r.Data["user"].(user.User)
		detailType := r.Data["detailType"].(string)
		if !isLogin(detailType) {
			logger.Info("当前用户未登录,展示部分详情")
		}
		reportDetail, err := report.GetReportById(reportId, isLogin(detailType), userInfo.Id)
		if err != nil {
			r.FailedResult("获取研报详情失败", result)
			err = exception.New(exception.GetReportFailed)
			return
		}
		r.SuccessResult("获取研报详情成功", reportDetail, result)
		return
	})
}
func convertToRecordCount(req *RecordCountReq) report.RecordCount {
	additionStr, _ := json.Marshal(req.Additional)
	return report.RecordCount{
		ReportId:   req.ReportId,
		TraceId:    req.TraceId,
		IpAddress:  req.IpAddress,
		Location:   req.Location,
		Referer:    req.Referer,
		Additional: string(additionStr),
	}
}