waybill_task.go 9.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322
  1. package service
  2. import (
  3. "cold-logistics/app/admin/model"
  4. "cold-logistics/app/admin/service/dto"
  5. cDto "cold-logistics/common/dto"
  6. "cold-logistics/common/global"
  7. model2 "cold-logistics/common/model"
  8. "cold-logistics/common/nats/nats_server"
  9. "errors"
  10. "fmt"
  11. "gogs.baozhida.cn/zoie/OAuth-core/service"
  12. "gorm.io/gorm"
  13. "sort"
  14. "time"
  15. )
  16. type WaybillTask struct {
  17. service.Service
  18. }
  19. func WaybillTaskTimeScopes(startTime, endTime string) func(db *gorm.DB) *gorm.DB {
  20. return func(db *gorm.DB) *gorm.DB {
  21. if len(startTime) == 0 && len(endTime) == 0 {
  22. return db
  23. }
  24. return db.Where("(start_time between ? and ?) or (end_time between ? and ?)",
  25. startTime, endTime, startTime, endTime)
  26. }
  27. }
  28. // GetPage 获取WaybillTask列表
  29. func (e *WaybillTask) GetPage(c *dto.WaybillTaskGetPageReq, list *[]model.WaybillTask, count *int64) error {
  30. var err error
  31. var data model.WaybillTask
  32. var waybill model.Waybill
  33. err = e.Orm.Model(&waybill).Where("waybill_no = ?", c.WaybillNo).First(&waybill).Error
  34. if err != nil {
  35. e.Log.Errorf("db error: %s", err)
  36. return errors.New("获取运单信息失败")
  37. }
  38. err = e.Orm.Model(&data).
  39. Scopes(
  40. cDto.MakeCondition(c.GetNeedSearch()),
  41. ).
  42. Preload("Warehouse").Preload("Car").Preload("CoolerBox").
  43. Find(list).Limit(-1).Offset(-1).
  44. Count(count).Error
  45. if err != nil {
  46. e.Log.Errorf("db error: %s", err)
  47. return global.GetFailedErr
  48. }
  49. // 获取公司秘钥
  50. var company model.SysDept
  51. company, err = model.GetCompanyById(waybill.DeptId)
  52. if err != nil {
  53. e.Log.Errorf("db error: %s", err)
  54. return model.GetCompanyKeyErr
  55. }
  56. for i := 0; i < len(*list); i++ {
  57. // 获取传感器信息
  58. deviceSensorList, _, _ := nats_server.Cold_CompanyDeviceSensor_List_ByKey((*list)[i].Sn, company.ColdKey)
  59. (*list)[i].DeviceSensorList = deviceSensorList
  60. }
  61. return nil
  62. }
  63. // GetPage 获取WaybillTask列表
  64. func (e *WaybillTask) GetDataPage(c *dto.WaybillTaskGetDataPageReq) (list []nats_server.DeviceData_R, count int64, err error) {
  65. var waybill model.Waybill
  66. err = e.Orm.Where("waybill_no = ?", c.WaybillNo).First(&waybill).Error
  67. if err != nil {
  68. e.Log.Errorf("db error: %s", err)
  69. err = errors.New("获取运单信息错误!")
  70. return
  71. }
  72. var task model.WaybillTask
  73. err = e.Orm.Where("id = ? and waybill_no = ?", c.TaskId, c.WaybillNo).First(&task).Error
  74. if err != nil {
  75. e.Log.Errorf("db error: %s", err)
  76. err = errors.New("获取运单信息错误!")
  77. return
  78. }
  79. // 获取最后一个任务id
  80. var lastWaybillTask model.WaybillTask
  81. err = e.Orm.Model(&lastWaybillTask).Where("waybill_no = ?", c.WaybillNo).Last(&lastWaybillTask).Error
  82. if err != nil {
  83. e.Log.Errorf("db error: %s", err)
  84. err = errors.New("获取运单信息错误!")
  85. return
  86. }
  87. var afterStartTime bool
  88. var beforeEndTime bool
  89. if len(c.StartTime) > 0 {
  90. var st time.Time
  91. st, err = time.ParseInLocation("2006-01-02 15:04:05", c.StartTime, time.Local)
  92. if err != nil {
  93. err = errors.New("解析时间字符串出错!")
  94. return
  95. }
  96. fmt.Println("st", st)
  97. fmt.Println("task.StartTime", task.StartTime.Local())
  98. if task.StartTime.Local().After(st) || st.Truncate(time.Second).Equal(task.StartTime.Local().Truncate(time.Second)) {
  99. c.StartTime = task.StartTime.String()
  100. } else {
  101. afterStartTime = true
  102. }
  103. } else {
  104. c.StartTime = task.StartTime.String()
  105. }
  106. if len(c.EndTime) > 0 {
  107. var et time.Time
  108. et, err = time.ParseInLocation("2006-01-02 15:04:05", c.EndTime, time.Local)
  109. if err != nil {
  110. err = errors.New("解析时间字符串出错!")
  111. return
  112. }
  113. fmt.Println("et", et)
  114. fmt.Println("task.EndTime", task.EndTime.Local())
  115. if et.After(task.EndTime.Local()) || et.Truncate(time.Second).Equal(task.EndTime.Local().Truncate(time.Second)) {
  116. c.EndTime = task.EndTime.String()
  117. } else {
  118. beforeEndTime = true
  119. }
  120. } else {
  121. c.EndTime = task.EndTime.String()
  122. }
  123. T_snid := ""
  124. for _, id := range c.T_ids {
  125. T_snid += fmt.Sprintf("%s,%d|", task.Sn, id)
  126. }
  127. list, count, err = nats_server.Cold_ReadDeviceDataListBy_T_snid(T_snid, c.StartTime, c.EndTime, 0, 9999)
  128. firstMap := map[int]nats_server.DeviceData_R{}
  129. lastMap := map[int]nats_server.DeviceData_R{}
  130. if count > 0 {
  131. for _, v := range c.T_ids {
  132. for i := 0; i < len(list); i++ {
  133. if afterStartTime {
  134. break
  135. }
  136. if v == list[i].T_id {
  137. if list[i].T_time != task.StartTime.String() {
  138. data := list[len(list)-1-i]
  139. data.T_time = task.StartTime.String()
  140. firstMap[v] = data
  141. count += 1
  142. }
  143. break
  144. }
  145. }
  146. if waybill.Status == model.WaybillStatusReceipt {
  147. for i := len(list) - 1; i >= 0; i-- {
  148. if beforeEndTime {
  149. break
  150. }
  151. if v == list[i].T_id {
  152. if c.TaskId == lastWaybillTask.Id && list[i].T_time != task.EndTime.String() && !task.EndTime.Local().IsZero() {
  153. data := list[len(list)-1-i]
  154. data.T_time = task.EndTime.String()
  155. lastMap[v] = data
  156. count += 1
  157. }
  158. break
  159. }
  160. }
  161. }
  162. }
  163. }
  164. for _, data := range firstMap {
  165. list = append(list, data)
  166. }
  167. for _, data := range lastMap {
  168. list = append(list, data)
  169. }
  170. sort.Slice(list, func(i, j int) bool {
  171. // 先按 T_time 字段排序,如果 T_time 相同则按 T_name 字段排序
  172. if list[i].T_time == list[j].T_time {
  173. return list[i].T_name < list[j].T_name
  174. }
  175. return list[i].T_time > list[j].T_time
  176. })
  177. // 计算总页数
  178. totalPages := (int(count) + c.PageSize - 1) / c.PageSize
  179. // 如果n超出范围,返回错误
  180. if c.Page > totalPages {
  181. return []nats_server.DeviceData_R{}, count, err
  182. }
  183. if c.Page < 1 {
  184. c.Page = 1
  185. }
  186. // 计算当前页的数据
  187. start := (c.Page - 1) * c.PageSize
  188. end := start + c.PageSize
  189. if end > int(count) {
  190. end = int(count)
  191. }
  192. currentPage := list[start:end]
  193. return currentPage, count, err
  194. }
  195. func (e *WaybillTask) GetNewestDataPage(c *dto.WaybillTaskGetNewestDataPageReq) (data nats_server.DeviceData_R, err error) {
  196. var waybill model.Waybill
  197. err = e.Orm.Model(&waybill).Where("waybill_no = ?", c.WaybillNo).First(&waybill).Error
  198. if err != nil {
  199. e.Log.Errorf("db error: %s", err)
  200. err = errors.New("获取运单信息错误!")
  201. return
  202. }
  203. // 获取公司秘钥
  204. var company model.SysDept
  205. company, err = model.GetCompanyById(waybill.DeptId)
  206. if err != nil {
  207. e.Log.Errorf("db error: %s", err)
  208. err = model.GetCompanyKeyErr
  209. return
  210. }
  211. var taskList []model.WaybillTask
  212. err = e.Orm.Where("waybill_no = ?", c.WaybillNo).Order("id desc").Find(&taskList).Error
  213. if err != nil {
  214. e.Log.Errorf("db error: %s", err)
  215. err = errors.New("获取运单信息任务失败!")
  216. return
  217. }
  218. if len(taskList) == 0 {
  219. err = errors.New("暂无温湿度记录信息!")
  220. return
  221. }
  222. var listAll []nats_server.DeviceData_R
  223. for _, task := range taskList {
  224. deviceSensorList, _, _ := nats_server.Cold_CompanyDeviceSensor_List_ByKey(task.Sn, company.ColdKey)
  225. if len(deviceSensorList) == 0 {
  226. continue
  227. }
  228. T_snid := fmt.Sprintf("%s,%d|", deviceSensorList[0].T_sn, deviceSensorList[0].T_id)
  229. if len(task.EndTime.String()) == 0 {
  230. task.EndTime = model2.Time(time.Now())
  231. }
  232. // 获取传感器信息
  233. list, _, _ := nats_server.Cold_ReadDeviceDataListBy_T_snid(T_snid, task.StartTime.String(), task.EndTime.String(), 0, 9999)
  234. if len(list) > 0 {
  235. listAll = append(listAll, list...)
  236. break
  237. }
  238. }
  239. if len(listAll) > 0 {
  240. sort.Slice(listAll, func(i, j int) bool {
  241. return listAll[i].T_time > listAll[j].T_time
  242. })
  243. data = listAll[0]
  244. return
  245. }
  246. err = errors.New("暂无温湿度记录信息!")
  247. return
  248. }
  249. func (e *WaybillTask) GetLocus(c *dto.WaybillGetLocusReq) ([]nats_server.DeviceData_R2, error) {
  250. var err error
  251. var data model.WaybillTask
  252. var waybill model.Waybill
  253. var taskList []model.WaybillTask
  254. locusList := make([]nats_server.DeviceData_R2, 0)
  255. err = e.Orm.Model(&waybill).Where("waybill_no = ?", c.WaybillNo).First(&waybill).Error
  256. if err != nil {
  257. e.Log.Errorf("db error: %s", err)
  258. return locusList, errors.New("获取运单信息失败")
  259. }
  260. // 获取公司秘钥
  261. var company model.SysDept
  262. company, err = model.GetCompanyById(waybill.DeptId)
  263. if err != nil {
  264. e.Log.Errorf("db error: %s", err)
  265. return locusList, model.GetCompanyKeyErr
  266. }
  267. err = e.Orm.Model(&data).
  268. Scopes(
  269. cDto.MakeCondition(c.GetNeedSearch()),
  270. ).Where("car_id > 0 or cooler_box_id > 0 ").
  271. Find(&taskList).Error
  272. if err != nil {
  273. e.Log.Errorf("db error: %s", err)
  274. return locusList, global.GetFailedErr
  275. }
  276. for i := 0; i < len(taskList); i++ {
  277. // 获取传感器信息
  278. deviceSensorList, _, _ := nats_server.Cold_CompanyDeviceSensor_List_ByKey(taskList[i].Sn, company.ColdKey)
  279. if len(deviceSensorList) > 0 {
  280. T_snid := fmt.Sprintf("%s,%d|", taskList[i].Sn, deviceSensorList[0].T_id)
  281. dataList, _, err := nats_server.Cold_ReadDeviceDataListBy_T_snidForLocus(T_snid, taskList[i].StartTime.String(), taskList[i].EndTime.String(), 0, 9999)
  282. if err != nil {
  283. e.Log.Errorf("nats 获取轨迹信息失败: %s", err)
  284. return locusList, global.GetFailedErr
  285. }
  286. // 倒序
  287. sort.Slice(dataList, func(i, j int) bool {
  288. if dataList[i].T_time < dataList[j].T_time {
  289. return true
  290. }
  291. return false
  292. })
  293. locusList = append(locusList, dataList...)
  294. }
  295. }
  296. return locusList, nil
  297. }