article.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774
  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 _, v := range list {
  157. if v.IsSpecial == 0 {
  158. articleIds = append(articleIds, v.ArticleId)
  159. }
  160. }
  161. // 报告关联产业信息
  162. industryMap := make(map[int][]*models.IndustrialManagementIdInt, 0)
  163. if len(articleIds) > 0 {
  164. var industryCond string
  165. var industryPars []interface{}
  166. industryCond += ` AND mg.article_id IN (` + utils.GetOrmInReplace(len(articleIds)) + `)`
  167. industryPars = append(industryPars, articleIds)
  168. industryList, e := models.GetIndustrialListByarticleId(industryPars, industryCond)
  169. if e != nil {
  170. err = errors.New("GetIndustrialListByarticleId" + e.Error())
  171. return
  172. }
  173. for i := range industryList {
  174. v := industryList[i]
  175. industryMap[v.ArticleId] = append(industryMap[v.ArticleId], &models.IndustrialManagementIdInt{
  176. ArticleId: v.ArticleId,
  177. IndustrialManagementId: v.IndustrialManagementId,
  178. IndustryName: v.IndustryName,
  179. ChartPermissionId: v.ChartPermissionId,
  180. })
  181. }
  182. }
  183. //// 处理文章收藏字段
  184. //mapCollect, e := GetUserAticleCollectMap(user)
  185. //if e != nil {
  186. // err = errors.New("GetUserAticleCollectMap" + e.Error())
  187. // return
  188. //}
  189. //var articleIds []int
  190. //for _, v := range list {
  191. // articleIds = append(articleIds, v.ArticleId)
  192. //}
  193. articleMapPv := GetArticleHistoryByArticleId(articleIds) //文章Pv
  194. articleCollectMap, _ := GetCygxArticleCollectMap(user.UserId) //用户收藏的文章
  195. articleCollectNumMap, _ := GetCygxArticleCollectNumMapByArtcileIds(articleIds) //文章收藏的数量
  196. for k, v := range list {
  197. if len(industryMap[v.ArticleId]) > 0 {
  198. list[k].List = industryMap[v.ArticleId]
  199. } else {
  200. list[k].List = make([]*models.IndustrialManagementIdInt, 0)
  201. }
  202. //if _, ok := mapCollect[v.ArticleId]; ok {
  203. // list[k].IsCollect = true
  204. //}
  205. if v.IsSpecial == 0 {
  206. list[k].Pv = articleMapPv[v.ArticleId]
  207. list[k].IsCollect = articleCollectMap[v.ArticleId]
  208. list[k].CollectNum = articleCollectNumMap[v.ArticleId]
  209. }
  210. }
  211. if len(list) == 0 {
  212. list = make([]*models.ArticleListResp, 0)
  213. }
  214. items = list
  215. return
  216. }
  217. // HandleArticleStock 处理报告关联的个股标签
  218. func HandleArticleStock(stock string) (items []*models.ComapnyNameResp) {
  219. sliceSubjects := strings.Split(stock, "/")
  220. if len(sliceSubjects) > 0 {
  221. for _, vSubject := range sliceSubjects {
  222. sliceKuohao := strings.Split(vSubject, "(") //过滤括号
  223. sliceXiahuaxian := strings.Split(sliceKuohao[0], "-") //过滤下划线
  224. subject := sliceXiahuaxian[0]
  225. items = append(items, &models.ComapnyNameResp{ComapnyName: subject})
  226. }
  227. }
  228. return
  229. }
  230. // 弘则报告发布日期在三个月以内的
  231. func GetArticNewLabelWhithActivity3Month() (labelMap map[int]bool, err error) {
  232. var condition string
  233. var pars []interface{}
  234. condition += ` AND publish_date <= ? AND article_id < ? `
  235. pars = append(pars, time.Now().AddDate(0, -3, 0), utils.SummaryArticleId)
  236. articleList, e := models.GetArticleList(condition, pars)
  237. if e != nil {
  238. err = errors.New("GetArticleList, Err: " + e.Error())
  239. return
  240. }
  241. var articleIds []int
  242. for _, v := range articleList {
  243. articleIds = append(articleIds, v.ArticleId)
  244. }
  245. if len(articleIds) == 0 {
  246. return
  247. }
  248. pars = make([]interface{}, 0)
  249. condition = ` AND article_id IN (` + utils.GetOrmInReplace(len(articleIds)) + `)`
  250. pars = append(pars, articleIds)
  251. industrialList, e := models.GetIndustrialArticleGroupManagementList(condition, pars)
  252. if e != nil {
  253. err = errors.New("GetIndustrialArticleGroupManagementList, Err: " + e.Error())
  254. return
  255. }
  256. labelMap = make(map[int]bool, 0)
  257. var industrialIds []int
  258. for _, v := range industrialList {
  259. industrialIds = append(industrialIds, v.IndustrialManagementId)
  260. }
  261. // 获取活动关联的产业
  262. var groupCond string
  263. var groupPars []interface{}
  264. groupCond += ` AND b.industrial_management_id IN (` + utils.GetOrmInReplace(len(industrialIds)) + `) AND b.source = 1 `
  265. groupPars = append(groupPars, industrialIds)
  266. groups, e := models.GetActivityIndustryRelationList(groupCond, groupPars)
  267. if e != nil {
  268. err = errors.New("获取活动产业关联列表失败, Err: " + e.Error())
  269. return
  270. }
  271. for _, v := range groups {
  272. labelMap[v.ActivityId] = true
  273. }
  274. return
  275. }
  276. // GetSpecialArticleDetailUserPower 处理用户查看专项调研文章详情的权限
  277. func GetSpecialArticleDetailUserPower(user *models.WxUserItem, articleInfo *models.ArticleDetail) (havePower bool, err error) {
  278. userType, _, e := GetUserType(user.CompanyId)
  279. if e != nil {
  280. err = errors.New("GetSpecialUserType, Err: " + e.Error())
  281. return
  282. }
  283. // 永续客户、大套餐客户可以查看行业升级套餐客户 权限
  284. if userType == 1 || userType == 2 {
  285. havePower = true
  286. return
  287. }
  288. permissionStr, e := GetCompanyPermissionUpgrade(user.CompanyId)
  289. if e != nil {
  290. err = errors.New("GetCompanyPermissionUpgrade, Err: " + e.Error())
  291. return
  292. }
  293. reportMapDetail, e := models.GetdetailByCategoryIdPush(articleInfo.CategoryId)
  294. if e != nil {
  295. err = errors.New("GetdetailByCategoryIdPush, Err: " + e.Error())
  296. return
  297. }
  298. if reportMapDetail == nil {
  299. err = errors.New("GetdetailByCategoryIdP,获取详情失败, Err: ")
  300. return
  301. }
  302. //如果没有对应的升级权限,则返回
  303. if !strings.Contains(permissionStr, reportMapDetail.ChartPermissionName) {
  304. return
  305. } else {
  306. havePower = true
  307. }
  308. return
  309. }
  310. //处理核心观点的展示规则
  311. //func ArticleAnnotation(item *models.ArticleListResp) (annotation string) {
  312. // if item.Annotation != "" {
  313. // annotation = strings.Replace(item.Annotation, "<br>", "", -1)
  314. // }
  315. // return
  316. //}
  317. func GetReportContentTextSubNew(content string) (contentSub string, err error) {
  318. content = html.UnescapeString(content)
  319. doc, errdoc := goquery.NewDocumentFromReader(strings.NewReader(content))
  320. if errdoc != nil {
  321. err = errdoc
  322. return
  323. }
  324. docText := doc.Text()
  325. bodyRune := []rune(docText)
  326. bodyRuneLen := len(bodyRune)
  327. body := string(bodyRune[:bodyRuneLen])
  328. contentSub = body
  329. contentSub = strings.Replace(contentSub, "Powered by Froala Editor", "", -1)
  330. contentSub = strings.Replace(contentSub, " ", "", -1)
  331. 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)
  332. return
  333. }
  334. // 处理核心观点的展示规则
  335. func ArticleAnnotation(item *models.ArticleListResp) (annotation string) {
  336. if item.ArticleId >= utils.SummaryArticleId {
  337. item.Annotation = YxArticleAnnotation(item)
  338. }
  339. if item.Annotation != "" {
  340. annotation = strings.Replace(item.Annotation, "<br>", "", -1)
  341. } else {
  342. return
  343. }
  344. bodyText, _ := GetReportContentTextSubNew(annotation)
  345. if bodyText == "" {
  346. return
  347. }
  348. if annotation != "" {
  349. annotation = html.UnescapeString(annotation)
  350. doc, _ := goquery.NewDocumentFromReader(strings.NewReader(annotation))
  351. docText := doc.Text()
  352. mapDoc := make(map[int]string)
  353. mapSortRepeat := make(map[string]string)
  354. var mapSort []int
  355. p := doc.Find("p")
  356. p.Each(func(tk int, pd *goquery.Selection) {
  357. pdText := pd.Text()
  358. pdText = strings.Replace(pdText, " ", "", -1)
  359. if pdText != "" {
  360. textLen := strings.Index(docText, pdText)
  361. if mapSortRepeat[strconv.Itoa(textLen)] == "" {
  362. mapDoc[(strings.Index(docText, pdText))] = pdText
  363. mapSort = append(mapSort, textLen)
  364. mapSortRepeat[strconv.Itoa(textLen)] = strconv.Itoa(textLen)
  365. }
  366. }
  367. })
  368. li := doc.Find("li")
  369. li.Each(func(tk int, li *goquery.Selection) {
  370. liText := li.Text()
  371. liText = strings.Replace(liText, " ", "", -1)
  372. if liText != "" {
  373. textLen := strings.Index(docText, liText)
  374. if mapSortRepeat[strconv.Itoa(textLen)] == "" {
  375. mapDoc[(strings.Index(docText, liText))] = strconv.Itoa(tk+1) + "." + liText
  376. mapSort = append(mapSort, textLen)
  377. mapSortRepeat[strconv.Itoa(textLen)] = strconv.Itoa(textLen)
  378. }
  379. }
  380. })
  381. ul := doc.Find("ul")
  382. ul.Each(func(tk int, ul *goquery.Selection) {
  383. ulText := ul.Text()
  384. ulText = strings.Replace(ulText, " ", "", -1)
  385. if ulText != "" {
  386. textLen := strings.Index(docText, ulText)
  387. if mapSortRepeat[strconv.Itoa(textLen)] == "" {
  388. mapDoc[(strings.Index(docText, ulText))] = ulText
  389. mapSort = append(mapSort, textLen)
  390. mapSortRepeat[strconv.Itoa(textLen)] = strconv.Itoa(textLen)
  391. }
  392. }
  393. })
  394. if len(mapSort) == 0 {
  395. return
  396. } else {
  397. //排序
  398. sort.Ints(mapSort)
  399. var annotationHtml string
  400. for _, vSort := range mapSort {
  401. for k, v := range mapDoc {
  402. if k == vSort && v != "" {
  403. annotationHtml += v + "<br>"
  404. }
  405. }
  406. }
  407. annotationHtml = strings.TrimRight(annotationHtml, "<br>")
  408. annotationHtml = "<p>" + annotationHtml + "</p>"
  409. annotation = annotationHtml
  410. }
  411. }
  412. return
  413. }
  414. // 处理核心观点的展示规则
  415. func AnnotationHtml(bodyText string) (annotation string) {
  416. if bodyText == "" {
  417. return
  418. }
  419. annotation = bodyText
  420. annotation = html.UnescapeString(annotation)
  421. doc, _ := goquery.NewDocumentFromReader(strings.NewReader(annotation))
  422. docText := doc.Text()
  423. mapDoc := make(map[int]string)
  424. var mapSort []int
  425. p := doc.Find("p")
  426. p.Each(func(tk int, pd *goquery.Selection) {
  427. pdText := pd.Text()
  428. pdText = strings.Replace(pdText, " ", "", -1)
  429. if pdText != "" {
  430. textLen := strings.Index(docText, pdText)
  431. mapDoc[(strings.Index(docText, pdText))] = pdText
  432. mapSort = append(mapSort, textLen)
  433. }
  434. })
  435. li := doc.Find("li")
  436. li.Each(func(tk int, li *goquery.Selection) {
  437. liText := li.Text()
  438. liText = strings.Replace(liText, " ", "", -1)
  439. if liText != "" {
  440. textLen := strings.Index(docText, liText)
  441. mapDoc[(strings.Index(docText, liText))] = strconv.Itoa(tk+1) + "." + liText
  442. mapSort = append(mapSort, textLen)
  443. }
  444. })
  445. ul := doc.Find("ul")
  446. ul.Each(func(tk int, ul *goquery.Selection) {
  447. ulText := ul.Text()
  448. ulText = strings.Replace(ulText, " ", "", -1)
  449. if ulText != "" {
  450. textLen := strings.Index(docText, ulText)
  451. mapDoc[(strings.Index(docText, ulText))] = ulText
  452. mapSort = append(mapSort, textLen)
  453. }
  454. })
  455. if len(mapSort) == 0 {
  456. return
  457. } else {
  458. //排序
  459. sort.Ints(mapSort)
  460. var annotationHtml string
  461. for _, vSort := range mapSort {
  462. for k, v := range mapDoc {
  463. if k == vSort && v != "" {
  464. annotationHtml += v + "<br>"
  465. }
  466. }
  467. }
  468. annotationHtml = strings.TrimRight(annotationHtml, "<br>")
  469. annotationHtml = "<p>" + annotationHtml + "</p>"
  470. annotation = annotationHtml
  471. }
  472. return
  473. }
  474. // 处理产品内测展示规则
  475. func ProductInteriorHtml(bodyText string) (annotation string) {
  476. if bodyText == "" {
  477. return
  478. }
  479. sliceBody := strings.Split(bodyText, "</p>")
  480. annotation, _ = GetReportContentTextSub(sliceBody[0])
  481. return
  482. }
  483. // 解析研选内容中的核心观点
  484. func YxArticleAnnotation(article *models.ArticleListResp) (annotation string) {
  485. //如果不规范,就获取内容主体
  486. if strings.Count(article.Body, "<hr") == 0 {
  487. //如果内容不规范而且,还有图片,就把核心观点置空
  488. if article.BodyImg != "" {
  489. return
  490. }
  491. annotation, _ = GetReportContentTextSub(article.Body)
  492. return
  493. }
  494. body := strings.ReplaceAll(article.Body, "<strong>", "")
  495. body = strings.ReplaceAll(body, "</strong>", "")
  496. body = strings.ReplaceAll(body, "</ol>", "</div>")
  497. body = strings.ReplaceAll(body, "<ol>", "<div>")
  498. body = strings.ReplaceAll(body, "</li>", "</p>")
  499. body = strings.ReplaceAll(body, "<li>", "<p>")
  500. re, _ := regexp.Compile("<strong.*?>")
  501. body = re.ReplaceAllString(body, "")
  502. reLi, _ := regexp.Compile("<li.*?>")
  503. body = reLi.ReplaceAllString(body, "")
  504. var plus int
  505. coreIndex := strings.Index(body, "核心观点:")
  506. plus = 15
  507. if coreIndex == -1 {
  508. coreIndex = strings.Index(body, "核心观点:")
  509. plus = 13
  510. }
  511. if coreIndex == -1 {
  512. coreIndex = strings.Index(body, "核心观点")
  513. plus = 12
  514. }
  515. if coreIndex == -1 {
  516. coreIndex = strings.Index(body, "核心结论:")
  517. plus = 15
  518. }
  519. if coreIndex == -1 {
  520. coreIndex = strings.Index(body, "核心结论:")
  521. plus = 13
  522. }
  523. if coreIndex == -1 {
  524. coreIndex = strings.Index(body, "核心结论")
  525. plus = 12
  526. }
  527. endIndex := strings.Index(body, "<hr")
  528. if coreIndex != -1 && endIndex != -1 {
  529. body = body[coreIndex+plus : endIndex]
  530. }
  531. annotation = body
  532. return
  533. }
  534. // 获取研选类型的文章分类Id
  535. func GetYanXuanArticleTypeIds() (articleTypeIds string, err error) {
  536. var condition string
  537. condition = " AND is_show_yanx = 1 "
  538. listType, e := models.GetCygxArticleTypeListCondition(condition)
  539. if e != nil {
  540. err = errors.New("GetCygxArticleTypeListCondition, Err: " + e.Error())
  541. return
  542. }
  543. for _, v := range listType {
  544. articleTypeIds += strconv.Itoa(v.ArticleTypeId) + ","
  545. }
  546. articleTypeIds = strings.TrimRight(articleTypeIds, ",")
  547. if articleTypeIds == "" {
  548. err = errors.New("研选分类ID不能为空")
  549. return
  550. }
  551. return
  552. }
  553. // GetUserAticleCollectMap 获取用户收藏的文章ID
  554. func GetUserAticleCollectMap(user *models.WxUserItem) (respMap map[int]int, err error) {
  555. list, e := models.GetCygxArticleCollectListByUser(user.UserId)
  556. if e != nil {
  557. err = errors.New("GetCygxArticleCollectListByUser, Err: " + e.Error())
  558. return
  559. }
  560. articleMap := make(map[int]int)
  561. for _, v := range list {
  562. articleMap[v.ArticleId] = v.ArticleId
  563. }
  564. respMap = articleMap
  565. return
  566. }
  567. // 通过接解析带有Md5的文章链接获取文章ID
  568. func GetReportLinkToArticleid(reportLink string) (articleId int, err error) {
  569. defer func() {
  570. if err != nil {
  571. go utils.SendAlarmMsg("通过接解析带有Md5的文章链接获取文章ID失败"+err.Error(), 2)
  572. }
  573. }()
  574. var artMd5 string
  575. //处理Md5的
  576. strnum1 := strings.Index(reportLink, "id=")
  577. if strnum1 > 0 {
  578. sliceId := strings.Split(reportLink, "id=")
  579. if len(sliceId) > 1 {
  580. reportLink = sliceId[1]
  581. sliceMd5Id := strings.Split(reportLink, "&")
  582. artMd5 = sliceMd5Id[0]
  583. }
  584. if artMd5 != "" {
  585. detail, errArt := models.GetArticleDetailByIdMd5(artMd5)
  586. if errArt != nil && errArt.Error() != utils.ErrNoRow() {
  587. err = errArt
  588. return
  589. }
  590. if detail != nil {
  591. articleId = detail.ArticleId
  592. }
  593. }
  594. } else {
  595. //处理活动的
  596. linkList := strings.Split(reportLink, "/")
  597. if linkList[len(linkList)-1] != "" {
  598. linkArticleId, _ := strconv.Atoi(linkList[len(linkList)-1])
  599. if linkArticleId > 0 {
  600. articleInfo, errArt := models.GetArticleDetailById(linkArticleId)
  601. if errArt != nil && errArt.Error() != utils.ErrNoRow() {
  602. err = errArt
  603. return
  604. }
  605. if articleInfo != nil {
  606. articleId = articleInfo.ArticleId
  607. }
  608. }
  609. }
  610. }
  611. return
  612. }
  613. // GetArticleStockMap 获取个股标签所对应的文章ID
  614. func GetArticleStockMap() (mapResp map[string]int, err error) {
  615. defer func() {
  616. if err != nil {
  617. go utils.SendAlarmMsg("获取个股标签所对应的文章ID失败"+err.Error(), 2)
  618. }
  619. }()
  620. list, err := models.GetArticleStock()
  621. if err != nil && err.Error() != utils.ErrNoRow() {
  622. return
  623. }
  624. mapResp = make(map[string]int, 0)
  625. if len(list) > 0 {
  626. //一对一精准匹配
  627. for _, v := range list {
  628. sliceSubjects := strings.Split(v.Stock, "/")
  629. if len(sliceSubjects) > 0 {
  630. for _, vSubject := range sliceSubjects {
  631. sliceKuohao := strings.Split(vSubject, "(") //过滤括号
  632. sliceXiahuaxian := strings.Split(sliceKuohao[0], "-") //过滤下划线
  633. subject := sliceXiahuaxian[0]
  634. mapResp[subject] = v.ArticleId
  635. }
  636. }
  637. }
  638. }
  639. return
  640. }
  641. // 用户报告操作行为,模板消息推送
  642. func ArticleUserRemind(user *models.WxUserItem, articleDetail *models.ArticleDetail, source int) (err error) {
  643. defer func() {
  644. if err != nil {
  645. go utils.SendAlarmMsg("同步策略平台阅读数据失败", 2)
  646. go utils.SendEmail(utils.APPNAME+"【"+utils.RunMode+"】"+"失败提醒", "GetCeLueArticlePv ErrMsg:"+err.Error(), utils.EmailSendToUsers)
  647. }
  648. }()
  649. countUser, err := models.GetUserRemind(user.UserId)
  650. if err != nil {
  651. return err
  652. }
  653. if countUser == 0 {
  654. return err
  655. }
  656. var sourceMsg string
  657. if source == 1 {
  658. sourceMsg = "阅读报告"
  659. } else {
  660. sourceMsg = "收藏报告"
  661. }
  662. //获取销售手机号
  663. sellerItemQy, err := models.GetSellerByCompanyIdCheckFicc(user.CompanyId, 2)
  664. if err != nil && err.Error() != utils.ErrNoRow() {
  665. return err
  666. }
  667. if sellerItemQy != nil {
  668. openIdList, err := models.GetWxOpenIdByMobileList(sellerItemQy.Mobile)
  669. if err != nil {
  670. return err
  671. }
  672. var keyword1 string
  673. var keyword2 string
  674. keyword1 = articleDetail.Title
  675. keyword2 = fmt.Sprint("互动:", sourceMsg, ",", user.RealName, "--", user.CompanyName)
  676. SendWxMsgWithArticleUserRemind(keyword1, keyword2, openIdList, articleDetail.ArticleId)
  677. }
  678. return
  679. }
  680. // GetAiQianYanArtilceList 获取AI前沿几篇文章
  681. func GetAiQianYanArtilceList(startSize, pageSize int) (items []*models.HomeArticle, total int, err error) {
  682. defer func() {
  683. if err != nil {
  684. go utils.SendAlarmMsg("获取AI前沿几篇文章失败"+err.Error(), 2)
  685. }
  686. }()
  687. var condition string
  688. var pars []interface{}
  689. condition += ` AND title LIKE '%AI前沿%' AND publish_status = 1 ORDER BY publish_date DESC `
  690. articleList, e := models.GetCygxCygxArticleList(condition, pars, startSize, pageSize)
  691. if e != nil {
  692. err = errors.New("GetCygxCygxArticleList, Err: " + e.Error())
  693. return
  694. }
  695. total, e = models.GetCygxArticleCount(condition, pars)
  696. if e != nil {
  697. err = errors.New("GetCygxArticleCount, Err: " + e.Error())
  698. return
  699. }
  700. for _, v := range articleList {
  701. item := new(models.HomeArticle)
  702. item.ArticleId = v.ArticleId
  703. item.Title = v.Title
  704. item.Abstract = v.Abstract
  705. item.Annotation = v.Annotation
  706. item.PublishDate = v.PublishDate
  707. item.CategoryId = strconv.Itoa(v.CategoryId)
  708. item.Body = v.Body
  709. items = append(items, item)
  710. }
  711. return
  712. }
  713. // GetYxArticleIdMap 获取研选文章ID
  714. func GetYxArticleIdMap(articleIds []int) (mapResp map[int]bool) {
  715. var err error
  716. defer func() {
  717. if err != nil {
  718. go utils.SendAlarmMsg("获取研选文章ID失败,GetYxArticleIdMap"+err.Error(), 2)
  719. }
  720. }()
  721. var condition string
  722. var pars []interface{}
  723. condition = ` AND article_type_id > 0 `
  724. if len(articleIds) > 0 {
  725. condition += ` AND article_id IN (` + utils.GetOrmInReplace(len(articleIds)) + `)`
  726. pars = append(pars, articleIds)
  727. }
  728. articleList, e := models.GetArticleList(condition, pars)
  729. if e != nil {
  730. err = errors.New("GetArticleList, Err: " + e.Error())
  731. return
  732. }
  733. mapResp = make(map[int]bool, 0)
  734. for _, v := range articleList {
  735. mapResp[v.ArticleId] = true
  736. }
  737. return
  738. }