wechat_platform.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804
  1. package llm
  2. import (
  3. "encoding/json"
  4. "eta/eta_api/cache"
  5. "eta/eta_api/controllers"
  6. "eta/eta_api/models"
  7. "eta/eta_api/models/rag"
  8. "eta/eta_api/models/rag/request"
  9. "eta/eta_api/models/rag/response"
  10. "eta/eta_api/models/system"
  11. "eta/eta_api/services"
  12. "eta/eta_api/services/elastic"
  13. "eta/eta_api/utils"
  14. "fmt"
  15. "github.com/rdlucklib/rdluck_tools/paging"
  16. "html"
  17. "strings"
  18. "time"
  19. )
  20. // WechatPlatformController
  21. // @Description: 微信公众号管理
  22. type WechatPlatformController struct {
  23. controllers.BaseAuthController
  24. }
  25. // TagList
  26. // @Title 获取ppt列表
  27. // @Description 获取ppt列表接口
  28. // @Param PageSize query int true "每页数据条数"
  29. // @Param CurrentIndex query int true "当前页页码,从1开始"
  30. // @Param KeyWord query string true "搜索关键词"
  31. // @Success 200 {object} models.TagListResp
  32. // @router /tag/list [get]
  33. func (c *WechatPlatformController) TagList() {
  34. br := new(models.BaseResponse).Init()
  35. defer func() {
  36. c.Data["json"] = br
  37. c.ServeJSON()
  38. }()
  39. sysUser := c.SysUser
  40. if sysUser == nil {
  41. br.Msg = "请登录"
  42. br.ErrMsg = "请登录,SysUser Is Empty"
  43. return
  44. }
  45. pageSize, _ := c.GetInt("PageSize")
  46. currentIndex, _ := c.GetInt("CurrentIndex")
  47. keyWord := c.GetString("KeyWord")
  48. var startSize int
  49. if pageSize <= 0 {
  50. pageSize = utils.PageSize20
  51. }
  52. if currentIndex <= 0 {
  53. currentIndex = 1
  54. }
  55. startSize = utils.StartIndex(currentIndex, pageSize)
  56. var condition string
  57. var pars []interface{}
  58. if keyWord != "" {
  59. condition += fmt.Sprintf(` AND %s like ?`, rag.WechatPlatformColumns.Nickname)
  60. pars = append(pars, `%`+keyWord+`%`)
  61. }
  62. obj := new(rag.Tag)
  63. total, list, err := obj.GetPageListByCondition(condition, pars, startSize, pageSize)
  64. if err != nil {
  65. br.Msg = "获取失败"
  66. br.ErrMsg = "获取失败,Err:" + err.Error()
  67. return
  68. }
  69. page := paging.GetPaging(currentIndex, pageSize, total)
  70. resp := new(response.TagListResp)
  71. resp.Paging = page
  72. resp.List = list
  73. br.Ret = 200
  74. br.Success = true
  75. br.Msg = "获取成功"
  76. br.Data = resp
  77. }
  78. // Add
  79. // @Title 新增公众号
  80. // @Description 新增公众号
  81. // @Param request body request.AddWechatPlatformReq true "type json string"
  82. // @Success 200 Ret=200 新增成功
  83. // @router /wechat_platform/add [post]
  84. func (c *WechatPlatformController) Add() {
  85. br := new(models.BaseResponse).Init()
  86. defer func() {
  87. c.Data["json"] = br
  88. c.ServeJSON()
  89. }()
  90. var req request.AddWechatPlatformReq
  91. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  92. if err != nil {
  93. br.Msg = "参数解析异常!"
  94. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  95. return
  96. }
  97. req.Name = strings.TrimSpace(req.Name)
  98. if req.Name == "" {
  99. br.Msg = "请输入公众号名称"
  100. br.IsSendEmail = false
  101. return
  102. }
  103. req.Link = strings.TrimSpace(req.Link)
  104. if req.Link == "" {
  105. br.Msg = "请输入文章链接"
  106. br.IsSendEmail = false
  107. return
  108. }
  109. // 名称校验
  110. {
  111. var condition string
  112. var pars []interface{}
  113. condition += fmt.Sprintf(` AND %s = ?`, rag.WechatPlatformColumns.Nickname)
  114. pars = append(pars, req.Name)
  115. obj := new(rag.WechatPlatform)
  116. item, err := obj.GetByCondition(condition, pars)
  117. if err != nil && !utils.IsErrNoRow(err) {
  118. br.Msg = "公众号信息获取失败"
  119. br.ErrMsg = "公众号信息获取失败,Err:" + err.Error()
  120. return
  121. }
  122. if item.WechatPlatformId > 0 {
  123. br.Msg = "公众号名称重复"
  124. br.IsSendEmail = false
  125. return
  126. }
  127. }
  128. // 链接校验
  129. {
  130. var condition string
  131. var pars []interface{}
  132. condition += fmt.Sprintf(` AND %s = ?`, rag.WechatPlatformColumns.ArticleLink)
  133. pars = append(pars, req.Link)
  134. obj := new(rag.WechatPlatform)
  135. item, err := obj.GetByCondition(condition, pars)
  136. if err != nil && !utils.IsErrNoRow(err) {
  137. br.Msg = "公众号信息获取失败"
  138. br.ErrMsg = "公众号信息获取失败,Err:" + err.Error()
  139. return
  140. }
  141. if item.WechatPlatformId > 0 {
  142. br.Msg = "公众号链接重复"
  143. br.IsSendEmail = false
  144. return
  145. }
  146. }
  147. item := &rag.WechatPlatform{
  148. WechatPlatformId: 0,
  149. FakeId: "",
  150. Nickname: req.Name,
  151. Alias: "",
  152. RoundHeadImg: "",
  153. ServiceType: 0,
  154. Signature: "",
  155. Verified: 0,
  156. ArticleLink: req.Link,
  157. Enabled: 0,
  158. SysUserId: c.SysUser.AdminId,
  159. ModifyTime: time.Now(),
  160. CreateTime: time.Now(),
  161. }
  162. err = item.Add(req.TagIdList)
  163. if err != nil {
  164. br.Msg = "添加失败"
  165. br.ErrMsg = "添加失败,Err:" + err.Error()
  166. return
  167. }
  168. // 异步新增公众号
  169. go cache.AddWechatArticleOpToCache(item.WechatPlatformId, `add`)
  170. br.Ret = 200
  171. br.Success = true
  172. br.Msg = `添加成功`
  173. }
  174. // FollowList
  175. // @Title 我关注的接口
  176. // @Description 我关注的接口
  177. // @Param PageSize query int true "每页数据条数"
  178. // @Param CurrentIndex query int true "当前页页码,从1开始"
  179. // @Param KeyWord query string true "搜索关键词"
  180. // @Success 200 {object} []*rag.WechatPlatform
  181. // @router /wechat_platform/list/follow [get]
  182. func (c *WechatPlatformController) FollowList() {
  183. br := new(models.BaseResponse).Init()
  184. defer func() {
  185. c.Data["json"] = br
  186. c.ServeJSON()
  187. }()
  188. sysUser := c.SysUser
  189. if sysUser == nil {
  190. br.Msg = "请登录"
  191. br.ErrMsg = "请登录,SysUser Is Empty"
  192. return
  193. }
  194. pageSize, _ := c.GetInt("PageSize")
  195. currentIndex, _ := c.GetInt("CurrentIndex")
  196. keyWord := c.GetString("KeyWord")
  197. var startSize int
  198. if pageSize <= 0 {
  199. pageSize = utils.PageSize20
  200. }
  201. if currentIndex <= 0 {
  202. currentIndex = 1
  203. }
  204. startSize = utils.StartIndex(currentIndex, pageSize)
  205. var condition string
  206. var pars []interface{}
  207. if keyWord != "" {
  208. condition += fmt.Sprintf(` AND %s like ?`, rag.WechatPlatformColumns.Nickname)
  209. pars = append(pars, `%`+keyWord+`%`)
  210. }
  211. condition += fmt.Sprintf(` AND b.%s = ?`, rag.WechatPlatformUserMappingColumns.SysUserID)
  212. pars = append(pars, c.SysUser.AdminId)
  213. obj := new(rag.WechatPlatformUserMapping)
  214. list, err := obj.GetListByCondition(condition, pars, startSize, pageSize)
  215. if err != nil {
  216. br.Msg = "获取失败"
  217. br.ErrMsg = "获取失败,Err:" + err.Error()
  218. return
  219. }
  220. br.Ret = 200
  221. br.Success = true
  222. br.Msg = "获取成功"
  223. br.Data = list
  224. }
  225. // PublicList
  226. // @Title 公共列表
  227. // @Description 公共列表
  228. // @Param PageSize query int true "每页数据条数"
  229. // @Param CurrentIndex query int true "当前页页码,从1开始"
  230. // @Param KeyWord query string true "搜索关键词"
  231. // @Success 200 {object} models.WechatPlatformListResp
  232. // @router /wechat_platform/list/public [get]
  233. func (c *WechatPlatformController) PublicList() {
  234. br := new(models.BaseResponse).Init()
  235. defer func() {
  236. c.Data["json"] = br
  237. c.ServeJSON()
  238. }()
  239. sysUser := c.SysUser
  240. if sysUser == nil {
  241. br.Msg = "请登录"
  242. br.ErrMsg = "请登录,SysUser Is Empty"
  243. return
  244. }
  245. pageSize, _ := c.GetInt("PageSize")
  246. currentIndex, _ := c.GetInt("CurrentIndex")
  247. keyWord := c.GetString("KeyWord")
  248. var startSize int
  249. if pageSize <= 0 {
  250. pageSize = utils.PageSize20
  251. }
  252. if currentIndex <= 0 {
  253. currentIndex = 1
  254. }
  255. startSize = utils.StartIndex(currentIndex, pageSize)
  256. var condition string
  257. var pars []interface{}
  258. if keyWord != "" {
  259. condition += fmt.Sprintf(` AND %s like ?`, rag.WechatPlatformColumns.Nickname)
  260. pars = append(pars, `%`+keyWord+`%`)
  261. }
  262. obj := new(rag.WechatPlatformUserMapping)
  263. list, err := obj.GetListByCondition(condition, pars, startSize, 100000)
  264. if err != nil {
  265. br.Msg = "获取失败"
  266. br.ErrMsg = "获取失败,Err:" + err.Error()
  267. return
  268. }
  269. resp := make([]response.WechatPlatformPublicListResp, 0)
  270. if list != nil && len(list) > 0 {
  271. userIdList := make([]int, 0)
  272. uerIdMap := make(map[int]bool)
  273. userFollowIndexMap := make(map[int]int)
  274. for _, v := range list {
  275. if _, ok := uerIdMap[v.FollowUserId]; !ok {
  276. userIdList = append(userIdList, v.FollowUserId)
  277. uerIdMap[v.FollowUserId] = true
  278. }
  279. index, ok := userFollowIndexMap[v.FollowUserId]
  280. if !ok {
  281. userFollowIndexMap[v.FollowUserId] = len(resp)
  282. resp = append(resp, response.WechatPlatformPublicListResp{
  283. UserId: v.FollowUserId,
  284. List: []*rag.UserFollowWechatPlatform{v},
  285. })
  286. } else {
  287. resp[index].List = append(resp[index].List, v)
  288. }
  289. }
  290. userList, err := system.GetAdminListByIdList(userIdList)
  291. if err != nil {
  292. br.Msg = "获取失败"
  293. br.ErrMsg = "获取失败,Err:" + err.Error()
  294. return
  295. }
  296. userNameMap := make(map[int]*system.Admin)
  297. for _, v := range userList {
  298. userNameMap[v.AdminId] = v
  299. }
  300. for k, v := range resp {
  301. userInfo, ok := userNameMap[v.UserId]
  302. if !ok {
  303. continue
  304. }
  305. resp[k].Name = userInfo.RealName + `关注`
  306. }
  307. }
  308. br.Ret = 200
  309. br.Success = true
  310. br.Msg = "获取成功"
  311. br.Data = resp
  312. }
  313. // PublicList
  314. // @Title 公共列表
  315. // @Description 公共列表
  316. // @Param PageSize query int true "每页数据条数"
  317. // @Param CurrentIndex query int true "当前页页码,从1开始"
  318. // @Param KeyWord query string true "搜索关键词"
  319. // @Success 200 {object} models.WechatPlatformListResp
  320. // @router /wechat_platform/op [post]
  321. func (c *WechatPlatformController) Op() {
  322. br := new(models.BaseResponse).Init()
  323. defer func() {
  324. c.Data["json"] = br
  325. c.ServeJSON()
  326. }()
  327. sysUser := c.SysUser
  328. if sysUser == nil {
  329. br.Msg = "请登录"
  330. br.ErrMsg = "请登录,SysUser Is Empty"
  331. return
  332. }
  333. var req request.OpWechatPlatformReq
  334. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  335. if err != nil {
  336. br.Msg = "参数解析异常!"
  337. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  338. return
  339. }
  340. if req.WechatPlatformId <= 0 {
  341. br.Msg = "参数错误"
  342. return
  343. }
  344. if req.Status < 0 || req.Status > 1 {
  345. br.Msg = "参数错误"
  346. return
  347. }
  348. obj := rag.WechatPlatform{}
  349. wechatPlatform, err := obj.GetById(req.WechatPlatformId)
  350. if err != nil {
  351. br.Msg = "修改失败"
  352. br.ErrMsg = "修改失败,Err:" + err.Error()
  353. if utils.IsErrNoRow(err) {
  354. br.Msg = "公众号不存在"
  355. br.IsSendEmail = false
  356. }
  357. return
  358. }
  359. wechatPlatform.Enabled = req.Status
  360. wechatPlatform.ModifyTime = time.Now()
  361. err = wechatPlatform.Update([]string{"enabled", `modify_time`})
  362. if err != nil {
  363. br.Msg = "修改失败"
  364. br.ErrMsg = "修改失败,Err:" + err.Error()
  365. return
  366. }
  367. // 异步处理公众号下面的文章
  368. go services.AddOrEditEsWechatPlatformId(wechatPlatform.WechatPlatformId)
  369. br.Ret = 200
  370. br.Success = true
  371. br.Msg = "修改成功"
  372. }
  373. // Refresh
  374. // @Title 公共列表
  375. // @Description 公共列表
  376. // @Success 200 {object} models.WechatPlatformListResp
  377. // @router /wechat_platform/refresh [post]
  378. func (c *WechatPlatformController) Refresh() {
  379. br := new(models.BaseResponse).Init()
  380. defer func() {
  381. c.Data["json"] = br
  382. c.ServeJSON()
  383. }()
  384. sysUser := c.SysUser
  385. if sysUser == nil {
  386. br.Msg = "请登录"
  387. br.ErrMsg = "请登录,SysUser Is Empty"
  388. return
  389. }
  390. var req request.RefreshWechatPlatformReq
  391. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  392. if err != nil {
  393. br.Msg = "参数解析异常!"
  394. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  395. return
  396. }
  397. if req.WechatPlatformId <= 0 {
  398. br.Msg = "参数错误"
  399. return
  400. }
  401. obj := rag.WechatPlatform{}
  402. wechatPlatform, err := obj.GetById(req.WechatPlatformId)
  403. if err != nil {
  404. br.Msg = "修改失败"
  405. br.ErrMsg = "修改失败,Err:" + err.Error()
  406. if utils.IsErrNoRow(err) {
  407. br.Msg = "公众号不存在"
  408. br.IsSendEmail = false
  409. }
  410. return
  411. }
  412. if wechatPlatform.FakeId != `` {
  413. br.Msg = "公众号已添加成功"
  414. br.ErrMsg = "公众号已添加成功"
  415. br.IsSendEmail = false
  416. return
  417. }
  418. go cache.AddWechatArticleOpToCache(wechatPlatform.WechatPlatformId, `add`)
  419. br.Ret = 200
  420. br.Success = true
  421. br.Msg = "刷新成功"
  422. }
  423. // ArticleList
  424. // @Title 我关注的接口
  425. // @Description 我关注的接口
  426. // @Param PageSize query int true "每页数据条数"
  427. // @Param CurrentIndex query int true "当前页页码,从1开始"
  428. // @Param WechatPlatformId query int true "微信公众号id"
  429. // @Param KeyWord query string true "搜索关键词"
  430. // @Success 200 {object} []*rag.WechatPlatform
  431. // @router /wechat_platform/article/list [get]
  432. func (c *WechatPlatformController) ArticleList() {
  433. br := new(models.BaseResponse).Init()
  434. defer func() {
  435. c.Data["json"] = br
  436. c.ServeJSON()
  437. }()
  438. sysUser := c.SysUser
  439. if sysUser == nil {
  440. br.Msg = "请登录"
  441. br.ErrMsg = "请登录,SysUser Is Empty"
  442. return
  443. }
  444. pageSize, _ := c.GetInt("PageSize")
  445. currentIndex, _ := c.GetInt("CurrentIndex")
  446. wechatPlatformId, _ := c.GetInt("WechatPlatformId")
  447. keyWord := c.GetString("KeyWord")
  448. var startSize int
  449. if pageSize <= 0 {
  450. pageSize = utils.PageSize20
  451. }
  452. if currentIndex <= 0 {
  453. currentIndex = 1
  454. }
  455. startSize = utils.StartIndex(currentIndex, pageSize)
  456. //var condition string
  457. //var pars []interface{}
  458. //
  459. //if keyWord != "" {
  460. // condition += fmt.Sprintf(` AND (b.%s like ? or a.%s like ? ) `, rag.WechatPlatformColumns.Nickname, rag.WechatArticleColumns.Title)
  461. // pars = append(pars, `%`+keyWord+`%`, `%`+keyWord+`%`)
  462. //}
  463. //
  464. //if wechatPlatformId > 0 {
  465. // condition += fmt.Sprintf(` AND a.%s = ?`, rag.WechatArticleColumns.WechatPlatformID)
  466. // pars = append(pars, wechatPlatformId)
  467. //}
  468. //
  469. //condition += fmt.Sprintf(` AND b.%s = ? `, rag.WechatPlatformColumns.Enabled)
  470. //pars = append(pars, 1)
  471. var total int
  472. viewList := make([]rag.WechatArticleView, 0)
  473. if keyWord == `` {
  474. var condition string
  475. var pars []interface{}
  476. if keyWord != "" {
  477. condition += fmt.Sprintf(` AND (b.%s like ? or a.%s like ? ) `, rag.WechatPlatformColumns.Nickname, rag.WechatArticleColumns.Title)
  478. pars = append(pars, `%`+keyWord+`%`, `%`+keyWord+`%`)
  479. }
  480. if wechatPlatformId > 0 {
  481. condition += fmt.Sprintf(` AND a.%s = ?`, rag.WechatArticleColumns.WechatPlatformID)
  482. pars = append(pars, wechatPlatformId)
  483. }
  484. //condition += fmt.Sprintf(` AND b.%s = ? `, rag.WechatPlatformColumns.Enabled)
  485. //pars = append(pars, 1)
  486. obj := new(rag.WechatArticle)
  487. tmpTotal, list, err := obj.GetPageListByPlatformCondition(condition, pars, startSize, pageSize)
  488. if err != nil {
  489. br.Msg = "获取失败"
  490. br.ErrMsg = "获取失败,Err:" + err.Error()
  491. return
  492. }
  493. total = tmpTotal
  494. if list != nil && len(list) > 0 {
  495. viewList = obj.ArticleAndPlatformListToViewList(list)
  496. }
  497. } else {
  498. sortMap := map[string]string{
  499. //"ArticleCreateTime": "desc",
  500. //"WechatArticleId": "desc",
  501. }
  502. tmpTotal, list, err := elastic.WechatArticleEsSearch(keyWord, wechatPlatformId, startSize, pageSize, sortMap)
  503. if err != nil {
  504. br.Msg = "获取失败"
  505. br.ErrMsg = "获取失败,Err:" + err.Error()
  506. return
  507. }
  508. total = int(tmpTotal)
  509. if list != nil && len(list) > 0 {
  510. viewList = list[0].ArticleAndPlatformListToViewList(list)
  511. }
  512. }
  513. // 标签处理
  514. {
  515. wechatPlateIdList := make([]int, 0)
  516. for _, v := range viewList {
  517. wechatPlateIdList = append(wechatPlateIdList, v.WechatPlatformId)
  518. }
  519. obj := rag.WechatPlatformTagMapping{}
  520. mappingList, err := obj.GetWechatPlatformTagListByWechatPlatformIdList(wechatPlateIdList)
  521. if err != nil {
  522. br.Msg = "获取失败"
  523. br.ErrMsg = "获取失败,Err:" + err.Error()
  524. return
  525. }
  526. wechatPlatformIdMap := make(map[int]*rag.WechatPlatformTagItem)
  527. for _, v := range mappingList {
  528. wechatPlatformIdMap[v.WechatPlatformId] = v
  529. }
  530. for k, v := range viewList {
  531. item, ok := wechatPlatformIdMap[v.WechatPlatformId]
  532. if !ok {
  533. continue
  534. }
  535. viewList[k].TagName = item.TagName
  536. viewList[k].TagId = item.TagId
  537. }
  538. }
  539. page := paging.GetPaging(currentIndex, pageSize, total)
  540. resp := response.WechatArticleListListResp{
  541. List: viewList,
  542. Paging: page,
  543. }
  544. br.Ret = 200
  545. br.Success = true
  546. br.Msg = "获取成功"
  547. br.Data = resp
  548. }
  549. // ArticleList
  550. // @Title 我关注的接口
  551. // @Description 我关注的接口
  552. // @Param WechatArticleId query int true "文章id"
  553. // @Success 200 {object} []*rag.WechatArticle
  554. // @router /wechat_platform/article/detail [get]
  555. func (c *WechatPlatformController) ArticleDetail() {
  556. br := new(models.BaseResponse).Init()
  557. defer func() {
  558. c.Data["json"] = br
  559. c.ServeJSON()
  560. }()
  561. sysUser := c.SysUser
  562. if sysUser == nil {
  563. br.Msg = "请登录"
  564. br.ErrMsg = "请登录,SysUser Is Empty"
  565. return
  566. }
  567. wechatArticleId, _ := c.GetInt("WechatArticleId")
  568. if wechatArticleId <= 0 {
  569. br.Msg = "请选择文章"
  570. br.IsSendEmail = false
  571. return
  572. }
  573. obj := new(rag.WechatArticle)
  574. item, err := obj.GetById(wechatArticleId)
  575. if err != nil {
  576. br.Msg = "获取失败"
  577. br.ErrMsg = "获取失败,Err:" + err.Error()
  578. return
  579. }
  580. if item.IsDeleted == 1 {
  581. br.Msg = "文章已删除"
  582. br.IsSendEmail = false
  583. return
  584. }
  585. resp := item.ToView()
  586. resp.Content = html.UnescapeString(item.Content)
  587. // 获取摘要信息
  588. {
  589. abstractObj := rag.WechatArticleAbstract{}
  590. abstractItem, err := abstractObj.GetByWechatArticleId(wechatArticleId)
  591. if err != nil && !utils.IsErrNoRow(err) {
  592. br.Msg = "获取失败"
  593. br.ErrMsg = "获取失败,Err:" + err.Error()
  594. return
  595. }
  596. resp.Abstract = abstractItem.Content
  597. }
  598. br.Ret = 200
  599. br.Success = true
  600. br.Msg = "获取成功"
  601. br.Data = resp
  602. }
  603. // ArticleList
  604. // @Title 我关注的接口
  605. // @Description 我关注的接口
  606. // @Param WechatArticleId query int true "文章id"
  607. // @Success 200 {object} []*rag.WechatPlatform
  608. // @router /wechat_platform/article/del [get]
  609. func (c *WechatPlatformController) ArticleDel() {
  610. br := new(models.BaseResponse).Init()
  611. defer func() {
  612. c.Data["json"] = br
  613. c.ServeJSON()
  614. }()
  615. sysUser := c.SysUser
  616. if sysUser == nil {
  617. br.Msg = "请登录"
  618. br.ErrMsg = "请登录,SysUser Is Empty"
  619. return
  620. }
  621. wechatArticleId, _ := c.GetInt("WechatArticleId")
  622. if wechatArticleId <= 0 {
  623. br.Msg = "请选择文章"
  624. br.IsSendEmail = false
  625. return
  626. }
  627. obj := new(rag.WechatArticle)
  628. item, err := obj.GetById(wechatArticleId)
  629. if err != nil {
  630. br.Msg = "获取失败"
  631. br.ErrMsg = "获取失败,Err:" + err.Error()
  632. return
  633. }
  634. if item.IsDeleted == 1 {
  635. br.Msg = "文章已删除"
  636. br.IsSendEmail = false
  637. return
  638. }
  639. item.IsDeleted = 1
  640. err = item.Update([]string{"is_deleted"})
  641. if err != nil {
  642. br.Msg = "删除失败"
  643. br.ErrMsg = "删除失败,Err:" + err.Error()
  644. return
  645. }
  646. // 修改ES信息
  647. services.AddOrEditEsWechatArticle(item.WechatArticleId)
  648. br.Ret = 200
  649. br.Success = true
  650. br.Msg = "删除成功"
  651. }
  652. //func init() {
  653. // //obj := rag.WechatPlatform{}
  654. // //item, _ := obj.GetByID(2)
  655. // //fmt.Println(llm.BeachAddWechatPlatform(item))
  656. //
  657. // obj := rag.WechatArticle{}
  658. // //item, _ := obj.GetById(30)
  659. // list, _ := obj.GetListByCondition(``, ` `, []interface{}{}, 0, 1000)
  660. // //llm.ArticleToTmpFile(item.TextContent)
  661. // for _, item := range list {
  662. // //llm.ArticleToKnowledge(item)
  663. // llm.GenerateArticleAbstract(item)
  664. // }
  665. //}
  666. //func init() {
  667. // //obj := rag.WechatPlatform{}
  668. // //item, _ := obj.GetByID(2)
  669. // //fmt.Println(llm.BeachAddWechatPlatform(item))
  670. //
  671. // //obj := rag.WechatArticle{}
  672. // //list, _ := obj.GetListByCondition(`wechat_article_id,content`, ` `, []interface{}{}, 0, 1)
  673. // ////obj := rag.WechatPlatform{}
  674. // ////list, _ := obj.GetListByCondition(` AND wechat_platform_id !=1 `, []interface{}{}, 0, 100)
  675. // ////llm.ArticleToTmpFile(item.TextContent)
  676. // //for _, item := range list {
  677. // // //llm.ArticleToKnowledge(item)
  678. // // services.ReGenerateArticleAbstract(item)
  679. // //}
  680. //
  681. // // 重新生成摘要
  682. // {
  683. // obj := rag.WechatArticle{}
  684. // list, _ := obj.GetListByCondition(``, ` AND text_content !='' AND abstract_status = 0`, []interface{}{}, 0, 300)
  685. // for _, item := range list {
  686. // services.GenerateArticleAbstract(item)
  687. // }
  688. // }
  689. //
  690. // //// 删除摘要向量库
  691. // //{
  692. // // obj := rag.WechatArticleAbstract{}
  693. // // list, _ := obj.GetListByCondition(`vector_key,wechat_article_abstract_id`, ` AND wechat_article_id in (25,27) `, []interface{}{}, 0, 10000)
  694. // // fmt.Println(services.DelDoc(list))
  695. // //}
  696. //
  697. // fmt.Println("修复结束")
  698. //}
  699. //func init() {
  700. // //// 微信文章加到es
  701. // //{
  702. // // obj := rag.WechatArticle{}
  703. // // list, _ := obj.GetListByCondition(` wechat_article_id `, ` `, []interface{}{}, 0, 10000)
  704. // // total := len(list)
  705. // // for k, item := range list {
  706. // // fmt.Println(k, "/", total)
  707. // // services.AddOrEditEsWechatArticle(item.WechatArticleId)
  708. // // }
  709. // //
  710. // // fmt.Println("结束了")
  711. // //}
  712. //
  713. // //// 微信文章加到es
  714. // {
  715. // obj := rag.WechatArticleAbstract{}
  716. // list, _ := obj.GetListByCondition(` wechat_article_abstract_id `, ` `, []interface{}{}, 0, 10000)
  717. // total := len(list)
  718. // for k, item := range list {
  719. // fmt.Println(k, "/", total)
  720. // services.AddOrEditEsWechatArticleAbstract(item.WechatArticleAbstractId)
  721. // }
  722. //
  723. // fmt.Println("结束了")
  724. // }
  725. //
  726. //}