package models import ( "eta_gn/eta_api/global" "eta_gn/eta_api/models/report" "eta_gn/eta_api/utils" "time" ) // ReportChapter 报告章节 type ReportChapter struct { ReportChapterId int `gorm:"column:report_chapter_id;primaryKey;autoIncrement:true;not null" description:"报告章节ID"` ReportId int `gorm:"column:report_id" description:"报告ID"` ReportType string `gorm:"column:report_type" description:"报告类型 day-晨报 week-周报"` ClassifyIdFirst int `gorm:"column:classify_id_first" description:"一级分类id"` ClassifyNameFirst string `gorm:"column:classify_name_first" description:"一级分类名称"` TypeId int `gorm:"column:type_id" description:"品种ID"` TypeName string `gorm:"column:type_name" description:"品种名称"` Title string `gorm:"column:title" description:"标题"` Abstract string `gorm:"column:abstract" description:"摘要"` AddType int `gorm:"column:add_type" description:"新增方式：1:新增报告，2:继承报告"` Author string `gorm:"column:author" description:"作者"` Content string `gorm:"column:content" description:"内容"` ContentSub string `gorm:"column:content_sub" description:"内容前两个章节"` Stage int `gorm:"column:stage" description:"期数"` Trend string `gorm:"column:trend" description:"趋势观点"` Sort int `gorm:"column:sort" description:"排序: 数值越小越靠前"` IsEdit int `gorm:"column:is_edit" description:"是否已编辑 0-待编辑 1-已编辑"` PublishState int `gorm:"column:publish_state" description:"发布状态 1-待发布，2-已发布"` PublishTime time.Time `gorm:"column:publish_time" description:"发布时间"` VideoUrl string `gorm:"column:video_url" description:"音频文件URL"` VideoName string `gorm:"column:video_name" description:"音频文件名称"` VideoPlaySeconds string `gorm:"column:video_play_seconds" description:"音频播放时长"` VideoSize string `gorm:"column:video_size" description:"音频文件大小，单位M"` VideoKind int `gorm:"column:video_kind" description:"音频生成方式:1,手动上传,2:自动生成"` CreateTime time.Time `gorm:"column:create_time" description:"创建时间"` ModifyTime time.Time `gorm:"column:modify_time" description:"修改时间"` OriginalVideoUrl string `gorm:"column:original_video_url" description:"原始音频文件URL"` ContentStruct string `gorm:"column:content_struct" description:"内容组件"` LastModifyAdminId int `gorm:"column:last_modify_admin_id" description:"最后更新人ID"` LastModifyAdminName string `gorm:"column:last_modify_admin_name" description:"最后更新人姓名"` ContentModifyTime time.Time `gorm:"column:content_modify_time" description:"内容更新时间"` ReportLayout int8 `gorm:"column:report_layout" description:"报告布局，1：常规布局，2：智能布局。默认：1"` ReportCreateTime time.Time `gorm:"column:report_create_time" description:"报告时间创建时间"` VoiceGenerateType int `gorm:"column:voice_generate_type" description:"音频生成方式，0：系统生成，1：人工上传"` } // ReportChapterItem 报告章节详情 type ReportChapterItem struct { ReportChapterId int `gorm:"column:report_chapter_id;primaryKey;autoIncrement:true;not null" description:"报告章节ID"` ReportId int `gorm:"column:report_id" description:"报告ID"` ReportType string `gorm:"column:report_type" description:"报告类型 day-晨报 week-周报"` ClassifyIdFirst int `gorm:"column:classify_id_first" description:"一级分类id"` ClassifyNameFirst string `gorm:"column:classify_name_first" description:"一级分类名称"` TypeId int `gorm:"column:type_id" description:"品种ID"` TypeName string `gorm:"column:type_name" description:"品种名称"` Title string `gorm:"column:title" description:"标题"` Abstract string `gorm:"column:abstract" description:"摘要"` AddType int `gorm:"column:add_type" description:"新增方式：1:新增报告，2:继承报告"` Author string `gorm:"column:author" description:"作者"` Content string `gorm:"column:content" description:"内容"` ContentSub string `gorm:"column:content_sub" description:"内容前两个章节"` Stage int `gorm:"column:stage" description:"期数"` Trend string `gorm:"column:trend" description:"趋势观点"` Sort int `gorm:"column:sort" description:"排序: 数值越小越靠前"` IsEdit int `gorm:"column:is_edit" description:"是否已编辑 0-待编辑 1-已编辑"` PublishState int `gorm:"column:publish_state" description:"发布状态 1-待发布，2-已发布"` PublishTime string `gorm:"column:publish_time" description:"发布时间"` VideoUrl string `gorm:"column:video_url" description:"音频文件URL"` VideoName string `gorm:"column:video_name" description:"音频文件名称"` VideoPlaySeconds string `gorm:"column:video_play_seconds" description:"音频播放时长"` VideoSize string `gorm:"column:video_size" description:"音频文件大小，单位M"` VideoKind int `gorm:"column:video_kind" description:"音频生成方式:1,手动上传,2:自动生成"` CreateTime string `gorm:"column:create_time" description:"创建时间"` ModifyTime string `gorm:"column:modify_time" description:"修改时间"` OriginalVideoUrl string `gorm:"column:original_video_url" description:"原始音频文件URL"` ContentStruct string `gorm:"column:content_struct" description:"内容组件"` LastModifyAdminId int `gorm:"column:last_modify_admin_id" description:"最后更新人ID"` LastModifyAdminName string `gorm:"column:last_modify_admin_name" description:"最后更新人姓名"` ContentModifyTime string `gorm:"column:content_modify_time" description:"内容更新时间"` ReportLayout int8 `gorm:"column:report_layout" description:"报告布局，1：常规布局，2：智能布局。默认：1"` ReportCreateTime string `gorm:"column:report_create_time" description:"报告时间创建时间"` } // ReportChapterItemResp // @Description: 章节详情（带有一些额外的数据） type ReportChapterItemResp struct { ReportChapterItem GrandAdminIdList []int `description:"授权的用户id列表"` PermissionIdList []int `description:"关联的品种id列表"` CanEdit bool `description:"是否可编辑"` Editor string `description:"编辑人"` HeadImg string `description:"报告头图地址"` EndImg string `description:"报告尾图地址"` HeadStyle string `description:"版头样式"` EndStyle string `description:"版尾样式"` } type ReportChapterResp struct { ReportChapterId int `description:"报告章节ID"` ReportId int `description:"报告ID"` ReportType string `description:"报告类型 day-晨报 week-周报"` TypeId int `description:"品种ID"` TypeName string `description:"品种名称"` TypeEditImg string `description:"后台编辑时的图片"` Title string `description:"标题"` Abstract string `description:"摘要"` Author string `description:"作者"` Content string `description:"内容"` ContentSub string `description:"内容前两个章节"` Stage int `description:"期数"` Trend string `description:"趋势观点"` Sort int `description:"排序: 数值越小越靠前"` IsEdit int `description:"是否已编辑 0-待编辑 1-已编辑"` PublishState int `description:"发布状态 1-待发布，2-已发布"` VideoUrl string `description:"音频文件URL"` VideoName string `description:"音频文件名称"` VideoPlaySeconds string `description:"音频播放时长"` VideoSize string `description:"音频文件大小，单位M"` VideoKind int `description:"音频生成方式:1,手动上传,2:自动生成"` PublishTime string `description:"发布时间"` CreateTime string `description:"创建时间"` ModifyTime string `description:"修改时间"` GrandAdminIdList []int `description:"授权的用户id列表"` PermissionIdList []int `description:"关联的品种id列表"` CanEdit bool `description:"是否可编辑"` Editor string `description:"编辑人"` IsAuth bool `description:"是否有权限"` } // GetChapterListByReportId 根据ReportId获取章节列表 func GetChapterListByReportId(reportId int) (list []*ReportChapter, err error) { sql := ` SELECT * FROM report_chapter WHERE report_id = ? ORDER BY sort ASC` err = global.DmSQL["rddp"].Raw(sql, reportId).Find(&list).Error return } // GetPublishedChapterListByReportId 根据ReportId获取已发布章节列表 func GetPublishedChapterListByReportId(reportId int) (list []*ReportChapter, err error) { sql := ` SELECT * FROM report_chapter WHERE report_id = ? AND publish_state = 2 ORDER BY sort ASC` err = global.DmSQL["rddp"].Raw(sql, reportId).Find(&list).Error return } // AddReportChapterReq // @Description: 新增报告章节请求体 type AddReportChapterReq struct { ReportId int `description:"报告ID"` Title string `description:"标题"` PermissionIdList []int `description:"报告关联的品种权限"` AdminIdList []int `description:"授权的编辑人id列表"` } // EditReportChapterReq 编辑报告章节请求体 type EditReportChapterReq struct { ReportChapterId int `description:"报告章节ID"` Title string `description:"标题"` AddType int `description:"新增方式：1:新增报告，2:继承报告"` Author string `description:"作者"` Content string `description:"内容"` TickerList []EditTickList `description:"指标信息"` CreateTime string `description:"发布时间"` VideoUrl string `description:"音频文件URL"` VideoName string `description:"音频文件名称"` VideoPlaySeconds string `description:"音频播放时长"` VideoSize string `description:"音频文件大小，单位M"` // 以下是智能研报相关 ContentStruct string `description:"内容组件"` HeadImg string `description:"报告头图地址"` EndImg string `description:"报告尾图地址"` CanvasColor string `description:"画布颜色"` HeadResourceId int `description:"版头资源ID"` EndResourceId int `description:"版尾资源ID"` IsManualSave bool `description:"是否手动保存：true：手动 false：自动"` } type EditTickList struct { Label string Ticker string Sort int } // GetReportChapterInfoById 根据主键获取报告章节 func GetReportChapterInfoById(reportChapterId int) (item *ReportChapter, err error) { sql := ` SELECT * FROM report_chapter WHERE report_chapter_id = ? ` err = global.DmSQL["rddp"].Raw(sql, reportChapterId).First(&item).Error return } // GetReportChapterItemById // @Description: 根据主键获取报告章节（时间格式为字符串的数据） // @author: Roc // @datetime 2024-06-27 14:10:29 // @param reportChapterId int // @return item *ReportChapterItem // @return err error func GetReportChapterItemById(reportChapterId int) (item *ReportChapterItem, err error) { sql := ` SELECT * FROM report_chapter WHERE report_chapter_id = ? ` err = global.DmSQL["rddp"].Raw(sql, reportChapterId).First(&item).Error return } // Add // @Description: 新增章节报告 // @author: Roc // @receiver chapterInfo // @datetime 2024-06-04 15:14:41 // @return err error func (chapterChapterInfo *ReportChapter) Add() (err error) { err = global.DmSQL["rddp"].Create(chapterChapterInfo).Error return } // UpdateChapter 更新报表章节 func (chapterChapterInfo *ReportChapter) UpdateChapter(cols []string) (err error) { err = global.DmSQL["rddp"].Select(cols).Updates(chapterChapterInfo).Error return } // EditChapterTrendTagReq 编辑章节趋势标签请求体 type EditChapterTrendTagReq struct { ReportChapterId int `description:"章节ID"` Trend string `description:"趋势"` } // UpdateChapterAndTicker 更新章节及ticker func UpdateChapterAndTicker(reportInfo *Report, chapterInfo *ReportChapter, updateCols []string, tickerList []*ReportChapterTicker) (err error) { // 更新报告的最近编辑人信息 if err = reportInfo.UpdateReport([]string{"LastModifyAdminId", "LastModifyAdminName", "ModifyTime"}); err != nil { return } // 更新章节 if err = chapterInfo.UpdateChapter(updateCols); err != nil { return } // 清空并新增章节ticker if err = ClearReportChapterTicker(chapterInfo.ReportChapterId); err != nil { return } tickerLen := len(tickerList) if tickerLen > 0 { for i := 0; i < tickerLen; i++ { _, tmpErr := InsertChapterTicker(tickerList[i]) if tmpErr != nil { return } } } return } // ReportChapterVideoList 报告章节音频列表 type ReportChapterVideoList struct { ReportId int `gorm:"column:report_id"` //`description:"报告ID"` ReportChapterId int `gorm:"column:report_chapter_id"` //`description:"报告章节ID"` VideoUrl string `gorm:"column:video_url"` //`description:"音频文件URL"` VideoName string `gorm:"column:video_name"` //`description:"音频文件名称"` VideoPlaySeconds string `gorm:"column:video_play_seconds"` //`description:"音频播放时长"` } // GetReportChapterVideoList 获取报告章节音频列表 func GetReportChapterVideoList(reportId int) (list []*ReportChapterVideoList, err error) { sql := ` SELECT report_id, report_chapter_id, video_url, video_name, video_play_seconds FROM report_chapter WHERE report_id = ? AND publish_state = 2 AND video_url != "" ORDER BY report_chapter_id ASC ` err = global.DmSQL["rddp"].Raw(sql, reportId).Find(&list).Error return } // GetReportChapterVideoListByReportIds 根据报告ID集合获取报告章节音频列表 func GetReportChapterVideoListByReportIds(reportIds []int) (list []*ReportChapterVideoList, err error) { if len(reportIds) == 0 { return } sql := ` SELECT report_id, report_chapter_id, video_url, video_name, video_play_seconds FROM report_chapter WHERE report_id IN (` + utils.GetOrmInReplace(len(reportIds)) + `) AND publish_state = 2 AND video_url != '' ORDER BY report_chapter_id ASC ` err = global.DmSQL["rddp"].Raw(sql, reportIds).Find(&list).Error return } // GetReportChapterVideoListByChapterIds 根据章节ID集合获取报告章节音频列表 func GetReportChapterVideoListByChapterIds(chapterIds []int) (list []*ReportChapterVideoList, err error) { if len(chapterIds) == 0 { return } sql := ` SELECT report_id, report_chapter_id, video_url, video_name, video_play_seconds FROM report_chapter WHERE report_chapter_id IN (` + utils.GetOrmInReplace(len(chapterIds)) + `) AND publish_state = 2 ORDER BY report_chapter_id ASC ` err = global.DmSQL["rddp"].Raw(sql, chapterIds).Find(&list).Error return } // PublishReportChapterReq 发布报告章节请求体 type PublishReportChapterReq struct { ReportChapterId int `description:"报告章节ID"` PublishReport int `description:"是否同时发布报告"` } // GetUnPublishedChapterList // @Description: 获取报告未发布的章节列表 // @author: Roc // @datetime 2024-06-14 15:59:23 // @param reportId int // @return list []*ReportChapter // @return err error func GetUnPublishedChapterList(reportId int) (list []*ReportChapter, err error) { sql := ` SELECT report_chapter_id,report_id,title FROM report_chapter WHERE report_id = ? AND publish_state = 1 ORDER BY sort ASC` err = global.DmSQL["rddp"].Raw(sql, reportId).Find(&list).Error return } // GetChapterListByChapterIds 根据ReportId获取章节列表 func GetChapterListByChapterIds(chapterIds []int) (list []*ReportChapter, err error) { if len(chapterIds) == 0 { return } sql := ` SELECT * FROM report_chapter WHERE report_chapter_id IN (` + utils.GetOrmInReplace(len(chapterIds)) + `) ORDER BY sort ASC` err = global.DmSQL["rddp"].Raw(sql, chapterIds).Find(&list).Error return } // UpdateReportChapterTypeNameByTypeId 更新章节类型名称 func UpdateReportChapterTypeNameByTypeId(typeId int, typeName string) (err error) { sql := `UPDATE report_chapter SET type_name = ? WHERE type_id = ?` err = global.DmSQL["rddp"].Exec(sql, typeName, typeId).Error return } // CountReportChapterByTypeId 通过章节类型ID获取章节数 func CountReportChapterByTypeId(typeId int) (count int, err error) { sql := ` SELECT COUNT(1) AS ct FROM report_chapter WHERE type_id = ?` err = global.DmSQL["rddp"].Raw(sql, typeId).Scan(&count).Error return } // AddReportChapter // @Description: 待添加的报告章节 type AddReportChapter struct { ReportChapter *ReportChapter GrantList []*report.ReportChapterGrant GrantPermissionList []*report.ReportChapterPermissionMapping } // EditReportChapterBaseInfoAndPermissionReq // @Description: 编辑报告章节的基础信息请求 type EditReportChapterBaseInfoAndPermissionReq struct { ReportChapterId int `description:"报告章节ID"` Title string `description:"标题"` PermissionIdList []int `description:"报告关联的品种权限"` AdminIdList []int `description:"授权的编辑人id列表"` } // GetReportChapterIdList // @Description: 获取报告的所有章节id列表 // @author: Roc // @datetime 2024-06-05 11:09:40 // @param reportId int // @return list []int // @return err error func GetReportChapterIdList(reportId int) (list []int, err error) { sql := ` SELECT report_chapter_id FROM report_chapter WHERE report_id = ? ORDER BY report_chapter_id ASC ` err = global.DmSQL["rddp"].Raw(sql, reportId).Scan(&list).Error return } // ReportChapterMoveReq // @Description: 报告章节移动请求 type ReportChapterMoveReq struct { ReportChapterId int `description:"报告章节id"` PrevReportChapterId int `description:"上一个兄弟节点报告章节id"` NextReportChapterId int `description:"下一个兄弟节点报告章节id"` } // GetReportChapterById // @Description: 获取具体章节 // @author: Roc // @receiver r // @datetime 2024-06-06 09:32:40 // @param reportChapterId int // @return item *ReportChapter // @return err error func (chapterChapterInfo *ReportChapter) GetReportChapterById(reportChapterId int) (item *ReportChapter, err error) { sql := ` SELECT * FROM report_chapter WHERE report_chapter_id = ?` err = global.DmSQL["rddp"].Raw(sql, reportChapterId).First(&item).Error return } // UpdateReportChapterSortByReportId // @Description: 根据父类id更新排序 // @author: Roc // @receiver chapterChapterInfo // @datetime 2024-06-06 09:39:28 // @param reportId int // @param reportChapterId int // @param nowSort int // @param updateSort string // @return err error func (chapterChapterInfo *ReportChapter) UpdateReportChapterSortByReportId(reportId, reportChapterId, nowSort int, updateSort string) (err error) { pars := make([]interface{}, 0) sql := ` update report_chapter set sort = ` + updateSort + ` WHERE report_id = ? AND sort > ?` pars = append(pars, reportId, nowSort) if reportChapterId > 0 { sql += ` or ( report_chapter_id > ? and sort = ? )` pars = append(pars, reportChapterId, nowSort) } err = global.DmSQL["rddp"].Exec(sql, pars...).Error return } // GetFirstReportChapterByReportId // @Description: 获取当前报告下，且排序数相同 的排序第一条的数据 // @author: Roc // @receiver chapterChapterInfo // @datetime 2024-06-06 09:45:32 // @param reportId int // @return item *ReportChapter // @return err error func (chapterChapterInfo *ReportChapter) GetFirstReportChapterByReportId(reportId int) (item *ReportChapter, err error) { sql := ` SELECT * FROM report_chapter WHERE 1 = 1 AND report_id = ? ORDER BY sort ASC, report_chapter_id ASC LIMIT 1` err = global.DmSQL["rddp"].Raw(sql, reportId).First(&item).Error return } // GetMaxSortByReportId // @Description: 获取最大的排序值 // @author: Roc // @receiver chapterChapterInfo // @datetime 2024-06-06 09:44:13 // @param reportId int // @return maxSort int // @return err error func (chapterChapterInfo *ReportChapter) GetMaxSortByReportId(reportId int) (maxSort int, err error) { sql := `SELECT COALESCE(MAX(sort), 0) AS sort FROM report_chapter WHERE report_id = ?` err = global.DmSQL["rddp"].Raw(sql, reportId).Scan(&maxSort).Error return } // Update // @Description: 数据变更 // @author: Roc // @receiver chapterChapterInfo // @datetime 2024-06-06 09:47:46 // @param cols []string // @return err error func (chapterChapterInfo *ReportChapter) Update(cols []string) (err error) { err = global.DmSQL["rddp"].Select(cols).Updates(chapterChapterInfo).Error return } // DelReportChapterReq // @Description: 删除报告章节请求体 type DelReportChapterReq struct { ReportChapterId int `description:"报告章节ID"` } // GetCountReportChapterByCondition // @Description: 根据条件获取章节数量 // @author: Roc // @datetime 2024-07-15 15:37:50 // @param condition string // @param pars []interface{} // @return count int // @return err error func GetCountReportChapterByCondition(condition string, pars []interface{}) (count int, err error) { sql := ` SELECT COUNT(1) AS count FROM report_chapter WHERE 1=1 ` if condition != "" { sql += condition } err = global.DmSQL["rddp"].Raw(sql, pars...).Scan(&count).Error return }