edb_refresh.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664
  1. package services
  2. import (
  3. "context"
  4. "eta/eta_task/models"
  5. "eta/eta_task/models/data_manage"
  6. "eta/eta_task/models/data_manage/edb_refresh"
  7. "eta/eta_task/services/alarm_msg"
  8. "eta/eta_task/services/data"
  9. "eta/eta_task/utils"
  10. "fmt"
  11. "strings"
  12. "sync"
  13. "time"
  14. )
  15. // ConfigRefreshData
  16. // @Description: 配置刷新数据
  17. // @author: Roc
  18. // @datetime 2024-01-10 13:55:05
  19. // @param cont context.Context
  20. // @return err error
  21. func ConfigRefreshData(cont context.Context) (err error) {
  22. errMsgList := make([]string, 0)
  23. defer func() {
  24. if err != nil {
  25. fmt.Println(err)
  26. }
  27. }()
  28. // 一期是只做wind、同花顺、钢联、有色
  29. now := time.Now()
  30. //now = time.Date(2023, 12, 31, 19, 10, 59, 0, time.Local)
  31. //now = time.Date(2023, 12, 31, 16, 50, 59, 0, time.Local)
  32. defaultSourceEdbInfoListMap, err := getDefaultRefreshData(now)
  33. if err != nil {
  34. errMsgList = append(errMsgList, "获取默认刷新数据失败,Err:"+err.Error())
  35. }
  36. sourceEdbInfoListMap, err := getConfigRefreshData(now)
  37. if err != nil {
  38. errMsgList = append(errMsgList, "获取指标配置刷新数据失败,Err:"+err.Error())
  39. }
  40. // 将两个合并
  41. allSourceEdbInfoListMap := mergeMaps(defaultSourceEdbInfoListMap, sourceEdbInfoListMap)
  42. wgNum := len(allSourceEdbInfoListMap)
  43. if wgNum <= 0 {
  44. return
  45. }
  46. wg := sync.WaitGroup{}
  47. wg.Add(wgNum)
  48. for _, edbList := range allSourceEdbInfoListMap {
  49. if edbList == nil {
  50. wg.Done()
  51. continue
  52. }
  53. if len(edbList) != 0 {
  54. go BaseRefreshData(&wg, edbList[0].Source, edbList[0].SubSource, edbList)
  55. }
  56. }
  57. wg.Wait()
  58. fmt.Println("Refresh End")
  59. return
  60. }
  61. // Function to merge two maps
  62. func mergeMaps(dst map[string][]*edb_refresh.EdbInfoListAndRefreshConfig, src map[string][]*edb_refresh.EdbInfoListAndRefreshConfig) (newMap map[string][]*edb_refresh.EdbInfoListAndRefreshConfig) {
  63. if dst == nil {
  64. return src
  65. }
  66. if src == nil {
  67. return dst
  68. }
  69. newMap = dst
  70. for k, v := range src {
  71. if newK, ok := newMap[k]; ok {
  72. newK = append(newK, v...)
  73. newMap[k] = newK
  74. } else {
  75. newMap[k] = v
  76. }
  77. }
  78. return newMap
  79. }
  80. // getDefaultRefreshData
  81. // @Description: 根据默认配置获取需要刷新的指标列表
  82. // @author: Roc
  83. // @datetime 2024-01-10 13:55:38
  84. // @param now time.Time
  85. // @return sourceEdbInfoListMap map[string][]*edb_refresh.EdbInfoListAndRefreshConfig
  86. // @return err error
  87. func getDefaultRefreshData(now time.Time) (sourceEdbInfoListMap map[string][]*edb_refresh.EdbInfoListAndRefreshConfig, err error) {
  88. errMsgList := make([]string, 0)
  89. defer func() {
  90. if err != nil {
  91. fmt.Println(err)
  92. }
  93. }()
  94. // 一期是只做wind、同花顺、钢联、有色
  95. sourceEdbInfoListMap = make(map[string][]*edb_refresh.EdbInfoListAndRefreshConfig)
  96. currTimeStr := getPreviousHalfHour(now)
  97. fmt.Println(currTimeStr)
  98. // 所有默认配置刷新项
  99. list := make([]*edb_refresh.EdbRefreshDefaultConfig, 0)
  100. //刷新频率,枚举值:每自然日、每交易日、每周、每旬、每月、每季、每半年、每年
  101. refreshFrequencyList := []string{"每自然日", "每交易日", "每周", "每旬", "每月", "每季", "每半年", "每年"}
  102. conf, err := models.GetBusinessConf()
  103. if err != nil {
  104. fmt.Println(err)
  105. utils.FileLog.Info("获取业务配置失败,Err:" + err.Error())
  106. return
  107. }
  108. // 获取钢联化工的数据获取方式
  109. mySteelChemicalDataMethod := "excel"
  110. if v, ok := conf["MySteelDataMethod"]; ok {
  111. if v == "api" {
  112. mySteelChemicalDataMethod = v
  113. }
  114. }
  115. utils.FileLog.Info("获取业务配置,MySteelDataMethod:" + mySteelChemicalDataMethod)
  116. // 获取各个刷新频率的配置
  117. for _, refreshFrequency := range refreshFrequencyList {
  118. // 获取刷新频率条件
  119. condition, pars, isHandler := getRefreshFrequencyCondition(now, refreshFrequency)
  120. if !isHandler {
  121. // 可能是非交易日,所以过滤不处理
  122. continue
  123. }
  124. condition += ` AND refresh_frequency = ? AND refresh_time = ?`
  125. pars = append(pars, refreshFrequency, currTimeStr)
  126. // 这两个是excel的数据源,他是从公共机更新的,需要过滤掉
  127. if mySteelChemicalDataMethod == "api" {
  128. // 钢联化工使用api的方式获取数据的,不需要过滤
  129. condition += ` AND source not in (?)`
  130. pars = append(pars, utils.DATA_SOURCE_YS)
  131. } else {
  132. condition += ` AND source not in (?,?)`
  133. pars = append(pars, utils.DATA_SOURCE_MYSTEEL_CHEMICAL, utils.DATA_SOURCE_YS)
  134. }
  135. tmpList, tmpErr := edb_refresh.GetListByCondition(condition, pars)
  136. if tmpErr != nil {
  137. err = tmpErr
  138. return
  139. }
  140. list = append(list, tmpList...)
  141. }
  142. // 更新的单元格数
  143. refreshDataNumMap := make(map[string]*edb_refresh.EdbRefreshDefaultConfig)
  144. // 数据源刷新频度的列表数组
  145. refreshDataFrequencyListMap := make(map[int]map[int][]string)
  146. wgNum := 0
  147. // 处理待刷新的数据源,整理成数组,方便获取对应的指标
  148. for _, item := range list {
  149. // 更新的单元格数
  150. key := fmt.Sprintf("%d_%d_%s", item.Source, item.SubSource, item.Frequency)
  151. refreshDataNumMap[key] = item
  152. // 数据源刷新频度的列表数组
  153. subSourceFrequencyList, ok := refreshDataFrequencyListMap[item.Source]
  154. if !ok {
  155. subSourceFrequencyList = make(map[int][]string)
  156. }
  157. frequencyList, ok := subSourceFrequencyList[item.SubSource]
  158. if !ok {
  159. wgNum++
  160. frequencyList = make([]string, 0)
  161. }
  162. subSourceFrequencyList[item.SubSource] = append(frequencyList, item.Frequency)
  163. refreshDataFrequencyListMap[item.Source] = subSourceFrequencyList
  164. }
  165. for source, subSourceFrequencyListMap := range refreshDataFrequencyListMap {
  166. switch source {
  167. case utils.DATA_SOURCE_MYSTEEL_CHEMICAL, utils.DATA_SOURCE_YS:
  168. // 只处理钢联化工使用api方式获取数据的情况
  169. if mySteelChemicalDataMethod == "api" {
  170. for subSource, frequencyList := range subSourceFrequencyListMap {
  171. items, tmpErr := data_manage.GetBaseFromMysteelChemicalIndexItems(frequencyList)
  172. if tmpErr != nil {
  173. errMsgList = append(errMsgList, fmt.Sprint("source:", source, "subSource:", subSource, "frequencyList:", strings.Join(frequencyList, ","), "err:", tmpErr.Error()))
  174. }
  175. indexList := make([]*edb_refresh.EdbInfoListAndRefreshConfig, 0)
  176. for _, v := range items {
  177. tmpConf := new(edb_refresh.EdbInfoListAndRefreshConfig)
  178. // 数据刷新的期数
  179. dataRefreshNum := utils.DATA_REFRESH
  180. key := fmt.Sprintf("%d_%d_%s", utils.DATA_SOURCE_MYSTEEL_CHEMICAL, 0, v.Frequency)
  181. if edbRefreshDefaultConfig, ok := refreshDataNumMap[key]; ok {
  182. if edbRefreshDefaultConfig.RefreshAllData == 1 { // 刷新所有数据期数
  183. dataRefreshNum = 0
  184. } else if edbRefreshDefaultConfig.RefreshDataNum > 0 { //
  185. dataRefreshNum = edbRefreshDefaultConfig.RefreshDataNum
  186. }
  187. }
  188. tmpConf.EdbCode = v.IndexCode
  189. tmpConf.EdbName = v.IndexName
  190. tmpConf.Source = utils.DATA_SOURCE_MYSTEEL_CHEMICAL
  191. tmpConf.Frequency = v.Frequency
  192. tmpConf.Unit = v.Unit
  193. tmpConf.StartDate, err = time.Parse(utils.FormatDate, v.StartDate)
  194. tmpConf.EndDate, err = time.Parse(utils.FormatDate, v.EndDate)
  195. tmpConf.ClassifyId = int(v.BaseFromMysteelChemicalClassifyId)
  196. tmpConf.DataRefreshNum = dataRefreshNum
  197. tmpConf.EdbInfoId = v.EdbInfoId
  198. indexList = append(indexList, tmpConf)
  199. }
  200. key := fmt.Sprint(source, "_", subSource)
  201. sourceEdbInfoListMap[key] = indexList
  202. }
  203. }
  204. // 其他情况不处理
  205. default:
  206. for subSource, frequencyList := range subSourceFrequencyListMap {
  207. edbList, tmpErr := edb_refresh.GetDefaultRefreshEdbInfoListBySourceAndSubSource(source, subSource, frequencyList)
  208. if tmpErr != nil {
  209. errMsgList = append(errMsgList, fmt.Sprint("source:", source, "subSource:", subSource, "frequencyList:", strings.Join(frequencyList, ","), "err:", tmpErr.Error()))
  210. }
  211. for _, v := range edbList {
  212. // 数据刷新的期数
  213. dataRefreshNum := utils.DATA_REFRESH
  214. key := fmt.Sprintf("%d_%d_%s", v.Source, v.SubSource, v.Frequency)
  215. if edbRefreshDefaultConfig, ok := refreshDataNumMap[key]; ok {
  216. if edbRefreshDefaultConfig.RefreshAllData == 1 { // 刷新所有数据期数
  217. dataRefreshNum = 0
  218. } else if edbRefreshDefaultConfig.RefreshDataNum > 0 { //
  219. dataRefreshNum = edbRefreshDefaultConfig.RefreshDataNum
  220. }
  221. }
  222. v.DataRefreshNum = dataRefreshNum
  223. }
  224. key := fmt.Sprint(source, "_", subSource)
  225. sourceEdbInfoListMap[key] = edbList
  226. }
  227. }
  228. }
  229. fmt.Println("Get Refresh End")
  230. return
  231. }
  232. // getConfigRefreshData
  233. // @Description: 根据指标配置获取需要刷新的指标列表
  234. // @author: Roc
  235. // @datetime 2024-01-10 13:55:59
  236. // @param now time.Time
  237. // @return sourceEdbInfoListMap map[string][]*edb_refresh.EdbInfoListAndRefreshConfig
  238. // @return err error
  239. func getConfigRefreshData(now time.Time) (sourceEdbInfoListMap map[string][]*edb_refresh.EdbInfoListAndRefreshConfig, err error) {
  240. defer func() {
  241. if err != nil {
  242. fmt.Println(err)
  243. }
  244. }()
  245. // 一期是只做wind、同花顺、钢联、有色
  246. sourceEdbInfoListMap = make(map[string][]*edb_refresh.EdbInfoListAndRefreshConfig)
  247. currTimeStr := getPreviousHalfHour(now)
  248. // 所有默认配置刷新项
  249. list := make([]*edb_refresh.EdbRefreshConfig, 0)
  250. //刷新频率,枚举值:每自然日、每交易日、每周、每旬、每月、每季、每半年、每年
  251. refreshFrequencyList := []string{"每自然日", "每交易日", "每周", "每旬", "每月", "每季", "每半年", "每年"}
  252. // 获取各个刷新频率的配置
  253. for _, refreshFrequency := range refreshFrequencyList {
  254. // 获取刷新频率条件
  255. condition, pars, isHandler := getRefreshFrequencyCondition(now, refreshFrequency)
  256. if !isHandler {
  257. // 可能是非交易日,所以过滤不处理
  258. continue
  259. }
  260. condition += ` AND refresh_frequency = ? AND refresh_time = ?`
  261. pars = append(pars, refreshFrequency, currTimeStr)
  262. tmpList, tmpErr := edb_refresh.GetEdbRefreshConfigListByCondition(condition, pars)
  263. if tmpErr != nil {
  264. err = tmpErr
  265. return
  266. }
  267. list = append(list, tmpList...)
  268. }
  269. // 配置列表
  270. configIdEdbRefreshConfigMap := make(map[int]*edb_refresh.EdbRefreshConfig)
  271. configIdList := make([]int, 0)
  272. for _, v := range list {
  273. configIdList = append(configIdList, v.EdbRefreshConfigId)
  274. configIdEdbRefreshConfigMap[v.EdbRefreshConfigId] = v
  275. }
  276. conf, err := models.GetBusinessConf()
  277. if err != nil {
  278. fmt.Println(err)
  279. return
  280. }
  281. // 获取钢联化工的数据获取方式
  282. mySteelChemicalDataMethod := "excel"
  283. if v, ok := conf["MySteelDataMethod"]; ok {
  284. if v == "api" {
  285. mySteelChemicalDataMethod = v
  286. }
  287. }
  288. // 当钢联的数据获取方式是api时,不用过滤
  289. var sourceList []int
  290. if mySteelChemicalDataMethod == "api" {
  291. sourceList = []int{utils.DATA_SOURCE_YS}
  292. } else {
  293. sourceList = []int{utils.DATA_SOURCE_MYSTEEL_CHEMICAL, utils.DATA_SOURCE_YS}
  294. }
  295. edbInfoList, err := edb_refresh.GetConfigRefreshEdbInfoListBySourceAndSubSource(sourceList, configIdList)
  296. if err != nil {
  297. return
  298. }
  299. for _, v := range edbInfoList {
  300. key := fmt.Sprint(v.Source, "_", v.SubSource)
  301. tmpList, ok := sourceEdbInfoListMap[key]
  302. if !ok {
  303. tmpList = make([]*edb_refresh.EdbInfoListAndRefreshConfig, 0)
  304. }
  305. // 数据刷新的期数
  306. dataRefreshNum := utils.DATA_REFRESH
  307. if edbRefreshConfig, ok2 := configIdEdbRefreshConfigMap[v.EdbRefreshConfigId]; ok2 {
  308. if edbRefreshConfig.RefreshAllData == 1 { // 刷新所有数据期数
  309. dataRefreshNum = 0
  310. } else if edbRefreshConfig.RefreshDataNum > 0 { //
  311. dataRefreshNum = edbRefreshConfig.RefreshDataNum
  312. }
  313. }
  314. v.DataRefreshNum = dataRefreshNum
  315. sourceEdbInfoListMap[key] = append(tmpList, v)
  316. }
  317. fmt.Println("Get ConfigRefreshData End")
  318. return
  319. }
  320. // BaseRefreshData
  321. // @Description: 基础数据刷新
  322. // @author: Roc
  323. // @datetime 2024-01-09 16:27:45
  324. // @param wg *sync.WaitGroup
  325. // @return err error
  326. func BaseRefreshData(wg *sync.WaitGroup, source, subSource int, items []*edb_refresh.EdbInfoListAndRefreshConfig) (err error) {
  327. errMsgList := make([]string, 0)
  328. defer func() {
  329. if err != nil {
  330. fmt.Println("来源:", source, ";子来源:", subSource, ";BaseRefreshData Err:"+err.Error())
  331. go alarm_msg.SendAlarmMsg(fmt.Sprint("来源:", source, ";子来源:", subSource, ";BaseRefreshData ErrMsg:"+err.Error()), 3)
  332. }
  333. if len(errMsgList) > 0 {
  334. errMsg := fmt.Sprint("来源:", source, ";子来源:", subSource, ";BaseRefreshData Err:"+strings.Join(errMsgList, "\n"))
  335. fmt.Println(errMsg)
  336. go alarm_msg.SendAlarmMsg(errMsg, 3)
  337. }
  338. wg.Done()
  339. }()
  340. // 数据刷新的期数
  341. dataRefreshNum := utils.DATA_REFRESH
  342. // 是否从最开始的日期更新
  343. var isRefreshByStartDate bool
  344. if source != utils.DATA_SOURCE_THS {
  345. for _, v := range items {
  346. // 如果暂停更新,那就过滤
  347. if v.NoUpdate == 1 {
  348. continue
  349. }
  350. if v.DataRefreshNum > 0 {
  351. dataRefreshNum = v.DataRefreshNum
  352. }
  353. startDate := ""
  354. if isRefreshByStartDate {
  355. startDate = v.StartDate.Format(utils.FormatDate)
  356. } else {
  357. if v.Frequency == "日度" {
  358. startDate = v.EndDate.AddDate(0, 0, -dataRefreshNum).Format(utils.FormatDate)
  359. } else if v.Frequency == "周度" {
  360. startDate = v.EndDate.AddDate(0, 0, -(dataRefreshNum * 7)).Format(utils.FormatDate)
  361. } else if v.Frequency == "旬度" {
  362. startDate = v.EndDate.AddDate(0, 0, -(dataRefreshNum * 10)).Format(utils.FormatDate)
  363. } else if v.Frequency == "月度" {
  364. startDate = v.EndDate.AddDate(0, -dataRefreshNum, 0).Format(utils.FormatDate)
  365. } else if v.Frequency == "季度" {
  366. startDate = v.EndDate.AddDate(0, -dataRefreshNum*3, 0).Format(utils.FormatDate)
  367. } else if v.Frequency == "半年度" {
  368. startDate = v.EndDate.AddDate(0, -dataRefreshNum*6, 0).Format(utils.FormatDate)
  369. } else if v.Frequency == "年度" {
  370. startDate = v.EndDate.AddDate(-dataRefreshNum, 0, 0).Format(utils.FormatDate)
  371. } else {
  372. startDate = v.EndDate.AddDate(0, 0, -utils.DATA_REFRESH).Format(utils.FormatDate)
  373. }
  374. }
  375. fmt.Println(startDate)
  376. // 数据更新
  377. resp, tmpErr := data.RefreshEdbData(v.EdbInfoId, v.Source, v.SubSource, v.EdbCode, startDate)
  378. if tmpErr != nil {
  379. errMsgList = append(errMsgList, v.EdbCode+"RefreshEdbData Err:"+tmpErr.Error())
  380. continue
  381. }
  382. if resp.Ret != 200 {
  383. errMsgList = append(errMsgList, v.EdbCode+";RefreshEdbData Err:"+resp.Msg+";ErrMsg:"+resp.ErrMsg)
  384. continue
  385. }
  386. }
  387. }
  388. // 同花顺控制速率, 每秒最多4次请求(同花顺EDB函数限制为5, 考虑到可能存在用户同时在使用, 这里批量刷新设置为4, 话说5确实是有点少了吧=_=!)
  389. if source == utils.DATA_SOURCE_THS {
  390. ticker := time.NewTicker(250 * time.Millisecond)
  391. defer ticker.Stop()
  392. for _, v := range items {
  393. <-ticker.C
  394. // 如果暂停更新,那就过滤
  395. if v.NoUpdate == 1 {
  396. continue
  397. }
  398. if v.DataRefreshNum > 0 {
  399. dataRefreshNum = v.DataRefreshNum
  400. }
  401. startDate := ""
  402. if isRefreshByStartDate {
  403. startDate = v.StartDate.Format(utils.FormatDate)
  404. } else {
  405. if v.Frequency == "日度" {
  406. startDate = v.EndDate.AddDate(0, 0, -dataRefreshNum).Format(utils.FormatDate)
  407. } else if v.Frequency == "周度" {
  408. startDate = v.EndDate.AddDate(0, 0, -(dataRefreshNum * 7)).Format(utils.FormatDate)
  409. } else if v.Frequency == "旬度" {
  410. startDate = v.EndDate.AddDate(0, 0, -(dataRefreshNum * 10)).Format(utils.FormatDate)
  411. } else if v.Frequency == "月度" {
  412. startDate = v.EndDate.AddDate(0, -dataRefreshNum, 0).Format(utils.FormatDate)
  413. } else if v.Frequency == "季度" {
  414. startDate = v.EndDate.AddDate(0, -dataRefreshNum*3, 0).Format(utils.FormatDate)
  415. } else if v.Frequency == "半年度" {
  416. startDate = v.EndDate.AddDate(0, -dataRefreshNum*6, 0).Format(utils.FormatDate)
  417. } else if v.Frequency == "年度" {
  418. startDate = v.EndDate.AddDate(-dataRefreshNum, 0, 0).Format(utils.FormatDate)
  419. } else {
  420. startDate = v.EndDate.AddDate(0, 0, -utils.DATA_REFRESH).Format(utils.FormatDate)
  421. }
  422. }
  423. fmt.Println(startDate)
  424. // 数据更新
  425. resp, tmpErr := data.RefreshEdbData(v.EdbInfoId, v.Source, v.SubSource, v.EdbCode, startDate)
  426. if tmpErr != nil {
  427. errMsgList = append(errMsgList, v.EdbCode+"RefreshEdbData Err:"+tmpErr.Error())
  428. continue
  429. }
  430. if resp.Ret != 200 {
  431. errMsgList = append(errMsgList, v.EdbCode+";RefreshEdbData Err:"+resp.Msg+";ErrMsg:"+resp.ErrMsg)
  432. continue
  433. }
  434. }
  435. }
  436. fmt.Println("来源:", source, ";子来源:", subSource, "刷新结束")
  437. return err
  438. }
  439. // getRefreshFrequencyCondition
  440. // @Description: 根据时间和刷新频率获取条件
  441. // @author: Roc
  442. // @datetime 2024-01-09 16:27:11
  443. // @param now time.Time
  444. // @param refreshFrequency string
  445. // @return condition string
  446. // @return pars []interface{}
  447. // @return isHandler bool
  448. func getRefreshFrequencyCondition(now time.Time, refreshFrequency string) (condition string, pars []interface{}, isHandler bool) {
  449. isHandler = true
  450. var dayNum int
  451. var isLastDay bool
  452. //刷新频率,枚举值:每自然日、每交易日、每周、每旬、每月、每季、每半年、每年
  453. switch refreshFrequency {
  454. case "每自然日":
  455. // 自然日不需要额外条件
  456. return
  457. case "每交易日":
  458. // 周六日不处理
  459. if now.Weekday() == time.Saturday || now.Weekday() == time.Sunday {
  460. isHandler = false
  461. }
  462. return
  463. case "每周":
  464. currWeekDay := now.Weekday()
  465. if currWeekDay == time.Sunday {
  466. currWeekDay = 7
  467. isLastDay = true
  468. }
  469. dayNum = int(currWeekDay)
  470. case "每旬":
  471. currDay := now.Day()
  472. if currDay <= 10 {
  473. dayNum = currDay
  474. // 如果是这旬的最后一天
  475. if currDay == 10 {
  476. isLastDay = true
  477. }
  478. } else if currDay <= 20 {
  479. dayNum = currDay - 10
  480. // 如果是这旬的最后一天
  481. if currDay == 20 {
  482. isLastDay = true
  483. }
  484. } else {
  485. dayNum = currDay - 20
  486. // 当月的最后一天
  487. monthLastDay := time.Date(now.Year(), now.Month()+1, 1, 0, 0, 0, 0, time.Local).AddDate(0, 0, -1)
  488. // 如果是这旬的最后一天
  489. if currDay == monthLastDay.Day() {
  490. isLastDay = true
  491. }
  492. }
  493. case "每月":
  494. // 当前日期
  495. currDay := now.Day()
  496. dayNum = currDay
  497. // 当期的最后一天
  498. monthLastDay := time.Date(now.Year(), now.Month()+1, 1, 0, 0, 0, 0, time.Local).AddDate(0, 0, -1)
  499. // 如果是这期的最后一天
  500. if currDay == monthLastDay.Day() {
  501. isLastDay = true
  502. }
  503. case "每季":
  504. // 当期的第一天 ; 当期的最后一天
  505. var startDay, endDay time.Time
  506. currMonth := now.Month()
  507. currDay := now.Day()
  508. if currMonth <= 3 {
  509. // 当季的第一天
  510. startDay = time.Date(now.Year(), 1, 1, 0, 0, 0, 0, time.Local)
  511. // 当季的最后一天
  512. endDay = time.Date(now.Year(), 4, 1, 0, 0, 0, 0, time.Local).AddDate(0, 0, -1)
  513. } else if currMonth <= 6 {
  514. // 当期的第一天
  515. startDay = time.Date(now.Year(), 4, 1, 0, 0, 0, 0, time.Local)
  516. // 当期的最后一天
  517. endDay = time.Date(now.Year(), 7, 1, 0, 0, 0, 0, time.Local).AddDate(0, 0, -1)
  518. } else if currMonth <= 9 {
  519. // 当期的第一天
  520. startDay = time.Date(now.Year(), 7, 1, 0, 0, 0, 0, time.Local)
  521. // 当期的最后一天
  522. endDay = time.Date(now.Year(), 10, 1, 0, 0, 0, 0, time.Local).AddDate(0, 0, -1)
  523. } else {
  524. // 当期的第一天
  525. startDay = time.Date(now.Year(), 10, 1, 0, 0, 0, 0, time.Local)
  526. // 当期的最后一天
  527. endDay = time.Date(now.Year()+1, 1, 1, 0, 0, 0, 0, time.Local).AddDate(0, 0, -1)
  528. }
  529. // 计算这期的第一天和当日的天数
  530. dayNum = utils.GetTimeSubDay(startDay, now) + 1
  531. // 如果是这期的最后一天
  532. if currMonth == endDay.Month() && currDay == endDay.Day() {
  533. isLastDay = true
  534. }
  535. case "每半年":
  536. // 当期的第一天 ; 当期的最后一天
  537. var startDay, endDay time.Time
  538. currMonth := now.Month()
  539. currDay := now.Day()
  540. if currMonth <= 6 {
  541. // 当期的第一天
  542. startDay = time.Date(now.Year(), 1, 1, 0, 0, 0, 0, time.Local)
  543. // 当期的最后一天
  544. endDay = time.Date(now.Year(), 7, 1, 0, 0, 0, 0, time.Local).AddDate(0, 0, -1)
  545. } else {
  546. // 当期的第一天
  547. startDay = time.Date(now.Year(), 7, 1, 0, 0, 0, 0, time.Local)
  548. // 当期的最后一天
  549. endDay = time.Date(now.Year()+1, 1, 1, 0, 0, 0, 0, time.Local).AddDate(0, 0, -1)
  550. }
  551. // 计算这期的第一天和当日的天数
  552. dayNum = utils.GetTimeSubDay(startDay, now) + 1
  553. // 如果是这期的最后一天
  554. if currMonth == endDay.Month() && currDay == endDay.Day() {
  555. isLastDay = true
  556. }
  557. case "每年":
  558. currMonth := now.Month()
  559. currDay := now.Day()
  560. // 当期的第一天
  561. startDay := time.Date(now.Year(), 1, 1, 0, 0, 0, 0, time.Local)
  562. // 当期的最后一天
  563. endDay := time.Date(now.Year()+1, 1, 1, 0, 0, 0, 0, time.Local).AddDate(0, 0, -1)
  564. // 计算这期的第一天和当日的天数
  565. dayNum = utils.GetTimeSubDay(startDay, now) + 1
  566. // 如果是这期的最后一天
  567. if currMonth == endDay.Month() && currDay == endDay.Day() {
  568. isLastDay = true
  569. }
  570. }
  571. // 如果是这期的最后一天,那么就是判断refresh_frequency_day是否配置为0,或者配置的天数大于这期的最大天数
  572. if isLastDay {
  573. condition += ` AND ( refresh_frequency_day = ? OR refresh_frequency_day >= ? )`
  574. pars = append(pars, 0, dayNum)
  575. } else {
  576. // 如果不是这期的最后一天,那么就是判断refresh_frequency_day是否等于配置的天数
  577. condition += ` AND refresh_frequency_day = ? `
  578. pars = append(pars, dayNum)
  579. }
  580. return
  581. }
  582. // getPreviousHalfHour
  583. // @Description: 根据传入的时间获取该时间的前整半小时的时间字符串
  584. // @author: Roc
  585. // @datetime 2024-01-09 14:27:34
  586. // @param now time.Time
  587. // @return string
  588. func getPreviousHalfHour(now time.Time) string {
  589. minute := now.Minute()
  590. if minute >= 30 {
  591. return fmt.Sprintf("%02d:%02d", now.Hour(), 30)
  592. }
  593. return fmt.Sprintf("%02d:%02d", now.Hour(), 0)
  594. }