123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916 |
- package report
- import (
- "encoding/json"
- "eta/eta_mini_ht_api/common/component/config"
- "eta/eta_mini_ht_api/common/component/es"
- logger "eta/eta_mini_ht_api/common/component/log"
- "eta/eta_mini_ht_api/common/contants"
- "eta/eta_mini_ht_api/common/utils/page"
- stringUtils "eta/eta_mini_ht_api/common/utils/string"
- configService "eta/eta_mini_ht_api/domian/config"
- analystService "eta/eta_mini_ht_api/domian/financial_analyst"
- userService "eta/eta_mini_ht_api/domian/user"
- "eta/eta_mini_ht_api/models"
- permissionDao "eta/eta_mini_ht_api/models/config"
- "eta/eta_mini_ht_api/models/eta"
- etaDao "eta/eta_mini_ht_api/models/eta"
- "eta/eta_mini_ht_api/models/ht"
- mediaDao "eta/eta_mini_ht_api/models/media"
- reportDao "eta/eta_mini_ht_api/models/report"
- userDao "eta/eta_mini_ht_api/models/user"
- "math/rand"
- "strconv"
- "strings"
- "time"
- )
- const (
- SourceETA = "ETA"
- SourceHT = "HT"
- DESC models.Order = "desc"
- ASC models.Order = "asc"
- ESColumn = "title"
- ESRangeColumn = "reportId"
- )
- var (
- sortField = []string{"_score:desc"}
- htConfig = config.GetConfig(contants.HT).(*config.HTBizConfig)
- )
- func elastic() *es.ESClient {
- return es.GetInstance()
- }
- // ESReport Report ES研报mapping
- type ESReport struct {
- ReportID int `json:"reportId"`
- OrgId int `json:"orgId"`
- Title string `json:"title"`
- Author string `json:"author"`
- Source reportDao.ReportSource `json:"source"`
- Abstract string `json:"abstract"`
- CoverSrc int `json:"coverSrc"`
- Status reportDao.ReportStatus `json:"status"`
- PublishedTime string `json:"publishedTime"`
- }
- type ReportDTO struct {
- ReportID int `json:"reportId"`
- OrgId int `json:"orgId"`
- Title string `json:"title"`
- Author string `json:"author"`
- AuthorInfo []Anthor `json:"authorInfo"`
- Source string `json:"source"`
- Abstract string `json:"abstract"`
- PublishedTime string `json:"publishedTime"`
- RiskLevel string `json:"riskLevel"`
- SecondPermission map[int]string `json:"-"`
- Permissions map[int]string `json:"-"`
- PermissionNames interface{} `json:"permissionNames"`
- Highlight []string `json:"highlight"`
- Detail json.RawMessage `json:"detail"`
- PdfUrl string `json:"pdfUrl"`
- CoverSrc int `json:"coverSrc"`
- CoverUrl string `json:"coverUrl"`
- Login bool `json:"login"`
- RiskLevelStatus string `json:"riskLevelStatus"`
- IsFree bool `json:"isFree"`
- IsSubscribe bool `json:"isSubscribe"`
- Price string `json:"price"`
- }
- type Detail struct {
- }
- type Anthor struct {
- Id int `json:"id"`
- Name string `json:"name"`
- HeadImgUrl string `json:"headImgUrl"`
- }
- func GetGetReportById(reportId int) (ReportDTO ReportDTO, err error) {
- report, err := reportDao.GetReportById(reportId)
- if err != nil {
- return
- }
- ReportDTO = convertReportDTO(report, true)
- authorStr := ReportDTO.Author
- authorNames := strings.Split(authorStr, ",")
- authorNames = stringUtils.RemoveEmptyStrings(authorNames)
- var authorList []Anthor
- if len(authorNames) > 0 {
- for _, name := range authorNames {
- var author analystService.FinancialAnalystDTO
- author, err = analystService.GetAnalystByName(name)
- var item Anthor
- if err != nil {
- item = Anthor{
- Id: 0,
- Name: name,
- HeadImgUrl: "",
- }
- } else {
- item = Anthor{
- Id: author.Id,
- Name: author.Name,
- HeadImgUrl: author.HeadImgUrl,
- }
- }
- authorList = append(authorList, item)
- }
- }
- ReportDTO.AuthorInfo = authorList
- return
- }
- func GetTotalPageCount() (total int64, latestId int64, err error) {
- return reportDao.GetTotalPageCount()
- }
- func GetTotalPageCountByAnalyst(analyst string, permissionIds []int) (total int64, latestId int64, ids []int) {
- ids, err := reportDao.GetReportsByAnalyst(analyst)
- if err != nil {
- logger.Error("查询研究研报告列表id失败:%v", err)
- return
- }
- //查询这些包含在列表中的权限的报告ids
- htOrgIds, err := GetHTReportIdsByPermissionIds(permissionIds)
- if err != nil {
- logger.Error("品种筛选ht报告id失败:%v", err)
- htOrgIds = []int{}
- }
- etaOrgIds, err := GetETAReportIdsByPermissionIds(permissionIds)
- if err != nil {
- logger.Error("品种筛选eta报告id失败:%v", err)
- etaOrgIds = []int{}
- }
- if len(etaOrgIds) == 0 && len(htOrgIds) == 0 {
- logger.Info("没有符合权限的研报")
- return
- }
- orgIds := make(map[string][]int, 2)
- if len(etaOrgIds) == 0 {
- orgIds["ETA"] = []int{}
- } else {
- orgIds["ETA"] = etaOrgIds
- }
- if len(htOrgIds) == 0 {
- orgIds["HT"] = []int{}
- } else {
- orgIds["HT"] = htOrgIds
- }
- permitReportIds, err := reportDao.GetReportIdListByOrgIds(orgIds)
- if err != nil {
- logger.Error("根据原始报告id获取报告id列表失败:%v", err)
- return
- }
- var filterReportIds []int
- for _, id := range ids {
- for _, permitReportId := range permitReportIds {
- if id == permitReportId {
- filterReportIds = append(filterReportIds, id)
- }
- }
- }
- if len(filterReportIds) == 0 {
- logger.Info("没有符合权限的研究员研报")
- return
- }
- ids = filterReportIds
- total = int64(len(filterReportIds))
- latestId = int64(findMax(filterReportIds))
- return
- }
- // findMaxWithError 函数用于找到整型数组中的最大值,并返回错误信息
- func findMax(nums []int) (max int) {
- if len(nums) == 0 {
- return 0
- }
- // 初始化最大值为数组的第一个元素
- max = nums[0]
- // 遍历数组,找到最大值
- for _, num := range nums {
- if num > max {
- max = num
- }
- }
- return
- }
- func SearchMaxReportId(key string) (total int64, reportId int64) {
- sort := []string{"reportId:desc"}
- request := matchAll(sort, key)
- //同步es
- re, err := elastic().Count(request)
- if err != nil {
- logger.Error("es搜索异常:%v", err)
- }
- count := re.Count
- total = int64(count)
- if total > 0 {
- request = match(key, 0, count, sort)
- re, err = elastic().Search(request)
- if err != nil {
- logger.Error("es搜索异常:%v", err)
- }
- hits := elastic().GetSource(re.Hits)
- data := hits[0].Source
- report := ReportDTO{}
- err = json.Unmarshal(data, &report)
- if err != nil {
- logger.Error("获取当前最大研报id失败:%v", err)
- return
- }
- reportId = int64(report.ReportID)
- }
- return
- }
- func SearchReportList(key string, ids []int, from int, size int, max int64) (reports []ReportDTO, err error) {
- //同步es
- var docIds []string
- for _, id := range ids {
- docIds = append(docIds, strconv.Itoa(id))
- }
- sorts := append(sortField, "publishedTime:desc")
- request := matchRangeByDocId(key, from, size, max, sorts, docIds)
- re, err := elastic().Search(request)
- if err != nil {
- logger.Error("es搜索异常:%v", err)
- }
- hits := elastic().GetSource(re.Hits)
- if len(hits) == 0 {
- reports = []ReportDTO{}
- return
- }
- for _, hit := range hits {
- var content map[string][]string
- err = json.Unmarshal(hit.Highlight, &content)
- report := ReportDTO{}
- err = json.Unmarshal(hit.Source, &report)
- if err != nil {
- logger.Error("解析研报数据失败:%v", err)
- continue
- }
- report.Highlight = content[ESColumn]
- report.Title = report.Highlight[0]
- report.PublishedTime = report.PublishedTime[:10]
- reports = append(reports, report)
- }
- return
- }
- func GetReportPageByAnalyst(pageInfo page.PageInfo, analyst string, reportIds []int) (list []ReportDTO, err error) {
- offset := page.StartIndex(pageInfo.Current, pageInfo.PageSize)
- reports, err := reportDao.GetReportPageByAnalyst(pageInfo.LatestId, pageInfo.PageSize, offset, analyst, reportIds)
- if err != nil {
- logger.Error("分页查询报告列表失败:%v", err)
- return
- }
- list = make([]ReportDTO, 0)
- if reports != nil {
- for _, report := range reports {
- dto := convertReportDTO(report, false)
- list = append(list, dto)
- }
- }
- return
- }
- func GetReportPageByOrgIds(pageInfo page.PageInfo, orgIds map[string][]int, searchAll bool) (list []ReportDTO, err error) {
- offset := page.StartIndex(pageInfo.Current, pageInfo.PageSize)
- reports, err := reportDao.GetReportPageByOrgIds(pageInfo.LatestId, pageInfo.PageSize, offset, orgIds, searchAll)
- if err != nil {
- logger.Error("分页查询报告列表失败:%v", err)
- return
- }
- list = make([]ReportDTO, 0)
- if reports != nil {
- for _, report := range reports {
- dto := convertReportDTO(report, false)
- list = append(list, dto)
- }
- }
- return
- }
- func GetNewReportByPublishTime(time time.Time) (reports []ReportDTO) {
- list := reportDao.GetNewReportByPublishTime(time)
- if list != nil {
- for _, report := range list {
- dto := convertReportDTO(report, false)
- reports = append(reports, dto)
- }
- }
- return
- }
- func GetReportPage(pageInfo page.PageInfo) (list []ReportDTO, err error) {
- offset := page.StartIndex(pageInfo.Current, pageInfo.PageSize)
- reports, err := reportDao.GetReportPage(pageInfo.LatestId, pageInfo.PageSize, offset)
- if err != nil {
- logger.Error("分页查询报告列表失败:%v", err)
- return
- }
- list = make([]ReportDTO, 0)
- if reports != nil {
- for _, report := range reports {
- dto := convertReportDTO(report, false)
- list = append(list, dto)
- }
- }
- return
- }
- func getETAReportFirstPermissions(id int) (permissionDTOs []configService.PermissionDTO) {
- classifyId, err := etaDao.GetReportClassifyById(id)
- if err != nil || classifyId == 0 {
- logger.Error("获取研报分类信息失败:%v", err)
- return
- }
- permissions, err := permissionDao.GetFirstPermissionsByClassifyID(classifyId)
- //permissions, err := etaDao.GetFirstPermissionsByClassifyID(classifyId)
- if err != nil {
- logger.Error("获取研报一级品种信息失败:%v", err)
- return
- }
- for _, permission := range permissions {
- permissionDTOs = append(permissionDTOs, convertPermissionDTO(permission))
- }
- return
- }
- func getHTReportFirstPermissions(id int) (permissionDTOs []configService.PermissionDTO) {
- report, err := reportDao.GetReportByOrgId(id, SourceHT)
- if err != nil {
- logger.Error("获取报告失败:%v", err)
- }
- permissionName := report.PlateName
- plateName, err := reportDao.GetPlateNameByPermissionName(permissionName)
- if err != nil {
- return []configService.PermissionDTO{}
- }
- return []configService.PermissionDTO{
- {
- PermissionId: 0,
- PermissionName: plateName,
- ParentId: 0,
- },
- }
- }
- func getETAReportSecondPermissions(id int) (permissionDTOs []configService.PermissionDTO) {
- classifyId, err := reportDao.GetReportClassifyById(id)
- //classifyId, err := etaDao.GetReportClassifyById(id)
- if err != nil || classifyId == 0 {
- logger.Error("获取研报分类信息失败:%v", err)
- return
- }
- permissions, err := permissionDao.GetSecondPermissionsByClassifyID(classifyId)
- //permissions, err := eta.GetSecondPermissionsByClassifyID(classifyId)
- if err != nil {
- logger.Error("获取研报二级品种信息失败:%v", err)
- return
- }
- for _, permission := range permissions {
- permissionDTOs = append(permissionDTOs, convertPermissionDTO(permission))
- }
- return
- }
- func getHTReportSecondPermissions(id int) (permissionDTOs []configService.PermissionDTO) {
- report, err := reportDao.GetReportByOrgId(id, SourceHT)
- if err != nil {
- logger.Error("获取报告失败:%v", err)
- }
- var permission permissionDao.Permission
- if report.PlateName != "" {
- permission, err = permissionDao.GetPermissionByName(report.PlateName)
- if err != nil {
- logger.Error("获取品种信息失败:%v", err)
- }
- }
- return []configService.PermissionDTO{
- {
- PermissionId: permission.PermissionId,
- PermissionName: report.PlateName,
- ParentId: permission.ParentId,
- RiskLevel: permission.RiskLevel,
- },
- }
- }
- func (es ESReport) GetId() string {
- return strconv.Itoa(es.ReportID)
- }
- func GetETALatestReportId() (id int, err error) {
- return reportDao.GetLatestReportIdBySource(reportDao.SourceETA)
- }
- func GetHTLatestReportId() (id int, err error) {
- return reportDao.GetLatestReportIdBySource(reportDao.SourceHT)
- }
- func InitETAReportList(list []eta.ETAReport) (err error) {
- logger.Info("同步研报数量%d", len(list))
- var reports []reportDao.Report
- for _, etaRp := range list {
- var coverSrc int
- //var permissions []etaDao.ChartPermission
- //permissions, err = etaDao.GetSecondPermissionsByClassifyID(etaRp.ClassifyID)
- var permissions []permissionDao.Permission
- permissions, err = permissionDao.GetSecondPermissionsByClassifyID(etaRp.ClassifyID)
- if err != nil || len(permissions) == 0 {
- logger.Error("获取研报二级品种信息失败:%v", err)
- coverSrc = 0
- } else {
- coverSrc = 0
- for _, permission := range permissions {
- permissionsId := permission.PermissionId
- var ids []int
- ids, err = mediaDao.GetIdsByPermissionId(permissionsId)
- if err != nil {
- logger.Error("获取图片资源失败:%v", err)
- continue
- }
- if ids == nil || len(ids) == 0 {
- continue
- }
- src := rand.NewSource(time.Now().UnixNano())
- r := rand.New(src)
- // 从切片中随机选择一个元素
- randomIndex := r.Intn(len(ids))
- coverSrc = ids[randomIndex]
- break
- }
- }
- destRp := convertEtaReport(etaRp, reportDao.StatusPublish)
- //destRp.Author = authorName
- destRp.CoverSrc = coverSrc
- reports = append(reports, destRp)
- //}
- }
- err = reportDao.BatchInsertReport(&reports)
- if err != nil {
- logger.Error("同步ETA研报失败:%v", err)
- return
- }
- return initES(reports)
- }
- func etaStatus(status int) reportDao.ReportStatus {
- if status == etaDao.Passed || status == etaDao.Published {
- return reportDao.StatusPublish
- } else {
- return reportDao.StatusUnPublish
- }
- }
- func SyncETAReportList(list []eta.ETAReport) (err error) {
- logger.Info("同步研报数量%d", len(list))
- var reports []reportDao.Report
- for _, etaRp := range list {
- var coverSrc int
- //var permissions []etaDao.ChartPermission
- //permissions, err = etaDao.GetSecondPermissionsByClassifyID(etaRp.ClassifyID)
- var permissions []permissionDao.Permission
- permissions, err = permissionDao.GetSecondPermissionsByClassifyID(etaRp.ClassifyID)
- if err != nil || len(permissions) == 0 {
- logger.Error("获取研报二级品种信息失败:%v", err)
- coverSrc = 0
- } else {
- coverSrc = 0
- for _, permission := range permissions {
- permissionsId := permission.PermissionId
- var ids []int
- ids, err = mediaDao.GetIdsByPermissionId(permissionsId)
- if err != nil {
- logger.Error("获取图片资源失败:%v", err)
- continue
- }
- if ids == nil || len(ids) == 0 {
- continue
- }
- src := rand.NewSource(time.Now().UnixNano())
- r := rand.New(src)
- // 从切片中随机选择一个元素
- randomIndex := r.Intn(len(ids))
- coverSrc = ids[randomIndex]
- break
- }
- }
- status := etaStatus(etaRp.State)
- destRp := convertEtaReport(etaRp, status)
- destRp.CoverSrc = coverSrc
- reports = append(reports, destRp)
- }
- esList, err := reportDao.InsertOrUpdateReport(reports, SourceETA)
- if esList == nil {
- return
- }
- return syncESAndSendMessage(esList)
- }
- type UpdateESReport struct {
- Title string `json:"title"`
- Author string `json:"author"`
- Abstract string `json:"abstract"`
- PublishedTime string `json:"publishedTime"`
- Status string `json:"status"`
- }
- func syncESAndSendMessage(reports []reportDao.Report) (err error) {
- var esReports []es.ESBase
- for _, etaRp := range reports {
- esRp := convertEsReport(etaRp)
- esReports = append(esReports, esRp)
- }
- //同步es
- for _, report := range reports {
- var exist bool
- exist, err = elastic().Exist(htConfig.GetReportIndex(), report.ID)
- if err != nil {
- logger.Error("查询es失败,reportId::%d,err:%v", report.ID, err)
- }
- if exist {
- update := UpdateESReport{
- Title: report.Title,
- Author: report.Author,
- PublishedTime: report.PublishedTime,
- Abstract: report.Abstract,
- Status: string(report.Status),
- }
- success := elastic().Update(htConfig.GetReportIndex(), report.ID, update)
- if !success {
- logger.Error("更新es失败,reportId::%d,err:%v", report.ID, err)
- }
- if report.Status == reportDao.StatusUnPublish {
- //隐藏热度搜索
- err = userDao.HiddenFlows(report.ID, userDao.ReportSourceType)
- if err != nil {
- logger.Error("隐藏热度搜索失败,reportId::%d,err:%v", report.ID, err)
- }
- } else {
- err = userDao.ShowFlows(report.ID, userDao.ReportSourceType)
- if err != nil {
- logger.Error("重置热度搜索失败,reportId::%d,err:%v", report.ID, err)
- }
- }
- } else {
- insert := ESReport{
- ReportID: report.ID,
- OrgId: report.OrgID,
- Title: report.Title,
- Author: report.Author,
- Source: report.Source,
- Abstract: report.Abstract,
- CoverSrc: report.CoverSrc,
- Status: report.Status,
- PublishedTime: report.PublishedTime,
- }
- success := elastic().CreateDocument(htConfig.GetReportIndex(), report.ID, insert)
- if !success {
- logger.Error("创建es文档失败,reportId::%d,err:%v", report.ID, err)
- }
- }
- }
- //err = elastic().BulkInsert(htConfig.GetReportIndex(), esReports)
- if err != nil {
- logger.Error("同步ETA研报到es失败:%v", err)
- return
- }
- //生产meta信息
- logger.Info("生成推送META信息")
- for _, report := range reports {
- if report.Status == reportDao.StatusUnPublish {
- logger.Info("报告取消发布,不需要生成推送消息")
- continue
- }
- var From string
- switch report.Source {
- case SourceETA:
- From = "ETA"
- case SourceHT:
- From = "HT"
- default:
- From = "UNKNOWN"
- }
- authors := strings.Split(report.Author, ",")
- authors = stringUtils.RemoveEmptyStrings(authors)
- if len(authors) > 0 {
- for _, authorName := range authors {
- userIds := userService.GetPostUser(authorName, report.PublishedTime)
- if len(userIds) > 0 {
- logger.Info("推送META信息,用户ID:%v", userIds)
- var author analystService.FinancialAnalystDTO
- author, err = analystService.GetAnalystByName(authorName)
- if err != nil {
- logger.Error("获取研报作者失败:%v", err)
- continue
- }
- usersStr := stringUtils.IntToStringSlice(userIds)
- Meta := userService.MetaData{
- AuthorName: author.Name,
- AuthorId: author.Id,
- SourceId: report.ID,
- PublishedTime: report.PublishedTime,
- }
- metaStr, _ := json.Marshal(Meta)
- toStr := strings.Join(usersStr, ",")
- metaContent := userService.MetaInfoDTO{
- From: From,
- Meta: string(metaStr),
- MetaType: "USER_NOTICE",
- SourceType: "REPORT",
- To: toStr,
- }
- err = userService.CreateMetaInfo(metaContent)
- if err != nil {
- logger.Error("创建Meta信息失败:%v", err)
- return err
- }
- }
- }
- }
- }
- return
- }
- func initES(reports []reportDao.Report) (err error) {
- var esReports []es.ESBase
- for _, etaRp := range reports {
- esRp := convertEsReport(etaRp)
- esReports = append(esReports, esRp)
- }
- //同步es
- err = elastic().BulkInsert(htConfig.GetReportIndex(), esReports)
- if err != nil {
- logger.Error("同步ETA研报到es失败:%v", err)
- return
- }
- return
- }
- func InitHTReportList(list []ht.HTReport) (noRecord bool, err error) {
- var reports []reportDao.Report
- //获取系统中ht品种名
- permissions, err := reportDao.GetGLAuthorNames()
- if err != nil {
- logger.Error("获取钢联研报作者失败:%v", err)
- return
- }
- for _, htRp := range list {
- for _, permission := range permissions {
- if htRp.PermissionName == permission.Permission {
- if permission.AuthorNames != "" {
- htRp.PublishUserName = permission.AuthorNames
- }
- destRp := convertHTReport(htRp, reportDao.StatusPublish)
- var coverSrc int
- var permissionId int
- permissionId, err = permissionDao.GetPermissionIdByName(htRp.PermissionName)
- //permissionId, err = etaDao.GetPermissionIdByName(htRp.PermissionName)
- if err != nil {
- logger.Error("HT获取eta品种id失败:%v", err)
- coverSrc = 0
- }
- var ids []int
- ids, err = mediaDao.GetIdsByPermissionId(permissionId)
- if err != nil {
- logger.Error("获取图片资源失败:%v", err)
- coverSrc = 0
- }
- if ids == nil || len(ids) == 0 {
- coverSrc = 0
- } else {
- src := rand.NewSource(time.Now().UnixNano())
- r := rand.New(src)
- // 从切片中随机选择一个元素
- randomIndex := r.Intn(len(ids))
- coverSrc = ids[randomIndex]
- }
- destRp.CoverSrc = coverSrc
- destRp.PlateName = htRp.PermissionName
- reports = append(reports, destRp)
- }
- //}
- }
- }
- if len(reports) == 0 {
- return true, nil
- } else {
- logger.Info("同步研报数量%d", len(reports))
- err = reportDao.BatchInsertReport(&reports)
- if err != nil {
- logger.Error("同步HT研报失败:%v", err)
- return false, err
- }
- return false, initES(reports)
- }
- }
- func htStatus(status int, isDelete int) reportDao.ReportStatus {
- if isDelete == 1 || status != ht.Publish {
- return reportDao.StatusUnPublish
- }
- return reportDao.StatusPublish
- }
- func SyncHTReportList(list []ht.HTReport) (noRecord bool, err error) {
- var reports []reportDao.Report
- permissions, err := reportDao.GetGLAuthorNames()
- if err != nil {
- logger.Error("获取钢联研报作者失败:%v", err)
- return
- }
- for _, htRp := range list {
- for _, permission := range permissions {
- if htRp.PermissionName == permission.Permission {
- if permission.AuthorNames != "" {
- htRp.PublishUserName = permission.AuthorNames
- }
- status := htStatus(htRp.Status, htRp.IsDelete)
- destRp := convertHTReport(htRp, status)
- var coverSrc int
- var permissionId int
- permissionId, err = permissionDao.GetPermissionIdByName(htRp.PermissionName)
- //permissionId, err = etaDao.GetPermissionIdByName(htRp.PermissionName)
- if err != nil {
- logger.Error("HT获取eta品种id失败:%v", err)
- coverSrc = 0
- }
- var ids []int
- ids, err = mediaDao.GetIdsByPermissionId(permissionId)
- if err != nil {
- logger.Error("获取图片资源失败:%v", err)
- coverSrc = 0
- }
- if ids == nil || len(ids) == 0 {
- coverSrc = 0
- } else {
- src := rand.NewSource(time.Now().UnixNano())
- r := rand.New(src)
- // 从切片中随机选择一个元素
- randomIndex := r.Intn(len(ids))
- coverSrc = ids[randomIndex]
- }
- destRp.CoverSrc = coverSrc
- destRp.PlateName = htRp.PermissionName
- reports = append(reports, destRp)
- }
- //}
- }
- }
- if len(reports) == 0 {
- return true, nil
- } else {
- logger.Info("同步研报数量%d", len(list))
- }
- esList, err := reportDao.InsertOrUpdateReport(reports, SourceHT)
- if esList == nil {
- return false, err
- }
- return false, syncESAndSendMessage(reports)
- }
- func GetListOrderByConditionWeekly(week bool, column string, limit int, order models.Order) (dtoList []ReportDTO, err error) {
- reports, err := reportDao.GetListOrderByCondition(week, column, limit, order)
- if err != nil {
- logger.Error("获取研报失败:%v", err)
- return
- }
- for _, reportItem := range reports {
- dto := convertReportDTO(reportItem, false)
- dtoList = append(dtoList, dto)
- }
- return
- }
- func GetListByCondition[T any](column string, ids []T) (dtoList []ReportDTO, err error) {
- var values []interface{}
- for _, id := range ids {
- values = append(values, id)
- }
- reports, err := reportDao.GetListByCondition(column, ids)
- if err != nil {
- logger.Error("获取研报失败:%v", err)
- return
- }
- for _, reportItem := range reports {
- dto := convertReportDTO(reportItem, false)
- dtoList = append(dtoList, dto)
- }
- return
- }
- func GetReportByIdListByOrgIds(orgIds map[string][]int) (ids []int, err error) {
- return reportDao.GetReportIdListByOrgIds(orgIds)
- }
- func GetTotalPageCountByPermissionIds(permissionIds []int, riskLevel string) (total int64, latestId int64, ids map[string][]int) {
- htOrgIds, err := GetHTReportIdsByPermissionIdsWithRiskLevel(permissionIds, riskLevel)
- if err != nil {
- logger.Error("品种筛选ht报告id失败:%v", err)
- htOrgIds = []int{}
- }
- etaOrgIds, err := GetETAReportIdsByPermissionIdsWithRiskLevel(permissionIds, riskLevel)
- if err != nil {
- logger.Error("品种筛选eta报告id失败:%v", err)
- etaOrgIds = []int{}
- }
- totalCol := int64(len(etaOrgIds) + len(htOrgIds))
- if totalCol == 0 {
- latestId = 0
- return
- }
- ids = make(map[string][]int, 2)
- if len(etaOrgIds) == 0 {
- ids["ETA"] = []int{}
- } else {
- ids["ETA"] = etaOrgIds
- }
- if len(htOrgIds) == 0 {
- ids["HT"] = []int{}
- } else {
- ids["HT"] = htOrgIds
- }
- total, latestId, err = reportDao.GetMaxIdByPermissionIds(ids)
- if err != nil {
- logger.Error("获取筛选报告的最大记录和记录数失败:%v", err)
- return
- }
- return
- }
- func convertEtaReport(etaRp eta.ETAReport, status reportDao.ReportStatus) reportDao.Report {
- return reportDao.Report{
- OrgID: etaRp.ID,
- Title: etaRp.Title,
- Abstract: etaRp.Abstract,
- Author: etaRp.Author,
- ClassifyId: etaRp.ClassifyID,
- CoverSrc: 0,
- PublishedTime: etaRp.PublishTime.Format(time.DateTime),
- Source: reportDao.SourceETA,
- SendStatus: reportDao.UNSEND,
- Status: status,
- }
- }
- func convertHTReport(etaRp ht.HTReport, status reportDao.ReportStatus) reportDao.Report {
- return reportDao.Report{
- OrgID: etaRp.Id,
- Title: etaRp.ReportName,
- Author: etaRp.PublishUserName,
- PublishedTime: etaRp.PublishedTime,
- CoverSrc: 0,
- Source: reportDao.SourceHT,
- SendStatus: reportDao.UNSEND,
- Status: status,
- }
- }
- func convertEsReport(report reportDao.Report) ESReport {
- return ESReport{
- ReportID: report.ID,
- Title: report.Title,
- OrgId: report.OrgID,
- Author: report.Author,
- Source: report.Source,
- Abstract: report.Abstract,
- Status: report.Status,
- CoverSrc: report.CoverSrc,
- PublishedTime: report.PublishedTime,
- }
- }
- func convertReportDTO(report reportDao.Report, fullTime bool) (reportDTO ReportDTO) {
- reportDTO = ReportDTO{
- ReportID: report.ID,
- Title: report.Title,
- OrgId: report.OrgID,
- Author: report.Author,
- Source: string(report.Source),
- CoverSrc: report.CoverSrc,
- Abstract: report.Abstract,
- PublishedTime: report.PublishedTime,
- }
- publishDate, err := time.Parse(time.DateTime, reportDTO.PublishedTime)
- if err == nil && !fullTime {
- reportDTO.PublishedTime = publishDate.Format(time.DateOnly)
- }
- return
- }
- func matchAll(sorts []string, key string) (request *es.ESQueryRequest) {
- req := new(es.ESQueryRequest)
- return req.CreateESQueryRequest(htConfig.GetReportIndex(), ESColumn, key, 0, 1, sorts, es.MatchAllByCondition).ByCondition("status", "PUBLISH")
- }
- func match(key string, from int, to int, sorts []string) (request *es.ESQueryRequest) {
- req := new(es.ESQueryRequest)
- return req.CreateESQueryRequest(htConfig.GetReportIndex(), ESColumn, key, from, to, sorts, es.MatchAllByCondition).ByCondition("status", "PUBLISH")
- }
- func matchRange(key string, from int, to int, max int64, sorts []string) (request *es.ESQueryRequest) {
- req := new(es.ESQueryRequest)
- return req.CreateESQueryRequest(htConfig.GetReportIndex(), ESColumn, key, from, to, sorts, es.RangeByCondition).Range(0, max, ESRangeColumn).ByCondition("status", "PUBLISH")
- }
- func matchRangeByDocId(key string, from int, to int, max int64, sorts []string, docIds []string) (request *es.ESQueryRequest) {
- req := new(es.ESQueryRequest)
- return req.CreateESQueryRequest(htConfig.GetReportIndex(), ESColumn, key, from, to, sorts, es.RangeByConditionWithDocIds).Range(0, max, ESRangeColumn).ByCondition("status", "PUBLISH").WithDocs(docIds)
- }
|