excel_info.go 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807
  1. package excel
  2. import (
  3. "eta/eta_api/models/data_manage"
  4. "eta/eta_api/services/alarm_msg"
  5. "eta/eta_api/utils"
  6. "fmt"
  7. "github.com/beego/beego/v2/client/orm"
  8. "strconv"
  9. "time"
  10. )
  11. // ExcelInfo excel表格详情表
  12. type ExcelInfo struct {
  13. ExcelInfoId int `orm:"column(excel_info_id);pk"`
  14. Source int `description:"表格来源,1:excel插件的表格,2:自定义表格,3:混合表格,4:自定义分析,默认:1"`
  15. ExcelType int `description:"表格类型,1:指标列,2:日期列,默认:1"`
  16. ExcelName string `description:"表格名称"`
  17. UniqueCode string `description:"表格唯一编码"`
  18. ExcelClassifyId int `description:"表格分类id"`
  19. SysUserId int `description:"操作人id"`
  20. SysUserRealName string `description:"操作人真实姓名"`
  21. Content string `description:"表格内容"`
  22. ExcelImage string `description:"表格图片"`
  23. FileUrl string `description:"表格下载地址"`
  24. Sort int `description:"排序字段,数字越小越排前面"`
  25. IsDelete int `description:"是否删除,0:未删除,1:已删除"`
  26. ModifyTime time.Time `description:"最近修改日期"`
  27. CreateTime time.Time `description:"创建日期"`
  28. IsJoinPermission int `description:"是否加入权限管控,0:不加入;1:加入;默认:0"`
  29. ParentId int `description:"表格的父级id"`
  30. BalanceType int `description:"平衡表类型:0 动态表,1静态表"`
  31. UpdateUserId int `description:"更新人id"`
  32. UpdateUserRealName string `description:"更新人真实姓名"`
  33. RelExcelInfoId int `description:"平衡表里静态表关联的动态表excel id"`
  34. VersionName string `description:"静态表版本名称"`
  35. }
  36. // Update 更新 excel表格基础信息
  37. func (excelInfo *ExcelInfo) Update(cols []string) (err error) {
  38. o := orm.NewOrmUsingDB("data")
  39. _, err = o.Update(excelInfo, cols...)
  40. return
  41. }
  42. type MyExcelInfoList struct {
  43. ExcelInfoId int `orm:"column(excel_info_id);pk"`
  44. Source int `description:"表格来源,1:excel插件的表格,2:自定义表格,默认:1"`
  45. ExcelType int `description:"表格类型,1:指标列,2:日期列,默认:1"`
  46. ExcelName string `description:"表格名称"`
  47. UniqueCode string `description:"表格唯一编码"`
  48. ExcelClassifyId int `description:"表格分类id"`
  49. SysUserId int `description:"操作人id"`
  50. SysUserRealName string `description:"操作人真实姓名"`
  51. ExcelImage string `description:"表格图片"`
  52. FileUrl string `description:"表格下载地址"`
  53. Sort int `description:"排序字段,数字越小越排前面"`
  54. ModifyTime time.Time `description:"最近修改日期"`
  55. CreateTime time.Time `description:"创建日期"`
  56. IsJoinPermission int `description:"是否加入权限管控,0:不加入;1:加入;默认:0"`
  57. HaveOperaAuth bool `description:"是否有数据权限"`
  58. UpdateUserId int `description:"更新人id"`
  59. UpdateUserRealName string `description:"更新人真实姓名"`
  60. Button ExcelInfoDetailButton `description:"操作权限"`
  61. CanEdit bool `description:"是否可编辑"`
  62. Editor string `description:"编辑人"`
  63. }
  64. type ExcelBaseInfo struct {
  65. ExcelInfoId int `orm:"column(excel_info_id);pk"`
  66. Source int `description:"表格来源,1:excel插件的表格,2:自定义表格,3:混合表格,4:自定义分析,默认:1"`
  67. ExcelType int `description:"表格类型,1:指标列,2:日期列,默认:1"`
  68. ExcelName string `description:"表格名称"`
  69. UniqueCode string `description:"表格唯一编码"`
  70. ExcelClassifyId int `description:"表格分类id"`
  71. SysUserId int `description:"操作人id"`
  72. SysUserRealName string `description:"操作人真实姓名"`
  73. }
  74. // AddExcelInfo 新增表格
  75. func AddExcelInfo(excelInfo *ExcelInfo, excelEdbMappingList []*ExcelEdbMapping, childExcel *ExcelInfo) (err error) {
  76. o, err := orm.NewOrmUsingDB("data").Begin()
  77. if err != nil {
  78. return
  79. }
  80. defer func() {
  81. if err != nil {
  82. _ = o.Rollback()
  83. } else {
  84. _ = o.Commit()
  85. }
  86. }()
  87. // 表格信息入库
  88. lastId, err := o.Insert(excelInfo)
  89. if err != nil {
  90. return
  91. }
  92. excelInfo.ExcelInfoId = int(lastId)
  93. // todo 判断如果是平衡表的父级,则新增一个名叫平衡表的子表, 内容为空
  94. if childExcel != nil {
  95. // 表格信息入库
  96. childExcel.ParentId = excelInfo.ExcelInfoId
  97. _, err = o.Insert(childExcel)
  98. if err != nil {
  99. err = fmt.Errorf("新增子表失败:%v", err)
  100. return
  101. }
  102. }
  103. // excel与指标的关联关系
  104. dataNum := len(excelEdbMappingList)
  105. if dataNum > 0 {
  106. for k, v := range excelEdbMappingList {
  107. v.ExcelInfoId = excelInfo.ExcelInfoId
  108. excelEdbMappingList[k] = v
  109. }
  110. _, err = o.InsertMulti(dataNum, excelEdbMappingList)
  111. }
  112. return
  113. }
  114. // EditExcelInfo 编辑表格
  115. func EditExcelInfo(excelInfo *ExcelInfo, updateExcelInfoParams []string, excelEdbMappingList []*ExcelEdbMapping) (err error) {
  116. o, err := orm.NewOrmUsingDB("data").Begin()
  117. if err != nil {
  118. return
  119. }
  120. defer func() {
  121. if err != nil {
  122. _ = o.Rollback()
  123. } else {
  124. _ = o.Commit()
  125. }
  126. }()
  127. // ETA表格信息变更
  128. _, err = o.Update(excelInfo, updateExcelInfoParams...)
  129. if err != nil {
  130. return
  131. }
  132. //更新父级分类时,同时更新子集分类
  133. if excelInfo.Source == utils.BALANCE_TABLE && excelInfo.ParentId == 0 {
  134. // 同步更新子表分类
  135. sql := `UPDATE FROM excel_info set excel_classify_id = ? WHERE parent_id=? `
  136. _, err = o.Raw(sql, excelInfo.ExcelClassifyId, excelInfo.ExcelInfoId).Exec()
  137. }
  138. // 删除关系表
  139. sql := `DELETE FROM excel_edb_mapping WHERE excel_info_id=? `
  140. _, err = o.Raw(sql, excelInfo.ExcelInfoId).Exec()
  141. // excel与指标的关联关系
  142. dataNum := len(excelEdbMappingList)
  143. if dataNum > 0 {
  144. for k, v := range excelEdbMappingList {
  145. v.ExcelInfoId = excelInfo.ExcelInfoId
  146. excelEdbMappingList[k] = v
  147. }
  148. _, err = o.InsertMulti(dataNum, excelEdbMappingList)
  149. }
  150. return
  151. }
  152. // GetExcelInfoAll 获取所有表格列表,用于分类展示
  153. func GetExcelInfoAll() (items []*ExcelClassifyItems, err error) {
  154. o := orm.NewOrmUsingDB("data")
  155. sql := ` SELECT excel_info_id,excel_classify_id,excel_name AS excel_classify_name,
  156. unique_code,sys_user_id,sys_user_real_name,is_join_permission
  157. FROM excel_info where is_delete=0 ORDER BY sort asc,create_time ASC `
  158. _, err = o.Raw(sql).QueryRows(&items)
  159. return
  160. }
  161. // GetNoContentExcelInfoAll 获取不含content的表格列表 用于分类展示
  162. func GetNoContentExcelInfoAll(source, userId int) (items []*ExcelClassifyItems, err error) {
  163. o := orm.NewOrmUsingDB("data")
  164. sql := ` SELECT excel_info_id,excel_classify_id,excel_name AS excel_classify_name,
  165. unique_code,sys_user_id,sys_user_real_name,sort,is_join_permission
  166. FROM excel_info where is_delete=0 AND source = ? `
  167. pars := []interface{}{source}
  168. if userId > 0 {
  169. sql += ` AND sys_user_id = ? `
  170. pars = append(pars, userId)
  171. }
  172. sql += ` ORDER BY sort asc,excel_info_id desc `
  173. _, err = o.Raw(sql, pars...).QueryRows(&items)
  174. return
  175. }
  176. // GetBalanceNoContentExcelInfoAll 获取不含content的平衡表表格列表 用于分类展示
  177. func GetBalanceNoContentExcelInfoAll(source int, excelInfoIds []int, userId int) (items []*ExcelClassifyItems, err error) {
  178. o := orm.NewOrmUsingDB("data")
  179. sql := ` SELECT excel_info_id,excel_classify_id,excel_name AS excel_classify_name,
  180. unique_code,sys_user_id,sys_user_real_name,sort,is_join_permission
  181. FROM excel_info where is_delete=0 AND source = ? AND parent_id = 0 AND balance_type=0 `
  182. pars := []interface{}{source}
  183. if userId > 0 {
  184. if len(excelInfoIds) > 0 {
  185. sql += ` AND (excel_info_id in (` + utils.GetOrmInReplace(len(excelInfoIds)) + `) or sys_user_id = ?)`
  186. pars = append(pars, excelInfoIds, userId)
  187. } else {
  188. sql += ` AND sys_user_id = ? `
  189. pars = append(pars, userId)
  190. }
  191. }
  192. sql += ` ORDER BY sort asc,excel_info_id desc `
  193. _, err = o.Raw(sql, pars...).QueryRows(&items)
  194. return
  195. }
  196. // GetAllExcelInfoBySource 根据来源获取包含content的表格列表
  197. func GetAllExcelInfoBySource(source int) (items []*ExcelInfo, err error) {
  198. o := orm.NewOrmUsingDB("data")
  199. sql := ` SELECT * FROM excel_info where is_delete=0 AND source = ? ORDER BY sort asc,create_time desc `
  200. _, err = o.Raw(sql, source).QueryRows(&items)
  201. return
  202. }
  203. // GetExcelInfoById 根据id 获取eta表格详情
  204. func GetExcelInfoById(excelInfoId int) (item *ExcelInfo, err error) {
  205. o := orm.NewOrmUsingDB("data")
  206. sql := ` SELECT * FROM excel_info WHERE excel_info_id=? AND is_delete=0 `
  207. err = o.Raw(sql, excelInfoId).QueryRow(&item)
  208. return
  209. }
  210. // GetExcelInfoByUnicode 编码获取表格
  211. func GetExcelInfoByUnicode(unicode string) (item *ExcelInfo, err error) {
  212. o := orm.NewOrmUsingDB("data")
  213. sql := ` SELECT * FROM excel_info WHERE unique_code = ? AND is_delete = 0 `
  214. err = o.Raw(sql, unicode).QueryRow(&item)
  215. return
  216. }
  217. func GetExcelInfoViewById(excelInfoId int) (item *ExcelInfoView, err error) {
  218. o := orm.NewOrmUsingDB("data")
  219. sql := ` SELECT * FROM excel_info WHERE excel_info_id=? AND is_delete=0 `
  220. err = o.Raw(sql, excelInfoId).QueryRow(&item)
  221. return
  222. }
  223. func GetExcelInfoCountByCondition(condition string, pars []interface{}) (count int, err error) {
  224. o := orm.NewOrmUsingDB("data")
  225. sql := ` SELECT COUNT(1) AS count FROM excel_info WHERE 1=1 AND is_delete=0 `
  226. if condition != "" {
  227. sql += condition
  228. }
  229. err = o.Raw(sql, pars).QueryRow(&count)
  230. return
  231. }
  232. func GetNoContentExcelInfoListByCondition(condition string, pars []interface{}, startSize, pageSize int) (items []*ExcelClassifyItems, err error) {
  233. o := orm.NewOrmUsingDB("data")
  234. sql := `SELECT excel_info_id,excel_classify_id,excel_name AS excel_classify_name,
  235. unique_code,sys_user_id,sys_user_real_name,sort,is_join_permission FROM excel_info WHERE 1=1 `
  236. if condition != "" {
  237. sql += condition
  238. }
  239. sql += ` AND is_delete=0 ORDER BY excel_info_id DESC LIMIT ?,? `
  240. _, err = o.Raw(sql, pars, startSize, pageSize).QueryRows(&items)
  241. return
  242. }
  243. func GetNoContentExcelInfoListByConditionNoPage(condition string, pars []interface{}) (items []*ExcelInfo, err error) {
  244. o := orm.NewOrmUsingDB("data")
  245. sql := `SELECT excel_info_id,excel_classify_id,excel_name,
  246. unique_code,sys_user_id,sys_user_real_name,sort,is_join_permission, parent_id, balance_type, update_user_id,update_user_real_name,rel_excel_info_id,version_name FROM excel_info WHERE is_delete=0 `
  247. if condition != "" {
  248. sql += condition
  249. }
  250. sql += ` ORDER BY excel_info_id DESC `
  251. _, err = o.Raw(sql, pars).QueryRows(&items)
  252. return
  253. }
  254. func GetExcelInfoListByCondition(condition string, pars []interface{}) (items []*ExcelInfo, err error) {
  255. o := orm.NewOrmUsingDB("data")
  256. sql := ` SELECT * FROM excel_info WHERE 1=1 AND is_delete=0 `
  257. if condition != "" {
  258. sql += condition
  259. }
  260. sql += ` ORDER BY sort asc, excel_info_id asc`
  261. _, err = o.Raw(sql, pars).QueryRows(&items)
  262. return
  263. }
  264. // GetNextExcelInfoByCondition 根据条件获取下一个表格
  265. func GetNextExcelInfoByCondition(condition string, pars []interface{}) (item *ExcelInfo, err error) {
  266. o := orm.NewOrmUsingDB("data")
  267. sql := ` SELECT * FROM excel_info WHERE 1=1 AND is_delete=0 `
  268. if condition != "" {
  269. sql += condition
  270. }
  271. sql += " ORDER BY sort asc , create_time desc LIMIT 1 "
  272. err = o.Raw(sql, pars).QueryRow(&item)
  273. return
  274. }
  275. // GetNextExcelInfo 根据分类id获取下一个excel表格
  276. func GetNextExcelInfo(classifyId, classifySort, source int) (item *ExcelInfo, err error) {
  277. o := orm.NewOrmUsingDB("data")
  278. sql := ` SELECT b.* FROM excel_classify AS a
  279. INNER JOIN excel_info AS b ON a.excel_classify_id=b.excel_classify_id
  280. WHERE (a.sort>? OR (a.sort=? and a.excel_classify_id>?) ) AND a.is_delete=0 AND b.is_delete=0
  281. AND a.source = ? AND b.source = ?
  282. ORDER BY a.sort ASC,b.sort asc,b.create_time desc
  283. LIMIT 1 `
  284. err = o.Raw(sql, classifySort, classifySort, classifyId, source, source).QueryRow(&item)
  285. return
  286. }
  287. // EditExcelInfoImage 修改excel表格的图片
  288. func EditExcelInfoImage(excelInfoId int, imageUrl string) (err error) {
  289. o := orm.NewOrmUsingDB("data")
  290. sql := ` UPDATE excel_info SET excel_image=?, modify_time = NOW() WHERE excel_info_id = ? AND is_delete=0 `
  291. _, err = o.Raw(sql, imageUrl, excelInfoId).Exec()
  292. if err != nil {
  293. fmt.Println("EditExcelInfoImage Err:", err.Error())
  294. return err
  295. }
  296. return
  297. }
  298. // GetExcelInfoByUniqueCode 根据unique_code来获取excel表格详情
  299. func GetExcelInfoByUniqueCode(uniqueCode string) (item *ExcelInfo, err error) {
  300. o := orm.NewOrmUsingDB("data")
  301. sql := ` SELECT * FROM excel_info WHERE unique_code=? AND is_delete=0 `
  302. err = o.Raw(sql, uniqueCode).QueryRow(&item)
  303. return
  304. }
  305. // GetFirstExcelInfoByClassifyId 获取当前分类下,且排序数相同 的排序第一条的数据
  306. func GetFirstExcelInfoByClassifyId(classifyId int) (item *ExcelInfo, err error) {
  307. o := orm.NewOrmUsingDB("data")
  308. sql := ` SELECT * FROM excel_info WHERE excel_classify_id=? AND is_delete=0 order by sort asc,excel_info_id desc limit 1`
  309. err = o.Raw(sql, classifyId).QueryRow(&item)
  310. return
  311. }
  312. // UpdateExcelInfoSortByClassifyId 根据表格id更新排序
  313. func UpdateExcelInfoSortByClassifyId(classifyId, nowSort, prevExcelInfoId int, updateSort string, source int) (err error) {
  314. o := orm.NewOrmUsingDB("data")
  315. sql := ` update excel_info set sort = ` + updateSort + ` WHERE excel_classify_id=? AND source=? AND is_delete=0 AND ( sort > ? `
  316. // todo 前一个兄弟节点后移
  317. if prevExcelInfoId > 0 {
  318. sql += ` or (excel_info_id < ` + fmt.Sprint(prevExcelInfoId) + ` and sort = ` + fmt.Sprint(nowSort) + `)`
  319. }
  320. sql += `)`
  321. _, err = o.Raw(sql, classifyId, source, nowSort).Exec()
  322. return
  323. }
  324. type ExcelInfoView struct {
  325. ExcelInfoId int `orm:"column(excel_info_id);pk"`
  326. ExcelName string `description:"来源名称"`
  327. ExcelClassifyId int `description:"表格分类id"`
  328. ExcelClassifyName string `description:"表格名称"`
  329. SysUserId int
  330. SysUserRealName string
  331. UniqueCode string `description:"表格唯一编码"`
  332. CreateTime time.Time
  333. ModifyTime time.Time
  334. DateType int `description:"日期类型:1:00年至今,2:10年至今,3:15年至今,4:年初至今,5:自定义时间"`
  335. StartDate string `description:"自定义开始日期"`
  336. EndDate string `description:"自定义结束日期"`
  337. IsSetName int `description:"设置名称"`
  338. EdbInfoIds string `description:"指标id"`
  339. ExcelType int `description:"生成样式:1:曲线图,2:季节性图"`
  340. Calendar string `description:"公历/农历"`
  341. SeasonStartDate string `description:"季节性图开始日期"`
  342. SeasonEndDate string `description:"季节性图开始日期"`
  343. ExcelImage string `description:"表格图片"`
  344. Sort int `description:"排序字段,数字越小越排前面"`
  345. IsAdd bool `description:"true:已加入我的图库,false:未加入我的图库"`
  346. MyExcelId int
  347. MyExcelClassifyId string `description:"我的表格分类,多个用逗号隔开"`
  348. ExcelClassify []*ExcelClassifyView
  349. EdbEndDate string `description:"指标最新更新日期"`
  350. LeftMin string `description:"表格左侧最小值"`
  351. LeftMax string `description:"表格左侧最大值"`
  352. RightMin string `description:"表格右侧最小值"`
  353. RightMax string `description:"表格右侧最大值"`
  354. }
  355. // GetExcelInfoByClassifyIdAndName 根据分类id和表格名获取表格信息
  356. func GetExcelInfoByClassifyIdAndName(classifyId int, excelName string) (item *ExcelInfo, err error) {
  357. o := orm.NewOrmUsingDB("data")
  358. sql := ` SELECT * FROM excel_info WHERE excel_classify_id = ? and excel_name=? AND is_delete=0 `
  359. err = o.Raw(sql, classifyId, excelName).QueryRow(&item)
  360. return
  361. }
  362. // GetNoContentExcelListByCondition 获取没有content的excel表格列表数据
  363. func GetNoContentExcelListByCondition(condition string, pars []interface{}, startSize, pageSize int) (item []*MyExcelInfoList, err error) {
  364. o := orm.NewOrmUsingDB("data")
  365. sql := ` SELECT excel_info_id,source,excel_type,excel_name,unique_code,excel_classify_id,sys_user_id,sys_user_real_name,excel_image,file_url,sort,create_time,modify_time,is_join_permission,update_user_id,update_user_real_name
  366. FROM excel_info WHERE 1=1 AND is_delete=0 `
  367. if condition != "" {
  368. sql += condition
  369. }
  370. //sql += " ORDER BY sort ASC,chart_info_id DESC LIMIT ?,? "
  371. sql += " ORDER BY create_time DESC LIMIT ?,? "
  372. _, err = o.Raw(sql, pars, startSize, pageSize).QueryRows(&item)
  373. return
  374. }
  375. // GetNoContentExcelListByConditionNoPage 获取没有content的excel表格列表数据
  376. func GetNoContentExcelListByConditionNoPage(condition string, pars []interface{}) (item []*MyExcelInfoList, err error) {
  377. o := orm.NewOrmUsingDB("data")
  378. sql := ` SELECT excel_info_id,source,excel_type,excel_name,unique_code,excel_classify_id,sys_user_id,sys_user_real_name,excel_image,file_url,sort,create_time,modify_time,is_join_permission,update_user_id,update_user_real_name
  379. FROM excel_info WHERE 1=1 AND is_delete=0 `
  380. if condition != "" {
  381. sql += condition
  382. }
  383. //sql += " ORDER BY sort ASC,chart_info_id DESC LIMIT ?,? "
  384. sql += " ORDER BY create_time DESC"
  385. _, err = o.Raw(sql, pars).QueryRows(&item)
  386. return
  387. }
  388. func GetExcelListCountByCondition(condition string, pars []interface{}) (count int, err error) {
  389. o := orm.NewOrmUsingDB("data")
  390. sql := ` SELECT COUNT(1) AS count FROM excel_info WHERE 1=1 AND is_delete=0 `
  391. if condition != "" {
  392. sql += condition
  393. }
  394. err = o.Raw(sql, pars).QueryRow(&count)
  395. return
  396. }
  397. // GetExcelViewInfoByExcelInfoId 根据excelInfoId 获取ETA表格详情
  398. func GetExcelViewInfoByExcelInfoId(excelInfoId int) (item *MyExcelInfoList, err error) {
  399. o := orm.NewOrmUsingDB("data")
  400. sql := ` SELECT excel_info_id,source,excel_type,excel_name,unique_code,excel_classify_id,sys_user_id,sys_user_real_name,excel_image,file_url,sort,create_time,modify_time,is_join_permission FROM excel_info WHERE excel_info_id = ? AND is_delete=0 `
  401. err = o.Raw(sql, excelInfoId).QueryRow(&item)
  402. return
  403. }
  404. // GetExcelInfoCountByClassifyId 根据分类id获取名下表格数量
  405. func GetExcelInfoCountByClassifyId(classifyId int) (total int64, err error) {
  406. o := orm.NewOrmUsingDB("data")
  407. sql := ` SELECT count(1) total FROM excel_info WHERE excel_classify_id = ? AND is_delete=0 `
  408. err = o.Raw(sql, classifyId).QueryRow(&total)
  409. return
  410. }
  411. // UpdateExcelInfoClassifyId 更改表格分类
  412. func UpdateExcelInfoClassifyId(classifyId, excelInfoId int) (err error) {
  413. o := orm.NewOrmUsingDB("data")
  414. sql := ` update excel_info set excel_classify_id = ? WHERE excel_info_id=? `
  415. _, err = o.Raw(sql, classifyId, excelInfoId).Exec()
  416. return
  417. }
  418. // GetNoContentExcelInfoByName 根据名称 获取eta表格详情
  419. func GetNoContentExcelInfoByName(excelName string, source int) (item *MyExcelInfoList, err error) {
  420. o := orm.NewOrmUsingDB("data")
  421. sql := ` SELECT excel_info_id,source,excel_type,excel_name,unique_code,excel_classify_id,sys_user_id,sys_user_real_name,excel_image,file_url,sort,create_time,modify_time,is_join_permission,update_user_id,update_user_real_name
  422. FROM excel_info WHERE excel_name = ? AND source = ? AND is_delete=0 `
  423. err = o.Raw(sql, excelName, source).QueryRow(&item)
  424. return
  425. }
  426. // GetNoContentExcelInfoByUniqueCode 根据unique_code来获取excel表格详情
  427. func GetNoContentExcelInfoByUniqueCode(uniqueCode string) (item *MyExcelInfoList, err error) {
  428. o := orm.NewOrmUsingDB("data")
  429. sql := ` SELECT excel_info_id,source,excel_type,excel_name,unique_code,excel_classify_id,sys_user_id,sys_user_real_name,excel_image,file_url,sort,create_time,modify_time,is_join_permission,update_user_id,update_user_real_name
  430. FROM excel_info WHERE unique_code=? AND is_delete=0 `
  431. err = o.Raw(sql, uniqueCode).QueryRow(&item)
  432. return
  433. }
  434. // GetNoContentExcelInfoByExcelId 根据表格id来获取excel表格详情
  435. func GetNoContentExcelInfoByExcelId(excelInfoId int) (item *MyExcelInfoList, err error) {
  436. o := orm.NewOrmUsingDB("data")
  437. sql := ` SELECT excel_info_id,source,excel_type,excel_name,unique_code,excel_classify_id,sys_user_id,sys_user_real_name,excel_image,file_url,sort,create_time,modify_time,is_join_permission,update_user_id,update_user_real_name
  438. FROM excel_info WHERE excel_info_id=? AND is_delete=0 `
  439. err = o.Raw(sql, excelInfoId).QueryRow(&item)
  440. return
  441. }
  442. // AddExcelInfoAndSheet 新增excel
  443. func AddExcelInfoAndSheet(excelInfo *ExcelInfo, sheetParamsList []AddExcelSheetParams) (err error) {
  444. o, err := orm.NewOrmUsingDB("data").Begin()
  445. if err != nil {
  446. return
  447. }
  448. defer func() {
  449. if err != nil {
  450. _ = o.Rollback()
  451. } else {
  452. _ = o.Commit()
  453. }
  454. }()
  455. // 表格信息入库
  456. lastId, err := o.Insert(excelInfo)
  457. if err != nil {
  458. return
  459. }
  460. excelInfo.ExcelInfoId = int(lastId)
  461. // sheet信息入库
  462. for _, sheetInfo := range sheetParamsList {
  463. dataNum := len(sheetInfo.DataList)
  464. //sheet信息入库
  465. excelSheetInfo := &ExcelSheet{
  466. ExcelSheetId: 0,
  467. ExcelInfoId: excelInfo.ExcelInfoId,
  468. SheetName: sheetInfo.SheetName,
  469. PageNum: dataNum,
  470. Index: sheetInfo.Index,
  471. Sort: sheetInfo.Sort,
  472. Config: sheetInfo.Config,
  473. CalcChain: sheetInfo.CalcChain,
  474. ModifyTime: time.Now(),
  475. CreateTime: time.Now(),
  476. }
  477. sheetId, tmpErr := o.Insert(excelSheetInfo)
  478. if tmpErr != nil {
  479. err = tmpErr
  480. return
  481. }
  482. excelSheetInfo.ExcelSheetId = int(sheetId)
  483. // data信息入库
  484. if dataNum > 0 {
  485. for k, _ := range sheetInfo.DataList {
  486. sheetInfo.DataList[k].ExcelSheetId = excelSheetInfo.ExcelSheetId
  487. sheetInfo.DataList[k].ExcelInfoId = excelSheetInfo.ExcelInfoId
  488. }
  489. _, tmpErr = o.InsertMulti(dataNum, sheetInfo.DataList)
  490. if tmpErr != nil {
  491. err = tmpErr
  492. return
  493. }
  494. }
  495. }
  496. return
  497. }
  498. // SaveExcelInfoAndSheet 编辑保存
  499. func SaveExcelInfoAndSheet(excelInfo *ExcelInfo, updateExcelInfoParam []string, sheetParamsList []AddExcelSheetParams) (err error) {
  500. o, err := orm.NewOrmUsingDB("data").Begin()
  501. if err != nil {
  502. return
  503. }
  504. defer func() {
  505. if err != nil {
  506. _ = o.Rollback()
  507. } else {
  508. _ = o.Commit()
  509. }
  510. }()
  511. // 表格信息入库
  512. _, err = o.Update(excelInfo, updateExcelInfoParam...)
  513. if err != nil {
  514. return
  515. }
  516. // 先删除历史的sheet信息
  517. sql := `DELETE FROM excel_sheet WHERE excel_info_id = ?`
  518. _, err = o.Raw(sql, excelInfo.ExcelInfoId).Exec()
  519. if err != nil {
  520. return
  521. }
  522. // 再删除历史sheet中的cell data信息
  523. sql = `DELETE FROM excel_sheet_data WHERE excel_info_id = ?`
  524. _, err = o.Raw(sql, excelInfo.ExcelInfoId).Exec()
  525. if err != nil {
  526. return
  527. }
  528. // sheet信息入库
  529. for _, sheetInfo := range sheetParamsList {
  530. dataNum := len(sheetInfo.DataList)
  531. //sheet信息入库
  532. excelSheetInfo := &ExcelSheet{
  533. ExcelSheetId: 0,
  534. ExcelInfoId: excelInfo.ExcelInfoId,
  535. SheetName: sheetInfo.SheetName,
  536. PageNum: dataNum,
  537. Index: sheetInfo.Index,
  538. Sort: sheetInfo.Sort,
  539. Config: sheetInfo.Config,
  540. CalcChain: sheetInfo.CalcChain,
  541. ModifyTime: time.Now(),
  542. CreateTime: time.Now(),
  543. }
  544. sheetId, tmpErr := o.Insert(excelSheetInfo)
  545. if tmpErr != nil {
  546. err = tmpErr
  547. return
  548. }
  549. excelSheetInfo.ExcelSheetId = int(sheetId)
  550. // data信息入库
  551. if dataNum > 0 {
  552. for k, _ := range sheetInfo.DataList {
  553. sheetInfo.DataList[k].ExcelSheetId = excelSheetInfo.ExcelSheetId
  554. sheetInfo.DataList[k].ExcelInfoId = excelSheetInfo.ExcelInfoId
  555. }
  556. _, tmpErr = o.InsertMulti(dataNum, sheetInfo.DataList)
  557. if tmpErr != nil {
  558. err = tmpErr
  559. return
  560. }
  561. }
  562. }
  563. return
  564. }
  565. // BatchRefreshExcelReq 批量刷新表格请求
  566. type BatchRefreshExcelReq struct {
  567. ExcelCodes []string `description:"表格编码"`
  568. PrimaryId int `description:"研报/智能研报/PPT主键"`
  569. SubId int `description:"报告章节ID"`
  570. Source string `description:"来源,枚举值:report、english_report、smart_report"`
  571. }
  572. // GetExcelMaxSortByClassifyId 获取当前分类下,且排序数最大的excel
  573. func GetExcelMaxSortByClassifyId(classifyId int, source int) (sort int, err error) {
  574. o := orm.NewOrmUsingDB("data")
  575. sql := ` SELECT Max(sort) AS sort FROM excel_info WHERE excel_classify_id=? AND source = ? AND is_delete=0 order by sort desc,excel_info_id desc limit 1`
  576. err = o.Raw(sql, classifyId, source).QueryRow(&sort)
  577. return
  578. }
  579. // GetNoContentExcelListByExcelInfoIdList
  580. // @Description: 根据ETA表格ID列表获取列表信息
  581. // @param excelInfoIdList []string
  582. // @return items
  583. // @return err
  584. func GetNoContentExcelListByExcelInfoIdList(excelInfoIdList []string) (items []*MyExcelInfoList, err error) {
  585. num := len(excelInfoIdList)
  586. if num <= 0 {
  587. return
  588. }
  589. o := orm.NewOrmUsingDB("data")
  590. sql := ` SELECT * FROM excel_info WHERE excel_info_id in (` + utils.GetOrmInReplace(num) + `) order by excel_info_id DESC `
  591. _, err = o.Raw(sql, excelInfoIdList).QueryRows(&items)
  592. return
  593. }
  594. func GetNoContentExcelListByExcelInfoIdAndParentId(excelInfoIdList []string) (items []*MyExcelInfoList, err error) {
  595. num := len(excelInfoIdList)
  596. if num <= 0 {
  597. return
  598. }
  599. o := orm.NewOrmUsingDB("data")
  600. sql := ` SELECT * FROM excel_info WHERE excel_info_id in (` + utils.GetOrmInReplace(num) + `) OR parent_id in (` + utils.GetOrmInReplace(num) + `) order by excel_info_id DESC `
  601. _, err = o.Raw(sql, excelInfoIdList, excelInfoIdList).QueryRows(&items)
  602. return
  603. }
  604. // GetNoContentExcelListByUserId
  605. // @Description: 根据ETA表格ID列表获取列表信息
  606. // @param userIdList []int
  607. // @return items
  608. // @return err
  609. func GetNoContentExcelListByUserId(userIdList []int) (items []*MyExcelInfoList, err error) {
  610. num := len(userIdList)
  611. if num <= 0 {
  612. return
  613. }
  614. o := orm.NewOrmUsingDB("data")
  615. sql := ` SELECT * FROM excel_info WHERE excel_info_id in (` + utils.GetOrmInReplace(num) + `) order by excel_info_id DESC `
  616. _, err = o.Raw(sql, userIdList).QueryRows(&items)
  617. return
  618. }
  619. // ModifyExcelInfoUserIdByCodeList
  620. // @Description: 根据编码列表修改表格信息
  621. // @author: Roc
  622. // @datetime 2024-03-26 14:16:30
  623. // @param excelIdList []string
  624. // @param userId int
  625. // @param userName string
  626. // @return err error
  627. func ModifyExcelInfoUserIdByCodeList(excelIdList []string, userId int, userName string) (err error) {
  628. num := len(excelIdList)
  629. if num <= 0 {
  630. return
  631. }
  632. o := orm.NewOrmUsingDB("data")
  633. sql := `UPDATE excel_info SET sys_user_id=?,sys_user_real_name=? WHERE excel_info_id in (` + utils.GetOrmInReplace(num) + `) `
  634. _, err = o.Raw(sql, userId, userName, excelIdList).Exec()
  635. return
  636. }
  637. // ModifyExcelInfoUserIdByOldUserId
  638. // @Description: 根据旧的用户id修改创建人
  639. // @author: Roc
  640. // @datetime 2024-03-25 19:14:59
  641. // @param oldUserId int
  642. // @param userId int
  643. // @param userName string
  644. // @return err error
  645. func ModifyExcelInfoUserIdByOldUserId(oldUserIdList []int, userId int, userName string) (err error) {
  646. num := len(oldUserIdList)
  647. if num <= 0 {
  648. return
  649. }
  650. o := orm.NewOrmUsingDB("data")
  651. sql := `UPDATE excel_info SET sys_user_id=?,sys_user_real_name=? WHERE is_delete=0 AND sys_user_id in (` + utils.GetOrmInReplace(num) + `) `
  652. _, err = o.Raw(sql, userId, userName, oldUserIdList).Exec()
  653. return
  654. }
  655. // GetChildExcelInfoByParentId 根据id 获取eta表格详情
  656. func GetChildExcelInfoByParentId(parentId int) (items []*ExcelInfo, err error) {
  657. o := orm.NewOrmUsingDB("data")
  658. sql := ` SELECT excel_info_id,source,excel_type,excel_name,unique_code,excel_classify_id,sys_user_id,sys_user_real_name,excel_image,file_url,sort,create_time,modify_time,is_join_permission, parent_id, balance_type, update_user_id,update_user_real_name FROM excel_info WHERE parent_id=? AND is_delete=0 order by sort asc, excel_info_id asc`
  659. _, err = o.Raw(sql, parentId).QueryRows(&items)
  660. return
  661. }
  662. // ExcelInfoDetailButton 操作按钮
  663. type ExcelInfoDetailButton struct {
  664. RefreshButton bool `description:"是否可刷新"`
  665. CopyButton bool `description:"是否可另存为"`
  666. DownloadButton bool `description:"是否可下载"`
  667. OpButton bool `description:"是否可编辑"`
  668. DeleteButton bool `description:"是否可删除"`
  669. OpEdbButton bool `description:"是否可生成指标"`
  670. RefreshEdbButton bool `description:"是否可刷新指标"`
  671. OpWorkerButton bool `description:"是否修改协作人"`
  672. }
  673. func GetExcelBaseInfoByExcelInfoIdList(excelInfoIdList []int) (items []*ExcelBaseInfo, err error) {
  674. num := len(excelInfoIdList)
  675. if num <= 0 {
  676. return
  677. }
  678. o := orm.NewOrmUsingDB("data")
  679. sql := ` SELECT excel_info_id,source,excel_type,excel_name,unique_code,excel_classify_id,sys_user_id,sys_user_real_name FROM excel_info WHERE excel_info_id in (` + utils.GetOrmInReplace(num) + `) order by excel_info_id DESC `
  680. _, err = o.Raw(sql, excelInfoIdList).QueryRows(&items)
  681. return
  682. }
  683. // ReplaceEdbInExcel 替换表格中的指标
  684. func ReplaceEdbInExcel(oldEdbInfo, newEdbInfo *data_manage.EdbInfo, updateExcelList []*ExcelInfo) (err error) {
  685. var errmsg string
  686. logMsg := `` // 记录替换的日志
  687. replaceTotal := 0
  688. o := orm.NewOrmUsingDB("data")
  689. to, err := o.Begin()
  690. if err != nil {
  691. return
  692. }
  693. defer func() {
  694. if err != nil {
  695. _ = to.Rollback()
  696. } else {
  697. _ = to.Commit()
  698. if logMsg != `` {
  699. utils.FileLog.Info(fmt.Sprintf("替换表格中的指标记录,替换总表格数:%d,旧的指标id:%d,新的指标id:%d;%s", replaceTotal, oldEdbInfo.EdbInfoId, newEdbInfo.EdbInfoId, logMsg))
  700. }
  701. }
  702. if errmsg != "" {
  703. fmt.Println("errmsg:" + errmsg)
  704. }
  705. if err != nil && errmsg != "" {
  706. go alarm_msg.SendAlarmMsg("替换替换配置中的指标记录失败提醒,errmsg:"+errmsg, 3)
  707. }
  708. }()
  709. // 替换表格中的内容
  710. for _, excelInfo := range updateExcelList {
  711. //更新配置中的指标A
  712. sql := `UPDATE excel_info SET content=?, modify_time=? WHERE excel_info_id=?`
  713. _, err = to.Raw(sql, excelInfo.Content, time.Now(), excelInfo.ExcelInfoId).Exec()
  714. if err != nil {
  715. errmsg = "更新表格内容失败:Err:" + err.Error()
  716. return
  717. }
  718. //更新配置中的指标B
  719. sql = `UPDATE excel_edb_mapping SET edb_info_id=?, modify_time=? WHERE excel_info_id=? and edb_info_id=?`
  720. _, err = to.Raw(sql, newEdbInfo.EdbInfoId, time.Now(), excelInfo.ExcelInfoId, oldEdbInfo.EdbInfoId).Exec()
  721. if err != nil {
  722. errmsg = "更新指标B关联图表配置信息失败:Err:" + err.Error()
  723. return
  724. }
  725. logMsg += `涉及到的表格id:` + strconv.Itoa(excelInfo.ExcelInfoId) + ";"
  726. replaceTotal += 1
  727. }
  728. return
  729. }