article.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759
  1. package services
  2. import (
  3. "errors"
  4. "fmt"
  5. "github.com/PuerkitoBio/goquery"
  6. "hongze/hongze_clpt/models"
  7. "hongze/hongze_clpt/utils"
  8. "html"
  9. "regexp"
  10. "sort"
  11. "strconv"
  12. "strings"
  13. "time"
  14. "unicode/utf8"
  15. )
  16. func FixArticleImgUrl(body string) (contentSub string, err error) {
  17. r := strings.NewReader(string(body))
  18. doc, err := goquery.NewDocumentFromReader(r)
  19. if err != nil {
  20. fmt.Println(err)
  21. }
  22. doc.Find("img").Each(func(i int, s *goquery.Selection) {
  23. src, _ := s.Attr("src")
  24. if i == 0 && src != "" {
  25. contentSub = src
  26. }
  27. })
  28. return
  29. }
  30. // GetReportContentTextSubByarticle 解析文章内容
  31. func GetReportContentTextSubByarticle(content, abstract string, articleId int) (contentSub string, err error) {
  32. var lenabstract int
  33. //如果不是研选就这么展示
  34. if articleId < utils.SummaryArticleId {
  35. abstract = html.UnescapeString(abstract)
  36. doc, errdoc := goquery.NewDocumentFromReader(strings.NewReader(abstract))
  37. if errdoc != nil {
  38. err = errdoc
  39. return
  40. }
  41. docabstract := doc.Text()
  42. lenabstract = utf8.RuneCountInString(docabstract)
  43. if lenabstract >= 20 {
  44. contentSub = docabstract
  45. return
  46. } else {
  47. contentSub, err = GetReportContentTextSub(content)
  48. }
  49. } else {
  50. contentSub, err = GetReportContentTextSub(content)
  51. }
  52. return
  53. }
  54. func GetReportContentTextSub(content string) (contentSub string, err error) {
  55. content = html.UnescapeString(content)
  56. doc, errdoc := goquery.NewDocumentFromReader(strings.NewReader(content))
  57. if errdoc != nil {
  58. err = errdoc
  59. return
  60. }
  61. docText := doc.Text()
  62. bodyRune := []rune(docText)
  63. bodyRuneLen := len(bodyRune)
  64. body := string(bodyRune[:bodyRuneLen])
  65. contentSub = body
  66. contentSub = strings.Replace(body, "Powered by Froala Editor", "", -1)
  67. contentSub = strings.Replace(body, "PoweredbyFroalaEditor", "", -1)
  68. contentSub = strings.Replace(body, " ", "", -1)
  69. return
  70. }
  71. func GetReportContentTextArticleBody(content string) (contentSub string) {
  72. contentSub = html.UnescapeString(content)
  73. contentSub = strings.Replace(contentSub, "<p data-f-id=\"pbf\" style=\"text-align: center; font-size: 14px; margin-top: 30px; opacity: 0.65; font-family: sans-serif;\">Powered by <a href=\"https://www.froala.com/wysiwyg-editor?pb=1\" title=\"Froala Editor\">Froala Editor</a></p>", "", -1)
  74. contentSub = strings.Replace(contentSub, "pre", "div", -1)
  75. return
  76. }
  77. // HandleArticleCategoryImg 预处理文章的封面图片
  78. func HandleArticleCategoryImg(list []*models.ArticleListResp, user *models.WxUserItem) (items []*models.ArticleListResp, err error) {
  79. //研选的五张图片
  80. detailResearch, e := models.GetConfigByCode("category_research_img_url")
  81. if e != nil {
  82. err = errors.New("获取研选的五张图片失败" + e.Error())
  83. return
  84. }
  85. researchList := strings.Split(detailResearch.ConfigValue, "{|}")
  86. //对应分类的所图片
  87. detailCategoryUrl, err := models.GetConfigByCode("category_map_img_url")
  88. if err != nil {
  89. err = errors.New("获取对应分类的所图片失败" + err.Error())
  90. return
  91. }
  92. categoryUrlList := strings.Split(detailCategoryUrl.ConfigValue, "{|}")
  93. mapCategoryUrl := make(map[string]string)
  94. var categoryId string
  95. var imgUrlChart string
  96. for _, v := range categoryUrlList {
  97. vslice := strings.Split(v, "_")
  98. categoryId = vslice[0]
  99. imgUrlChart = vslice[len(vslice)-1]
  100. mapCategoryUrl[categoryId] = imgUrlChart
  101. }
  102. mapChartPerssion := make(map[string]string)
  103. reportMappingList, err := models.GetReportMappingStrategyAll()
  104. if err != nil {
  105. err = errors.New("GetReportMappingStrategyAll err" + err.Error())
  106. return
  107. }
  108. for _, v := range reportMappingList {
  109. mapChartPerssion[strconv.Itoa(v.CategoryId)] = v.ChartPermissionName
  110. }
  111. for k, v := range list {
  112. if list[k].Annotation == "" {
  113. imgurl, _ := FixArticleImgUrl(html.UnescapeString(list[k].Body))
  114. if imgurl != "" {
  115. list[k].BodyImg = imgurl
  116. }
  117. }
  118. item := list[k]
  119. //如果文章一开始的内容是图片,优先展示第一张图片
  120. //newBody, _ := GetReportContentTextSubByarticle(item.Body, item.Annotation, item.ArticleId)
  121. list[k].Resource = item.Resource
  122. list[k].Annotation = ArticleAnnotation(item)
  123. list[k].Body = ""
  124. list[k].Abstract, _ = GetReportContentTextSub(v.Abstract)
  125. list[k].PublishDate = utils.StrTimeToTime(item.PublishDate).Format(utils.FormatDate) //时间字符串格式转时间格式
  126. list[k].ChartPermissionName = mapChartPerssion[v.CategoryId]
  127. //如果是研选系列的任意取五张图片的中的一张
  128. if v.CategoryId == "0" || v.ArticleId >= utils.SummaryArticleId {
  129. knum := v.ArticleId % 5
  130. list[k].ImgUrlPc = researchList[knum]
  131. } else {
  132. list[k].ImgUrlPc = mapCategoryUrl[v.CategoryId]
  133. }
  134. if list[k].ArticleId < utils.SummaryArticleId {
  135. list[k].HttpUrl = utils.StrategyPlatform + strconv.Itoa(v.ArticleId)
  136. list[k].IsNeedJump = true
  137. }
  138. list[k].Source = 1
  139. //添加行业默认图片
  140. if v.ImgUrlPc == "" {
  141. if v.ChartPermissionName == utils.YI_YAO_NAME {
  142. list[k].ImgUrlPc = utils.YI_YAO_OTHER_IMG
  143. } else if v.ChartPermissionName == utils.XIAO_FEI_NAME {
  144. list[k].ImgUrlPc = utils.XIAO_FEI_OTHER_IMG
  145. } else if v.ChartPermissionName == utils.KE_JI_NAME {
  146. list[k].ImgUrlPc = utils.KE_JI_OTHER_IMG
  147. } else if v.ChartPermissionName == utils.ZHI_ZAO_NAME {
  148. list[k].ImgUrlPc = utils.ZHI_ZAO_OTHER_IMG
  149. }
  150. }
  151. if v.ArticleTypeId > 0 {
  152. list[k].IsResearch = true
  153. }
  154. }
  155. articleIds := make([]int, 0)
  156. for i := range list {
  157. articleIds = append(articleIds, list[i].ArticleId)
  158. }
  159. // 报告关联产业信息
  160. industryMap := make(map[int][]*models.IndustrialManagementIdInt, 0)
  161. if len(articleIds) > 0 {
  162. var industryCond string
  163. var industryPars []interface{}
  164. industryCond += ` AND mg.article_id IN (` + utils.GetOrmInReplace(len(articleIds)) + `)`
  165. industryPars = append(industryPars, articleIds)
  166. industryList, e := models.GetIndustrialListByarticleId(industryPars, industryCond)
  167. if e != nil {
  168. err = errors.New("GetIndustrialListByarticleId" + e.Error())
  169. return
  170. }
  171. for i := range industryList {
  172. v := industryList[i]
  173. industryMap[v.ArticleId] = append(industryMap[v.ArticleId], &models.IndustrialManagementIdInt{
  174. ArticleId: v.ArticleId,
  175. IndustrialManagementId: v.IndustrialManagementId,
  176. IndustryName: v.IndustryName,
  177. ChartPermissionId: v.ChartPermissionId,
  178. })
  179. }
  180. }
  181. // 处理文章收藏字段
  182. mapCollect, e := GetUserAticleCollectMap(user)
  183. if e != nil {
  184. err = errors.New("GetUserAticleCollectMap" + e.Error())
  185. return
  186. }
  187. for k, v := range list {
  188. if len(industryMap[v.ArticleId]) > 0 {
  189. list[k].List = industryMap[v.ArticleId]
  190. } else {
  191. list[k].List = make([]*models.IndustrialManagementIdInt, 0)
  192. }
  193. if _, ok := mapCollect[v.ArticleId]; ok {
  194. list[k].IsCollect = true
  195. }
  196. }
  197. if len(list) == 0 {
  198. list = make([]*models.ArticleListResp, 0)
  199. }
  200. items = list
  201. return
  202. }
  203. // HandleArticleStock 处理报告关联的个股标签
  204. func HandleArticleStock(stock string) (items []*models.ComapnyNameResp) {
  205. sliceSubjects := strings.Split(stock, "/")
  206. if len(sliceSubjects) > 0 {
  207. for _, vSubject := range sliceSubjects {
  208. sliceKuohao := strings.Split(vSubject, "(") //过滤括号
  209. sliceXiahuaxian := strings.Split(sliceKuohao[0], "-") //过滤下划线
  210. subject := sliceXiahuaxian[0]
  211. items = append(items, &models.ComapnyNameResp{ComapnyName: subject})
  212. }
  213. }
  214. return
  215. }
  216. // 弘则报告发布日期在三个月以内的
  217. func GetArticNewLabelWhithActivity3Month() (labelMap map[int]bool, err error) {
  218. var condition string
  219. var pars []interface{}
  220. condition += ` AND publish_date <= ? AND article_id < ? `
  221. pars = append(pars, time.Now().AddDate(0, -3, 0), utils.SummaryArticleId)
  222. articleList, e := models.GetArticleList(condition, pars)
  223. if e != nil {
  224. err = errors.New("GetArticleList, Err: " + e.Error())
  225. return
  226. }
  227. var articleIds []int
  228. for _, v := range articleList {
  229. articleIds = append(articleIds, v.ArticleId)
  230. }
  231. if len(articleIds) == 0 {
  232. return
  233. }
  234. pars = make([]interface{}, 0)
  235. condition = ` AND article_id IN (` + utils.GetOrmInReplace(len(articleIds)) + `)`
  236. pars = append(pars, articleIds)
  237. industrialList, e := models.GetIndustrialArticleGroupManagementList(condition, pars)
  238. if e != nil {
  239. err = errors.New("GetIndustrialArticleGroupManagementList, Err: " + e.Error())
  240. return
  241. }
  242. labelMap = make(map[int]bool, 0)
  243. var industrialIds []int
  244. for _, v := range industrialList {
  245. industrialIds = append(industrialIds, v.IndustrialManagementId)
  246. }
  247. // 获取活动关联的产业
  248. var groupCond string
  249. var groupPars []interface{}
  250. groupCond += ` AND b.industrial_management_id IN (` + utils.GetOrmInReplace(len(industrialIds)) + `) AND b.source = 1 `
  251. groupPars = append(groupPars, industrialIds)
  252. groups, e := models.GetActivityIndustryRelationList(groupCond, groupPars)
  253. if e != nil {
  254. err = errors.New("获取活动产业关联列表失败, Err: " + e.Error())
  255. return
  256. }
  257. for _, v := range groups {
  258. labelMap[v.ActivityId] = true
  259. }
  260. return
  261. }
  262. // GetSpecialArticleDetailUserPower 处理用户查看专项调研文章详情的权限
  263. func GetSpecialArticleDetailUserPower(user *models.WxUserItem, articleInfo *models.ArticleDetail) (havePower bool, err error) {
  264. userType, _, e := GetUserType(user.CompanyId)
  265. if e != nil {
  266. err = errors.New("GetSpecialUserType, Err: " + e.Error())
  267. return
  268. }
  269. // 永续客户、大套餐客户可以查看行业升级套餐客户 权限
  270. if userType == 1 || userType == 2 {
  271. havePower = true
  272. return
  273. }
  274. permissionStr, e := GetCompanyPermissionUpgrade(user.CompanyId)
  275. if e != nil {
  276. err = errors.New("GetCompanyPermissionUpgrade, Err: " + e.Error())
  277. return
  278. }
  279. reportMapDetail, e := models.GetdetailByCategoryIdPush(articleInfo.CategoryId)
  280. if e != nil {
  281. err = errors.New("GetdetailByCategoryIdPush, Err: " + e.Error())
  282. return
  283. }
  284. if reportMapDetail == nil {
  285. err = errors.New("GetdetailByCategoryIdP,获取详情失败, Err: ")
  286. return
  287. }
  288. //如果没有对应的升级权限,则返回
  289. if !strings.Contains(permissionStr, reportMapDetail.ChartPermissionName) {
  290. return
  291. } else {
  292. havePower = true
  293. }
  294. return
  295. }
  296. //处理核心观点的展示规则
  297. //func ArticleAnnotation(item *models.ArticleListResp) (annotation string) {
  298. // if item.Annotation != "" {
  299. // annotation = strings.Replace(item.Annotation, "<br>", "", -1)
  300. // }
  301. // return
  302. //}
  303. func GetReportContentTextSubNew(content string) (contentSub string, err error) {
  304. content = html.UnescapeString(content)
  305. doc, errdoc := goquery.NewDocumentFromReader(strings.NewReader(content))
  306. if errdoc != nil {
  307. err = errdoc
  308. return
  309. }
  310. docText := doc.Text()
  311. bodyRune := []rune(docText)
  312. bodyRuneLen := len(bodyRune)
  313. body := string(bodyRune[:bodyRuneLen])
  314. contentSub = body
  315. contentSub = strings.Replace(contentSub, "Powered by Froala Editor", "", -1)
  316. contentSub = strings.Replace(contentSub, " ", "", -1)
  317. contentSub = strings.Replace(contentSub, "<p data-f-id=\"pbf\" style=\"text-align: center; font-size: 14px; margin-top: 30px; opacity: 0.65; font-family: sanered by <a href=\"https://www.froala.com/wysiwyg-editor?pb=1\" title=\"Froala Editor\">Froala Editor</a></p>", "", -1)
  318. return
  319. }
  320. // 处理核心观点的展示规则
  321. func ArticleAnnotation(item *models.ArticleListResp) (annotation string) {
  322. if item.ArticleId >= utils.SummaryArticleId {
  323. item.Annotation = YxArticleAnnotation(item)
  324. }
  325. if item.Annotation != "" {
  326. annotation = strings.Replace(item.Annotation, "<br>", "", -1)
  327. } else {
  328. return
  329. }
  330. bodyText, _ := GetReportContentTextSubNew(annotation)
  331. if bodyText == "" {
  332. return
  333. }
  334. if annotation != "" {
  335. annotation = html.UnescapeString(annotation)
  336. doc, _ := goquery.NewDocumentFromReader(strings.NewReader(annotation))
  337. docText := doc.Text()
  338. mapDoc := make(map[int]string)
  339. mapSortRepeat := make(map[string]string)
  340. var mapSort []int
  341. p := doc.Find("p")
  342. p.Each(func(tk int, pd *goquery.Selection) {
  343. pdText := pd.Text()
  344. pdText = strings.Replace(pdText, " ", "", -1)
  345. if pdText != "" {
  346. textLen := strings.Index(docText, pdText)
  347. if mapSortRepeat[strconv.Itoa(textLen)] == "" {
  348. mapDoc[(strings.Index(docText, pdText))] = pdText
  349. mapSort = append(mapSort, textLen)
  350. mapSortRepeat[strconv.Itoa(textLen)] = strconv.Itoa(textLen)
  351. }
  352. }
  353. })
  354. li := doc.Find("li")
  355. li.Each(func(tk int, li *goquery.Selection) {
  356. liText := li.Text()
  357. liText = strings.Replace(liText, " ", "", -1)
  358. if liText != "" {
  359. textLen := strings.Index(docText, liText)
  360. if mapSortRepeat[strconv.Itoa(textLen)] == "" {
  361. mapDoc[(strings.Index(docText, liText))] = strconv.Itoa(tk+1) + "." + liText
  362. mapSort = append(mapSort, textLen)
  363. mapSortRepeat[strconv.Itoa(textLen)] = strconv.Itoa(textLen)
  364. }
  365. }
  366. })
  367. ul := doc.Find("ul")
  368. ul.Each(func(tk int, ul *goquery.Selection) {
  369. ulText := ul.Text()
  370. ulText = strings.Replace(ulText, " ", "", -1)
  371. if ulText != "" {
  372. textLen := strings.Index(docText, ulText)
  373. if mapSortRepeat[strconv.Itoa(textLen)] == "" {
  374. mapDoc[(strings.Index(docText, ulText))] = ulText
  375. mapSort = append(mapSort, textLen)
  376. mapSortRepeat[strconv.Itoa(textLen)] = strconv.Itoa(textLen)
  377. }
  378. }
  379. })
  380. if len(mapSort) == 0 {
  381. return
  382. } else {
  383. //排序
  384. sort.Ints(mapSort)
  385. var annotationHtml string
  386. for _, vSort := range mapSort {
  387. for k, v := range mapDoc {
  388. if k == vSort && v != "" {
  389. annotationHtml += v + "<br>"
  390. }
  391. }
  392. }
  393. annotationHtml = strings.TrimRight(annotationHtml, "<br>")
  394. annotationHtml = "<p>" + annotationHtml + "</p>"
  395. annotation = annotationHtml
  396. }
  397. }
  398. return
  399. }
  400. // 处理核心观点的展示规则
  401. func AnnotationHtml(bodyText string) (annotation string) {
  402. if bodyText == "" {
  403. return
  404. }
  405. annotation = bodyText
  406. annotation = html.UnescapeString(annotation)
  407. doc, _ := goquery.NewDocumentFromReader(strings.NewReader(annotation))
  408. docText := doc.Text()
  409. mapDoc := make(map[int]string)
  410. var mapSort []int
  411. p := doc.Find("p")
  412. p.Each(func(tk int, pd *goquery.Selection) {
  413. pdText := pd.Text()
  414. pdText = strings.Replace(pdText, " ", "", -1)
  415. if pdText != "" {
  416. textLen := strings.Index(docText, pdText)
  417. mapDoc[(strings.Index(docText, pdText))] = pdText
  418. mapSort = append(mapSort, textLen)
  419. }
  420. })
  421. li := doc.Find("li")
  422. li.Each(func(tk int, li *goquery.Selection) {
  423. liText := li.Text()
  424. liText = strings.Replace(liText, " ", "", -1)
  425. if liText != "" {
  426. textLen := strings.Index(docText, liText)
  427. mapDoc[(strings.Index(docText, liText))] = strconv.Itoa(tk+1) + "." + liText
  428. mapSort = append(mapSort, textLen)
  429. }
  430. })
  431. ul := doc.Find("ul")
  432. ul.Each(func(tk int, ul *goquery.Selection) {
  433. ulText := ul.Text()
  434. ulText = strings.Replace(ulText, " ", "", -1)
  435. if ulText != "" {
  436. textLen := strings.Index(docText, ulText)
  437. mapDoc[(strings.Index(docText, ulText))] = ulText
  438. mapSort = append(mapSort, textLen)
  439. }
  440. })
  441. if len(mapSort) == 0 {
  442. return
  443. } else {
  444. //排序
  445. sort.Ints(mapSort)
  446. var annotationHtml string
  447. for _, vSort := range mapSort {
  448. for k, v := range mapDoc {
  449. if k == vSort && v != "" {
  450. annotationHtml += v + "<br>"
  451. }
  452. }
  453. }
  454. annotationHtml = strings.TrimRight(annotationHtml, "<br>")
  455. annotationHtml = "<p>" + annotationHtml + "</p>"
  456. annotation = annotationHtml
  457. }
  458. return
  459. }
  460. // 处理产品内测展示规则
  461. func ProductInteriorHtml(bodyText string) (annotation string) {
  462. if bodyText == "" {
  463. return
  464. }
  465. sliceBody := strings.Split(bodyText, "</p>")
  466. annotation, _ = GetReportContentTextSub(sliceBody[0])
  467. return
  468. }
  469. // 解析研选内容中的核心观点
  470. func YxArticleAnnotation(article *models.ArticleListResp) (annotation string) {
  471. //如果不规范,就获取内容主体
  472. if strings.Count(article.Body, "<hr") == 0 {
  473. //如果内容不规范而且,还有图片,就把核心观点置空
  474. if article.BodyImg != "" {
  475. return
  476. }
  477. annotation, _ = GetReportContentTextSub(article.Body)
  478. return
  479. }
  480. body := strings.ReplaceAll(article.Body, "<strong>", "")
  481. body = strings.ReplaceAll(body, "</strong>", "")
  482. body = strings.ReplaceAll(body, "</ol>", "</div>")
  483. body = strings.ReplaceAll(body, "<ol>", "<div>")
  484. body = strings.ReplaceAll(body, "</li>", "</p>")
  485. body = strings.ReplaceAll(body, "<li>", "<p>")
  486. re, _ := regexp.Compile("<strong.*?>")
  487. body = re.ReplaceAllString(body, "")
  488. reLi, _ := regexp.Compile("<li.*?>")
  489. body = reLi.ReplaceAllString(body, "")
  490. var plus int
  491. coreIndex := strings.Index(body, "核心观点:")
  492. plus = 15
  493. if coreIndex == -1 {
  494. coreIndex = strings.Index(body, "核心观点:")
  495. plus = 13
  496. }
  497. if coreIndex == -1 {
  498. coreIndex = strings.Index(body, "核心观点")
  499. plus = 12
  500. }
  501. if coreIndex == -1 {
  502. coreIndex = strings.Index(body, "核心结论:")
  503. plus = 15
  504. }
  505. if coreIndex == -1 {
  506. coreIndex = strings.Index(body, "核心结论:")
  507. plus = 13
  508. }
  509. if coreIndex == -1 {
  510. coreIndex = strings.Index(body, "核心结论")
  511. plus = 12
  512. }
  513. endIndex := strings.Index(body, "<hr")
  514. if coreIndex != -1 && endIndex != -1 {
  515. body = body[coreIndex+plus : endIndex]
  516. }
  517. annotation = body
  518. return
  519. }
  520. // 获取研选类型的文章分类Id
  521. func GetYanXuanArticleTypeIds() (articleTypeIds string, err error) {
  522. var condition string
  523. condition = " AND is_show_yanx = 1 "
  524. listType, e := models.GetCygxArticleTypeListCondition(condition)
  525. if e != nil {
  526. err = errors.New("GetCygxArticleTypeListCondition, Err: " + e.Error())
  527. return
  528. }
  529. for _, v := range listType {
  530. articleTypeIds += strconv.Itoa(v.ArticleTypeId) + ","
  531. }
  532. articleTypeIds = strings.TrimRight(articleTypeIds, ",")
  533. if articleTypeIds == "" {
  534. err = errors.New("研选分类ID不能为空")
  535. return
  536. }
  537. return
  538. }
  539. // GetUserAticleCollectMap 获取用户收藏的文章ID
  540. func GetUserAticleCollectMap(user *models.WxUserItem) (respMap map[int]int, err error) {
  541. list, e := models.GetCygxArticleCollectListByUser(user.UserId)
  542. if e != nil {
  543. err = errors.New("GetCygxArticleCollectListByUser, Err: " + e.Error())
  544. return
  545. }
  546. articleMap := make(map[int]int)
  547. for _, v := range list {
  548. articleMap[v.ArticleId] = v.ArticleId
  549. }
  550. respMap = articleMap
  551. return
  552. }
  553. // 通过接解析带有Md5的文章链接获取文章ID
  554. func GetReportLinkToArticleid(reportLink string) (articleId int, err error) {
  555. defer func() {
  556. if err != nil {
  557. go utils.SendAlarmMsg("通过接解析带有Md5的文章链接获取文章ID失败"+err.Error(), 2)
  558. }
  559. }()
  560. var artMd5 string
  561. //处理Md5的
  562. strnum1 := strings.Index(reportLink, "id=")
  563. if strnum1 > 0 {
  564. sliceId := strings.Split(reportLink, "id=")
  565. if len(sliceId) > 1 {
  566. reportLink = sliceId[1]
  567. sliceMd5Id := strings.Split(reportLink, "&")
  568. artMd5 = sliceMd5Id[0]
  569. }
  570. if artMd5 != "" {
  571. detail, errArt := models.GetArticleDetailByIdMd5(artMd5)
  572. if errArt != nil && errArt.Error() != utils.ErrNoRow() {
  573. err = errArt
  574. return
  575. }
  576. if detail != nil {
  577. articleId = detail.ArticleId
  578. }
  579. }
  580. } else {
  581. //处理活动的
  582. linkList := strings.Split(reportLink, "/")
  583. if linkList[len(linkList)-1] != "" {
  584. linkArticleId, _ := strconv.Atoi(linkList[len(linkList)-1])
  585. if linkArticleId > 0 {
  586. articleInfo, errArt := models.GetArticleDetailById(linkArticleId)
  587. if errArt != nil && errArt.Error() != utils.ErrNoRow() {
  588. err = errArt
  589. return
  590. }
  591. if articleInfo != nil {
  592. articleId = articleInfo.ArticleId
  593. }
  594. }
  595. }
  596. }
  597. return
  598. }
  599. // GetArticleStockMap 获取个股标签所对应的文章ID
  600. func GetArticleStockMap() (mapResp map[string]int, err error) {
  601. defer func() {
  602. if err != nil {
  603. go utils.SendAlarmMsg("获取个股标签所对应的文章ID失败"+err.Error(), 2)
  604. }
  605. }()
  606. list, err := models.GetArticleStock()
  607. if err != nil && err.Error() != utils.ErrNoRow() {
  608. return
  609. }
  610. mapResp = make(map[string]int, 0)
  611. if len(list) > 0 {
  612. //一对一精准匹配
  613. for _, v := range list {
  614. sliceSubjects := strings.Split(v.Stock, "/")
  615. if len(sliceSubjects) > 0 {
  616. for _, vSubject := range sliceSubjects {
  617. sliceKuohao := strings.Split(vSubject, "(") //过滤括号
  618. sliceXiahuaxian := strings.Split(sliceKuohao[0], "-") //过滤下划线
  619. subject := sliceXiahuaxian[0]
  620. mapResp[subject] = v.ArticleId
  621. }
  622. }
  623. }
  624. }
  625. return
  626. }
  627. // 用户报告操作行为,模板消息推送
  628. func ArticleUserRemind(user *models.WxUserItem, articleDetail *models.ArticleDetail, source int) (err error) {
  629. defer func() {
  630. if err != nil {
  631. go utils.SendAlarmMsg("同步策略平台阅读数据失败", 2)
  632. go utils.SendEmail(utils.APPNAME+"【"+utils.RunMode+"】"+"失败提醒", "GetCeLueArticlePv ErrMsg:"+err.Error(), utils.EmailSendToUsers)
  633. }
  634. }()
  635. countUser, err := models.GetUserRemind(user.UserId)
  636. if err != nil {
  637. return err
  638. }
  639. if countUser == 0 {
  640. return err
  641. }
  642. var sourceMsg string
  643. if source == 1 {
  644. sourceMsg = "阅读报告"
  645. } else {
  646. sourceMsg = "收藏报告"
  647. }
  648. //获取销售手机号
  649. sellerItemQy, err := models.GetSellerByCompanyIdCheckFicc(user.CompanyId, 2)
  650. if err != nil && err.Error() != utils.ErrNoRow() {
  651. return err
  652. }
  653. if sellerItemQy != nil {
  654. openIdList, err := models.GetWxOpenIdByMobileList(sellerItemQy.Mobile)
  655. if err != nil {
  656. return err
  657. }
  658. var keyword1 string
  659. var keyword2 string
  660. keyword1 = articleDetail.Title
  661. keyword2 = fmt.Sprint("互动:", sourceMsg, ",", user.RealName, "--", user.CompanyName)
  662. SendWxMsgWithArticleUserRemind(keyword1, keyword2, openIdList, articleDetail.ArticleId)
  663. }
  664. return
  665. }
  666. // GetAiQianYanArtilceList 获取AI前沿几篇文章
  667. func GetAiQianYanArtilceList(startSize, pageSize int) (items []*models.HomeArticle, total int, err error) {
  668. defer func() {
  669. if err != nil {
  670. go utils.SendAlarmMsg("获取AI前沿几篇文章失败"+err.Error(), 2)
  671. }
  672. }()
  673. var condition string
  674. var pars []interface{}
  675. condition += ` AND title LIKE '%AI前沿%' AND publish_status = 1 ORDER BY publish_date DESC `
  676. articleList, e := models.GetCygxCygxArticleList(condition, pars, startSize, pageSize)
  677. if e != nil {
  678. err = errors.New("GetCygxCygxArticleList, Err: " + e.Error())
  679. return
  680. }
  681. total, e = models.GetCygxArticleCount(condition, pars)
  682. if e != nil {
  683. err = errors.New("GetCygxArticleCount, Err: " + e.Error())
  684. return
  685. }
  686. for _, v := range articleList {
  687. item := new(models.HomeArticle)
  688. item.ArticleId = v.ArticleId
  689. item.Title = v.Title
  690. item.Abstract = v.Abstract
  691. item.Annotation = v.Annotation
  692. item.PublishDate = v.PublishDate
  693. item.CategoryId = strconv.Itoa(v.CategoryId)
  694. item.Body = v.Body
  695. items = append(items, item)
  696. }
  697. return
  698. }
  699. // GetYxArticleIdMap 获取研选文章ID
  700. func GetYxArticleIdMap(articleIds []int) (mapResp map[int]bool) {
  701. var err error
  702. defer func() {
  703. if err != nil {
  704. go utils.SendAlarmMsg("获取研选文章ID失败,GetYxArticleIdMap"+err.Error(), 2)
  705. }
  706. }()
  707. var condition string
  708. var pars []interface{}
  709. condition = ` AND article_type_id > 0 `
  710. if len(articleIds) > 0 {
  711. condition += ` AND article_id IN (` + utils.GetOrmInReplace(len(articleIds)) + `)`
  712. pars = append(pars, articleIds)
  713. }
  714. articleList, e := models.GetArticleList(condition, pars)
  715. if e != nil {
  716. err = errors.New("GetArticleList, Err: " + e.Error())
  717. return
  718. }
  719. mapResp = make(map[int]bool, 0)
  720. for _, v := range articleList {
  721. mapResp[v.ArticleId] = true
  722. }
  723. return
  724. }