TaskData.go 41 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481
  1. package controllers
  2. import (
  3. "ColdVerify_server/Nats"
  4. "ColdVerify_server/conf"
  5. "ColdVerify_server/lib"
  6. "ColdVerify_server/logs"
  7. "ColdVerify_server/models/Account"
  8. "ColdVerify_server/models/Device"
  9. "ColdVerify_server/models/System"
  10. "ColdVerify_server/models/Task"
  11. "errors"
  12. "fmt"
  13. beego "github.com/beego/beego/v2/server/web"
  14. "github.com/signintech/gopdf"
  15. "github.com/xuri/excelize/v2"
  16. "log"
  17. "math"
  18. "os"
  19. "sort"
  20. "strconv"
  21. "strings"
  22. "sync"
  23. "time"
  24. )
  25. type TaskDataController struct {
  26. beego.Controller
  27. }
  28. // 获取-
  29. func (c *TaskDataController) Extract_TaskData() {
  30. // 验证登录 User_is, User_r
  31. user_r, User_is := Account.Verification_Admin(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  32. if !User_is {
  33. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  34. c.ServeJSON()
  35. return
  36. }
  37. Time_start := c.GetString("Time_start")
  38. Time_end := c.GetString("Time_end")
  39. T_task_id := c.GetString("T_task_id")
  40. Task_r, is := Task.Read_Task(T_task_id)
  41. if !is {
  42. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_task_id 错误!"}
  43. c.ServeJSON()
  44. return
  45. }
  46. DeviceClass_List := Device.Read_DeviceClassList_List_id(Task_r.T_class)
  47. if len(DeviceClass_List) == 0 {
  48. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_Class_id 分类设备列表 为空!"}
  49. c.ServeJSON()
  50. return
  51. }
  52. logs.Println("----导入 :", DeviceClass_List)
  53. // 清空表
  54. Task.Truncate_TaskData(Task_r.T_task_id)
  55. // 插入 数据
  56. for _, v := range DeviceClass_List {
  57. logs.Println("---- :", v.T_sn, Task_r.T_task_id, Time_start, Time_end)
  58. Task.Import_TaskData(v.T_sn, v.T_id, Task_r.T_task_id, Time_start, Time_end)
  59. }
  60. // 提取数据后 将 当前任务 数据采集 标志 为 1
  61. Task_r.T_collection_state = 1
  62. if !Task.Update_Task(Task_r, "T_collection_state") {
  63. c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改失败!"}
  64. c.ServeJSON()
  65. return
  66. }
  67. System.Add_UserLogs_T(user_r.T_uuid, "任务", "修改", Task_r)
  68. System.Add_UserLogs(user_r.T_uuid, "提取数据", "提取数据"+Task_r.T_name, Task_r.T_task_id+"|"+Time_start+"|"+Time_end)
  69. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!"}
  70. c.ServeJSON()
  71. return
  72. }
  73. // 后台导出数据
  74. func (c *TaskDataController) Extract_TaskData_Back() {
  75. // 验证登录 User_is, User_r
  76. user_r, User_is := Account.Verification_Admin(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  77. if !User_is {
  78. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  79. c.ServeJSON()
  80. return
  81. }
  82. Time_start := c.GetString("Time_start")
  83. Time_end := c.GetString("Time_end")
  84. T_task_id := c.GetString("T_task_id")
  85. Task_r, is := Task.Read_Task(T_task_id)
  86. if !is {
  87. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_task_id 错误!"}
  88. c.ServeJSON()
  89. return
  90. }
  91. // 采集中
  92. if Task_r.T_collection_state == 2 {
  93. c.Data["json"] = lib.JSONS{Code: 202, Msg: "数据正采集中,请勿重复提交!"}
  94. c.ServeJSON()
  95. return
  96. }
  97. DeviceClass_List := Device.Read_DeviceClassList_List_id(Task_r.T_class)
  98. if len(DeviceClass_List) == 0 {
  99. c.Data["json"] = lib.JSONS{Code: 202, Msg: "设备列表为空,请先添加设备!"}
  100. c.ServeJSON()
  101. return
  102. }
  103. // 更新状态为采集中
  104. Task_r.T_collection_state = 2
  105. if !Task.Update_Task(Task_r, "T_collection_state") {
  106. c.Data["json"] = lib.JSONS{Code: 202, Msg: "导出数据失败!"}
  107. c.ServeJSON()
  108. return
  109. }
  110. resp := Nats.Extract_TaskData_Back{
  111. T_uuid: user_r.T_uuid,
  112. Time_start: Time_start,
  113. Time_end: Time_end,
  114. DeviceClassList: DeviceClass_List,
  115. Task: Task_r,
  116. }
  117. // 后台执行打包数据
  118. //b, _ := msgpack.Marshal(&data)
  119. //_ = lib.Nats.Publish("ColdVerify_Server_Extract_TaskData_Back", b)
  120. go func(resp Nats.Extract_TaskData_Back, Task_r Task.Task) {
  121. Task_rname := resp.Task
  122. // 清空表
  123. Task.Truncate_TaskData(Task_rname.T_task_id)
  124. //失败重试5次
  125. DeviceClassList := new(sync.Map)
  126. var count int
  127. for _, v := range resp.DeviceClassList {
  128. if strings.Contains(v.T_sn, "-") || len(v.T_sn) == 0 {
  129. // 从3.0平台导入
  130. continue
  131. }
  132. DeviceClassList.Store(fmt.Sprintf("%s|%s", v.T_sn, v.T_id), 5)
  133. //err = Task.Import_TaskData_Back(v.T_sn, v.T_id, resp.Task.T_task_id, resp.Time_start, resp.Time_end)
  134. //count++
  135. //time.Sleep(5 * time.Second)
  136. }
  137. DeviceClassList.Range(func(k, v interface{}) bool {
  138. count++
  139. return true
  140. })
  141. for count > 0 {
  142. DeviceClassList.Range(func(k, v any) bool {
  143. T_snid := strings.Split(k.(string), "|")
  144. T_sn := T_snid[0]
  145. T_id := T_snid[1]
  146. temp := v.(int)
  147. temp--
  148. DeviceClassList.Store(k, temp)
  149. err := Task.Import_TaskData_Back(T_sn, T_id, resp.Task.T_task_id, resp.Time_start, resp.Time_end)
  150. if err == nil || strings.Contains(err.Error(), "doesn't exist") {
  151. DeviceClassList.Delete(k)
  152. count--
  153. } else {
  154. logs.Error("设备数据同步到任务数据失败", err)
  155. DeviceClassList.Delete(k)
  156. count--
  157. }
  158. return true
  159. })
  160. }
  161. // 提取数据后 将 当前任务 数据采集 标志 为 1
  162. Task_r.T_collection_state = 1
  163. if !Task.Update_Task(Task_r, "T_collection_state") {
  164. c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改失败!"}
  165. c.ServeJSON()
  166. return
  167. }
  168. err := Task.CorrectionSecond(T_task_id)
  169. if err != nil {
  170. logs.Error(lib.FuncName(), err)
  171. return
  172. }
  173. // 删除重复数据
  174. _, err = Task.DeleteDeduplicate(T_task_id)
  175. if err != nil {
  176. logs.Error(lib.FuncName(), "sql 删除重复数据失败")
  177. return
  178. }
  179. }(resp, Task_r)
  180. System.Add_UserLogs_T(resp.T_uuid, "任务", "修改", Task_r)
  181. System.Add_UserLogs(resp.T_uuid, "提取数据", "提取数据"+Task_r.T_name, Task_r.T_task_id+"|"+resp.Time_start+"|"+resp.Time_end)
  182. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!"}
  183. c.ServeJSON()
  184. return
  185. }
  186. // 列表 -
  187. func (c *TaskDataController) TaskData_List() {
  188. // 验证登录 User_is, User_r
  189. _, User_is := Account.Verification_Admin(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  190. if !User_is {
  191. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  192. c.ServeJSON()
  193. return
  194. }
  195. var r_jsons lib.R_JSONS
  196. page, _ := c.GetInt("page")
  197. if page < 1 {
  198. page = 1
  199. }
  200. page_z, _ := c.GetInt("page_z")
  201. if page_z < 1 {
  202. page_z = conf.Page_size
  203. }
  204. Time_start := c.GetString("Time_start")
  205. Time_end := c.GetString("Time_end")
  206. T_sn := c.GetString("T_sn")
  207. T_id := c.GetString("T_id")
  208. T_layout_no := c.GetString("T_layout_no") // 替换之前的T_id
  209. if len(T_layout_no) > 0 {
  210. T_id = T_layout_no
  211. }
  212. T_task_id := c.GetString("T_task_id")
  213. Task_r, is := Task.Read_Task(T_task_id)
  214. if !is {
  215. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_task_id 错误!"}
  216. c.ServeJSON()
  217. return
  218. }
  219. if Task_r.T_delivery_state == 2 {
  220. c.Data["json"] = lib.JSONS{Code: 202, Msg: "数据采集中,请稍后!"}
  221. c.ServeJSON()
  222. return
  223. }
  224. // 查询设备列表
  225. dcList, _ := Device.Read_DeviceClassList_OrderList(Task_r.T_class, T_sn, "", "", page, 9999)
  226. // 保存布局编号和校准证书对应关系
  227. var deviceCertificateMap = make(map[string]string) // t_id, T_Certificate_sn
  228. for _, v := range dcList {
  229. deviceCertificateMap[v.T_id] = v.T_Certificate_sn
  230. }
  231. var cnt int64
  232. List, cnt := Task.Read_TaskData_ById_List(Task_r.T_task_id, T_sn, T_id, Time_start, Time_end, page, page_z)
  233. for i := 0; i < len(List); i++ {
  234. List[i].T_Certificate_sn = deviceCertificateMap[List[i].T_id]
  235. }
  236. page_size := math.Ceil(float64(cnt) / float64(page_z))
  237. r_jsons.List = List
  238. r_jsons.Page = page
  239. r_jsons.Page_size = int(page_size)
  240. r_jsons.Pages = lib.Func_page(int64(page), int64(page_size))
  241. r_jsons.Num = int(cnt)
  242. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
  243. c.ServeJSON()
  244. return
  245. }
  246. // 列表 -
  247. func (c *TaskDataController) UserTaskData_List() {
  248. // 验证登录 User_is, User_r
  249. User_r, User_is := Account.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  250. if !User_is {
  251. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  252. c.ServeJSON()
  253. return
  254. }
  255. var r_jsons lib.R_JSONS
  256. page, _ := c.GetInt("page")
  257. if page < 1 {
  258. page = 1
  259. }
  260. page_z, _ := c.GetInt("page_z")
  261. if page_z < 1 {
  262. page_z = conf.Page_size
  263. }
  264. Time_start := c.GetString("Time_start")
  265. Time_end := c.GetString("Time_end")
  266. T_sn := c.GetString("T_sn")
  267. T_id := c.GetString("T_id")
  268. T_layout_no := c.GetString("T_layout_no") // 替换之前的T_id
  269. if len(T_layout_no) > 0 {
  270. T_id = T_layout_no
  271. }
  272. T_task_id := c.GetString("T_task_id")
  273. Task_r, is := Task.Read_Task(T_task_id)
  274. if !is {
  275. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_task_id 错误!"}
  276. c.ServeJSON()
  277. return
  278. }
  279. if Task_r.T_delivery_state == 2 {
  280. c.Data["json"] = lib.JSONS{Code: 202, Msg: "数据采集中,请稍后!"}
  281. c.ServeJSON()
  282. return
  283. }
  284. if Task_r.T_uuid != User_r.T_uuid {
  285. c.Data["json"] = lib.JSONS{Code: 200, Msg: "无权访问!"}
  286. c.ServeJSON()
  287. return
  288. }
  289. if len(Time_start) == 0 {
  290. Time_start = Task_r.T_VerifyDeviceDataStartTime
  291. }
  292. if len(Time_end) == 0 {
  293. Time_end = Task_r.T_VerifyDeviceDataEndTime
  294. }
  295. // 查询设备列表
  296. dcList, _ := Device.Read_DeviceClassList_OrderList(Task_r.T_class, T_sn, "", "", page, 9999)
  297. // 保存布局编号和校准证书对应关系
  298. var deviceCertificateMap = make(map[string]string) // t_id, T_Certificate_sn
  299. for _, v := range dcList {
  300. deviceCertificateMap[v.T_id] = v.T_Certificate_sn
  301. }
  302. var cnt int64
  303. List, cnt := Task.Read_TaskData_ById_List(Task_r.T_task_id, T_sn, T_id, Time_start, Time_end, page, page_z)
  304. for i := 0; i < len(List); i++ {
  305. List[i].T_Certificate_sn = deviceCertificateMap[List[i].T_id]
  306. }
  307. page_size := math.Ceil(float64(cnt) / float64(page_z))
  308. r_jsons.List = List
  309. r_jsons.Page = page
  310. r_jsons.Page_size = int(page_size)
  311. r_jsons.Pages = lib.Func_page(int64(page), int64(page_size))
  312. r_jsons.Num = int(cnt)
  313. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
  314. c.ServeJSON()
  315. return
  316. }
  317. // 列表 -
  318. func (c *TaskDataController) TaskDataClass_List() {
  319. // 验证登录 User_is, User_r
  320. _, User_is := Account.Verification_Admin(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  321. if !User_is {
  322. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  323. c.ServeJSON()
  324. return
  325. }
  326. T_task_id := c.GetString("T_task_id")
  327. Task_r, is := Task.Read_Task(T_task_id)
  328. if !is {
  329. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_task_id 错误!"}
  330. c.ServeJSON()
  331. return
  332. }
  333. if Task_r.T_delivery_state == 2 {
  334. c.Data["json"] = lib.JSONS{Code: 202, Msg: "数据采集中,请稍后!"}
  335. c.ServeJSON()
  336. return
  337. }
  338. List := Task.Read_TaskData_ById_ClassList(Task_r.T_task_id)
  339. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: List}
  340. c.ServeJSON()
  341. return
  342. }
  343. // 添加-
  344. func (c *TaskDataController) TaskData_AddS() {
  345. // 验证登录 User_is, User_r
  346. user_r, User_is := Account.Verification_Admin(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  347. if !User_is {
  348. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  349. c.ServeJSON()
  350. return
  351. }
  352. T_task_id := c.GetString("T_task_id")
  353. Task_r, is := Task.Read_Task(T_task_id)
  354. if !is {
  355. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_task_id 错误!"}
  356. c.ServeJSON()
  357. return
  358. }
  359. //T_sn|T_id|T_t|T_rh|T_time?
  360. T_Data := c.GetString("T_Data")
  361. if len(T_Data) < 5 {
  362. c.Data["json"] = lib.JSONS{Code: 202, Msg: "err T_Data!"}
  363. c.ServeJSON()
  364. return
  365. }
  366. T_Data_list := strings.Split(T_Data, "?")
  367. println(len(T_Data_list), "len(T_Data_list)")
  368. var T_Data_list_x = 0
  369. for _, v := range T_Data_list {
  370. // 2022-08-09 14:25:00|27.7|55.2
  371. if len(v) < 5 {
  372. println(v, "len(v) < 5")
  373. continue
  374. }
  375. v_list := strings.Split(v, "|")
  376. is = Task.Add_TaskData(Task_r.T_task_id, v_list[0], v_list[1], v_list[2], v_list[3], v_list[4])
  377. if is {
  378. T_Data_list_x += 1
  379. }
  380. }
  381. System.Add_UserLogs(user_r.T_uuid, "任务数据", "添加数据"+Task_r.T_name, Task_r.T_task_id+"结果:"+string(len(T_Data_list))+"/"+string(T_Data_list_x)+"|=> "+T_Data)
  382. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: strconv.Itoa(T_Data_list_x)}
  383. c.ServeJSON()
  384. return
  385. }
  386. // 添加-
  387. func (c *TaskDataController) TaskData_Add() {
  388. // 验证登录 User_is, User_r
  389. user_r, User_is := Account.Verification_Admin(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  390. if !User_is {
  391. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  392. c.ServeJSON()
  393. return
  394. }
  395. T_sn := c.GetString("T_sn")
  396. T_id, _ := c.GetInt("T_id")
  397. T_t, _ := c.GetFloat("T_t")
  398. T_rh, _ := c.GetFloat("T_rh")
  399. T_time := c.GetString("T_time")
  400. T_task_id := c.GetString("T_task_id")
  401. Task_r, is := Task.Read_Task(T_task_id)
  402. if !is {
  403. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_task_id 错误!"}
  404. c.ServeJSON()
  405. return
  406. }
  407. is = Task.Add_TaskData(Task_r.T_task_id, T_sn, strconv.Itoa(T_id), fmt.Sprintf("%.2f", T_t), fmt.Sprintf("%.2f", T_rh), T_time)
  408. if !is {
  409. c.Data["json"] = lib.JSONS{Code: 202, Msg: "添加失败!"}
  410. c.ServeJSON()
  411. return
  412. }
  413. System.Add_UserLogs(user_r.T_uuid, "任务数据", "添加数据"+Task_r.T_name, Task_r.T_task_id+"|"+T_sn+"|"+strconv.Itoa(T_id)+"|"+fmt.Sprintf("%.2f", T_t)+"|"+fmt.Sprintf("%.2f", T_rh)+T_time)
  414. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!"}
  415. c.ServeJSON()
  416. return
  417. }
  418. // 修改-
  419. func (c *TaskDataController) TaskData_Up() {
  420. // 验证登录 User_is, User_r
  421. user_r, User_is := Account.Verification_Admin(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  422. if !User_is {
  423. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  424. c.ServeJSON()
  425. return
  426. }
  427. Id, err := c.GetInt("Id")
  428. if err != nil {
  429. c.Data["json"] = lib.JSONS{Code: 202, Msg: "Id 错误!"}
  430. c.ServeJSON()
  431. return
  432. }
  433. T_t, err := c.GetFloat("T_t")
  434. if err != nil {
  435. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_t 错误!"}
  436. c.ServeJSON()
  437. return
  438. }
  439. T_rh, err := c.GetFloat("T_rh")
  440. if err != nil {
  441. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_rh 错误!"}
  442. c.ServeJSON()
  443. return
  444. }
  445. T_time := c.GetString("T_time")
  446. T_task_id := c.GetString("T_task_id")
  447. Task_r, is := Task.Read_Task(T_task_id)
  448. if !is {
  449. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_task_id 错误!"}
  450. c.ServeJSON()
  451. return
  452. }
  453. is = Task.Up_TaskData(Task_r.T_task_id, strconv.Itoa(Id), fmt.Sprintf("%.2f", T_t), fmt.Sprintf("%.2f", T_rh), T_time)
  454. if !is {
  455. c.Data["json"] = lib.JSONS{Code: 202, Msg: "修改失败!"}
  456. c.ServeJSON()
  457. return
  458. }
  459. System.Add_UserLogs(user_r.T_uuid, "任务数据", "修改数据"+Task_r.T_name, Task_r.T_task_id+"|"+strconv.Itoa(Id)+"|"+fmt.Sprintf("%.2f", T_t)+"|"+fmt.Sprintf("%.2f", T_rh)+T_time)
  460. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!"}
  461. c.ServeJSON()
  462. return
  463. }
  464. // 删除-
  465. func (c *TaskDataController) TaskData_Del() {
  466. // 验证登录 User_is, User_r
  467. user_r, User_is := Account.Verification_Admin(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  468. if !User_is {
  469. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  470. c.ServeJSON()
  471. return
  472. }
  473. Id, err := c.GetInt("Id")
  474. if err != nil {
  475. c.Data["json"] = lib.JSONS{Code: 202, Msg: "Id 错误!"}
  476. c.ServeJSON()
  477. return
  478. }
  479. T_task_id := c.GetString("T_task_id")
  480. Task_r, is := Task.Read_Task(T_task_id)
  481. if !is {
  482. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_task_id 错误!"}
  483. c.ServeJSON()
  484. return
  485. }
  486. is = Task.Del_TaskData(Task_r.T_task_id, strconv.Itoa(Id))
  487. if !is {
  488. c.Data["json"] = lib.JSONS{Code: 202, Msg: "删除失败!"}
  489. c.ServeJSON()
  490. return
  491. }
  492. System.Add_UserLogs(user_r.T_uuid, "任务数据", "删除"+Task_r.T_name, Task_r.T_task_id+"|"+strconv.Itoa(Id))
  493. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!"}
  494. c.ServeJSON()
  495. return
  496. }
  497. // 删除-
  498. func (c *TaskDataController) TaskData_Del_t_id() {
  499. // 验证登录 User_is, User_r
  500. user_r, User_is := Account.Verification_Admin(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  501. if !User_is {
  502. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  503. c.ServeJSON()
  504. return
  505. }
  506. Id, err := c.GetInt("Id")
  507. if err != nil {
  508. c.Data["json"] = lib.JSONS{Code: 202, Msg: "Id 错误!"}
  509. c.ServeJSON()
  510. return
  511. }
  512. T_task_id := c.GetString("T_task_id")
  513. Task_r, is := Task.Read_Task(T_task_id)
  514. if !is {
  515. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_task_id 错误!"}
  516. c.ServeJSON()
  517. return
  518. }
  519. is = Task.Del_TaskData_t_id(Task_r.T_task_id, strconv.Itoa(Id))
  520. if !is {
  521. c.Data["json"] = lib.JSONS{Code: 202, Msg: "删除失败!"}
  522. c.ServeJSON()
  523. return
  524. }
  525. System.Add_UserLogs(user_r.T_uuid, "任务数据", "删除"+Task_r.T_name, Task_r.T_task_id+"|"+strconv.Itoa(Id))
  526. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!"}
  527. c.ServeJSON()
  528. return
  529. }
  530. // 列表 - 接口
  531. func (c *TaskDataController) Export_Data_Excel() {
  532. // 验证登录 User_is, User_r
  533. _, User_is := Account.Verification_Admin(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  534. if !User_is {
  535. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  536. c.ServeJSON()
  537. return
  538. }
  539. Time_start := c.GetString("Time_start")
  540. Time_end := c.GetString("Time_end")
  541. T_sn_str := c.GetString("T_sn_list") //865901058809339,865901058815849,865901058818991,865901058810568
  542. //T_id, err := c.GetInt("T_id")
  543. //if err != nil {
  544. // T_id = -1
  545. //
  546. //}
  547. T_task_id := c.GetString("T_task_id")
  548. Task_r, is := Task.Read_Task(T_task_id)
  549. if !is {
  550. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_task_id 错误!"}
  551. c.ServeJSON()
  552. return
  553. }
  554. T_sn_list := strings.Split(T_sn_str, ",")
  555. DeviceSensor_data_list := []Task.TaskData_{}
  556. for _, v := range T_sn_list {
  557. DeviceSensor_data, _ := Task.Read_TaskData_ById_List(Task_r.T_task_id, v, "", Time_start, Time_end, 1, 9999)
  558. DeviceSensor_data_list = append(DeviceSensor_data_list, DeviceSensor_data...)
  559. }
  560. f := excelize.NewFile() // 设置单元格的值
  561. // 这里设置表头
  562. f.SetCellValue("Sheet1", "A1", "编号")
  563. f.SetCellValue("Sheet1", "B1", "SN")
  564. f.SetCellValue("Sheet1", "C1", "温度℃")
  565. f.SetCellValue("Sheet1", "D1", "湿度%")
  566. f.SetCellValue("Sheet1", "E1", "记录时间")
  567. // 设置列宽
  568. f.SetColWidth("Sheet1", "A", "A", 7)
  569. f.SetColWidth("Sheet1", "B", "B", 20)
  570. f.SetColWidth("Sheet1", "C", "C", 10)
  571. f.SetColWidth("Sheet1", "D", "D", 10)
  572. f.SetColWidth("Sheet1", "E", "E", 22)
  573. line := 1
  574. // 循环写入数据
  575. for _, v := range DeviceSensor_data_list {
  576. line++
  577. f.SetCellValue("Sheet1", fmt.Sprintf("A%d", line), v.T_id)
  578. f.SetCellValue("Sheet1", fmt.Sprintf("B%d", line), v.T_sn)
  579. f.SetCellValue("Sheet1", fmt.Sprintf("C%d", line), strconv.FormatFloat(float64(v.T_t), 'f', 1, 64))
  580. f.SetCellValue("Sheet1", fmt.Sprintf("D%d", line), strconv.FormatFloat(float64(v.T_rh), 'f', 1, 64))
  581. f.SetCellValue("Sheet1", fmt.Sprintf("E%d", line), v.T_time)
  582. }
  583. logs.Println("DeviceSensor_data:", len(DeviceSensor_data_list))
  584. lib.Create_Dir("./ofile")
  585. timeStr := time.Now().Format("20060102150405")
  586. // 保存文件
  587. if err := f.SaveAs("ofile/" + timeStr + ".xlsx"); err != nil {
  588. logs.Error(lib.FuncName(), err)
  589. }
  590. if !lib.Pload_qiniu("ofile/"+timeStr+".xlsx", "ofile/"+timeStr+".xlsx") {
  591. c.Data["json"] = lib.JSONS{Code: 203, Msg: "oss!"}
  592. c.ServeJSON()
  593. return
  594. }
  595. //删除目录
  596. //err = os.Remove("ofile/" + timeStr + ".xlsx")
  597. //if err != nil {
  598. // logs.Error(lib.FuncName(),err)
  599. //}
  600. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: "https://bzdcoldverifyoss.baozhida.cn/" + "ofile/" + timeStr + ".xlsx"}
  601. c.ServeJSON()
  602. return
  603. }
  604. // 列表 - 接口
  605. func (c *TaskDataController) Export_Data_PDF() {
  606. // 验证登录 User_is, User_r
  607. _, User_is := Account.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  608. if !User_is {
  609. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  610. c.ServeJSON()
  611. return
  612. }
  613. Time_start := c.GetString("Time_start")
  614. Time_end := c.GetString("Time_end")
  615. T_sn_str := c.GetString("T_sn_list") //865901058809339,865901058815849,865901058818991,865901058810568
  616. //T_id, err := c.GetInt("T_id")
  617. //if err != nil {
  618. // T_id = -1
  619. //
  620. //}
  621. T_task_id := c.GetString("T_task_id")
  622. Task_r, is := Task.Read_Task(T_task_id)
  623. if !is {
  624. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_task_id 错误!"}
  625. c.ServeJSON()
  626. return
  627. }
  628. dcList, _ := Device.Read_DeviceClassList_OrderList(Task_r.T_class, "", "", "", 0, 9999)
  629. // 查询设备列表
  630. // 保存布局编号和校准证书对应关系
  631. var deviceCertificateMap = make(map[string]string) // t_id, T_Certificate_sn
  632. for _, v := range dcList {
  633. deviceCertificateMap[v.T_id] = v.T_Certificate_sn
  634. }
  635. T_sn_list := strings.Split(T_sn_str, ",")
  636. DeviceSensor_data_list := []Task.TaskData_{}
  637. for _, v := range T_sn_list {
  638. DeviceSensor_data, _ := Task.Read_TaskData_ById_List(Task_r.T_task_id, v, "", Time_start, Time_end, 1, 9999)
  639. DeviceSensor_data_list = append(DeviceSensor_data_list, DeviceSensor_data...)
  640. for i := 0; i < len(DeviceSensor_data_list); i++ {
  641. DeviceSensor_data_list[i].T_Certificate_sn = deviceCertificateMap[DeviceSensor_data_list[i].T_id]
  642. }
  643. }
  644. //------
  645. var err error
  646. pdf := &gopdf.GoPdf{}
  647. pdf.Start(gopdf.Config{PageSize: gopdf.Rect{W: 595.28, H: 841.89}}) //595.28, 841.89 = A4
  648. //err = GetFont(pdf, "LiberationSerif-Regular.ttf")
  649. //if err != nil {
  650. // log.Fatalln(err)
  651. //}
  652. //err = pdf.SetFont("Ubuntu-L", "", 14)
  653. //if err != nil {
  654. // log.Fatalln(err)
  655. //}
  656. err = pdf.AddTTFFont("simsun", "static/fonts/三极行楷简体-粗.ttf")
  657. if err != nil {
  658. c.Data["json"] = lib.JSONS{Code: 204, Msg: "ok!", Data: err}
  659. c.ServeJSON()
  660. return
  661. }
  662. err = pdf.SetFont("simsun", "", 24)
  663. if err != nil {
  664. c.Data["json"] = lib.JSONS{Code: 205, Msg: "ok!", Data: err}
  665. c.ServeJSON()
  666. return
  667. }
  668. pdf.SetGrayFill(0.5)
  669. pdf.SetMargins(0, 20, 0, 20)
  670. pdf.AddPage()
  671. //use path
  672. //pdf.Image("logo.png", 100, 50, &gopdf.Rect{W: 50, H: 50})
  673. textw, _ := pdf.MeasureTextWidth(Task_r.T_name)
  674. pdf.SetX((595 / 2) - (textw / 2))
  675. pdf.SetY(40)
  676. pdf.Text(Task_r.T_name)
  677. // 线
  678. pdf.SetLineWidth(2)
  679. pdf.SetLineType("dashed")
  680. pdf.Line(10, 60, 585, 60)
  681. err = pdf.AddTTFFont("wts", "static/fonts/MiSans-Medium.ttf")
  682. if err != nil {
  683. c.Data["json"] = lib.JSONS{Code: 206, Msg: "ok!", Data: err}
  684. c.ServeJSON()
  685. return
  686. }
  687. err = pdf.SetFont("wts", "", 12)
  688. if err != nil {
  689. c.Data["json"] = lib.JSONS{Code: 207, Msg: "ok!", Data: err}
  690. c.ServeJSON()
  691. return
  692. }
  693. t_ := Time_start + " / " + Time_end
  694. if len(Time_start) == 0 {
  695. t_ = "所有数据"
  696. }
  697. //fmt.Sprintf(" %.1f ", v.T_t)
  698. lib.RectFillColor(pdf, "提取数据时间段["+t_+"]", 14, 22, 80, 550, 40, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  699. lib.RectFillColor(pdf, "布局编号", 12, 22, 120, 80, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  700. lib.RectFillColor(pdf, "证书编号", 12, 92, 120, 150, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  701. lib.RectFillColor(pdf, "温度℃", 12, 242, 120, 100, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  702. lib.RectFillColor(pdf, "湿度%", 12, 342, 120, 100, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  703. //lib.RectFillColor(pdf, "温度范围", 12, 272, 120, 100, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  704. //lib.RectFillColor(pdf, "湿度范围", 12, 362, 120, 100, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  705. lib.RectFillColor(pdf, "记录时间", 12, 442, 120, 130, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  706. var y float64 = 140
  707. err = pdf.SetFont("wts", "", 10)
  708. if err != nil {
  709. c.Data["json"] = lib.JSONS{Code: 207, Msg: "ok!", Data: err}
  710. c.ServeJSON()
  711. return
  712. }
  713. for _, v := range DeviceSensor_data_list {
  714. //text := fmt.Sprintf(" %d ", i+1)
  715. var textH float64 = 25 // if text height is 25px.
  716. pdf.SetNewY(y, textH)
  717. y = pdf.GetY()
  718. //pdf.SetX(x) // must after pdf.SetNewY() called.
  719. //err = pdf.Text(text)
  720. //if err != nil {
  721. // log.Fatalln(err)
  722. //}
  723. T_t := fmt.Sprintf(" %.1f ", v.T_t)
  724. T_rh := fmt.Sprintf(" %.1f ", v.T_rh)
  725. T_time := fmt.Sprintf("%s", v.T_time)
  726. lib.RectFillColor(pdf, v.T_id, 10, 22, y, 80, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  727. lib.RectFillColor(pdf, v.T_Certificate_sn, 10, 92, y, 150, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  728. lib.RectFillColor(pdf, T_t, 10, 242, y, 100, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  729. lib.RectFillColor(pdf, T_rh, 10, 342, y, 100, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  730. lib.RectFillColor(pdf, T_time, 10, 442, y, 130, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  731. y += 20
  732. }
  733. timeStr := "ofile/" + time.Now().Format("20060102150405") + ".pdf"
  734. err = pdf.WritePdf(timeStr)
  735. if err != nil {
  736. c.Data["json"] = lib.JSONS{Code: 207, Msg: "ok!", Data: err}
  737. c.ServeJSON()
  738. return
  739. }
  740. if !lib.Pload_qiniu(timeStr, timeStr) {
  741. c.Data["json"] = lib.JSONS{Code: 203, Msg: "oss!"}
  742. c.ServeJSON()
  743. return
  744. }
  745. //删除目录
  746. err = os.Remove(timeStr)
  747. if err != nil {
  748. logs.Error(lib.FuncName(), err)
  749. }
  750. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: "https://bzdcoldverifyoss.baozhida.cn/" + timeStr}
  751. c.ServeJSON()
  752. return
  753. }
  754. // 列表 - 接口
  755. func (c *TaskDataController) Check() {
  756. // 验证登录 User_is, User_r
  757. _, User_is := Account.Verification_Admin(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  758. if !User_is {
  759. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  760. c.ServeJSON()
  761. return
  762. }
  763. T_task_id := c.GetString("T_task_id")
  764. Task_r, is := Task.Read_Task(T_task_id)
  765. if !is {
  766. c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_task_id 错误!"}
  767. c.ServeJSON()
  768. return
  769. }
  770. type R_JSONS struct {
  771. T_sn string // SN
  772. T_id string // 编号
  773. T_unm int // 数据量
  774. T_time_interval int64 // 时间间隔
  775. Time_start string // 开始
  776. Time_end string // 结束
  777. Result int // 200 OK, 201 可以补 , 202 不能补
  778. Result_str string // 提示内容
  779. Result_Time_start string // 2022-8-05 21:01
  780. Result_Time_defect int64 // 缺失时间间隔
  781. }
  782. var r_jsons []R_JSONS
  783. List := Task.Read_TaskData_ById_ClassList(Task_r.T_task_id)
  784. for _, v := range List {
  785. var r_json R_JSONS
  786. r_json.T_sn = v.T_sn
  787. r_json.Result = 200
  788. DeviceSensor_data := Task.Read_TaskData_ById_List_(Task_r.T_task_id, v.T_sn)
  789. r_json.T_unm = len(DeviceSensor_data)
  790. if r_json.T_unm > 2 {
  791. r_json.T_id = DeviceSensor_data[0].T_id
  792. r_json.Time_end = DeviceSensor_data[len(DeviceSensor_data)-1].T_time
  793. r_json.Time_start = DeviceSensor_data[0].T_time
  794. formatTime_a, _ := time.Parse("2006-1-2 15:04", DeviceSensor_data[0].T_time)
  795. formatTime_b, _ := time.Parse("2006-1-2 15:04", DeviceSensor_data[1].T_time)
  796. formatTime_x := formatTime_b.Unix() - formatTime_a.Unix()
  797. r_json.T_time_interval = formatTime_x
  798. println("formatTime_x:", v.T_id, v.T_sn, formatTime_x)
  799. if formatTime_x > 60*30 || formatTime_x < 60 {
  800. r_json.Result = 202
  801. r_json.Result_str = "数据异常(数据的第一个时间与第二个时间相差 " + strconv.FormatInt(formatTime_x, 10) + "秒),必须大于60秒,小于30分钟"
  802. r_jsons = append(r_jsons, r_json)
  803. continue
  804. }
  805. for data_i := 2; data_i < len(DeviceSensor_data); data_i++ {
  806. formatTime_a = formatTime_b
  807. formatTime_b, _ = time.Parse("2006-1-2 15:04", DeviceSensor_data[data_i].T_time)
  808. formatTime_ := formatTime_b.Unix() - formatTime_a.Unix()
  809. println("formatTime_x-:", formatTime_)
  810. if formatTime_ < formatTime_x {
  811. r_json.Result = 202
  812. r_json.Result_str = "开始时间:" + DeviceSensor_data[data_i-1].T_time + " 离下一条时间间隔:" + strconv.FormatInt(formatTime_, 10) + "秒,间隔小于时间间隔,不能自动补充"
  813. break
  814. }
  815. if formatTime_x != formatTime_ {
  816. if formatTime_ > 60*30 {
  817. r_json.Result = 202
  818. r_json.Result_str = "开始时间:" + DeviceSensor_data[data_i-1].T_time + " 离下一条时间间隔:" + strconv.FormatInt(formatTime_, 10) + "秒,相差时间大于30分钟,不能自动补充"
  819. break
  820. }
  821. r_json.Result = 201
  822. r_json.Result_str = "开始时间:" + DeviceSensor_data[data_i-1].T_time + " 离下一条时间间隔:" + strconv.FormatInt(formatTime_, 10) + "秒"
  823. r_json.Result_Time_start = DeviceSensor_data[data_i-1].T_time
  824. r_json.Result_Time_defect = formatTime_
  825. println(r_json.Result_str)
  826. break
  827. }
  828. }
  829. } else {
  830. r_json.Result = 202
  831. r_json.Result_str = "数据量太少 必须大于 2条以上!"
  832. }
  833. r_jsons = append(r_jsons, r_json)
  834. }
  835. c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
  836. c.ServeJSON()
  837. return
  838. }
  839. func (c *TaskDataController) TaskData_Temperature_Pdf() {
  840. // 验证登录管理员 User_is, User_r
  841. _, User_Admin_is := Account.Verification_Admin(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  842. // 验证登录用户 User_is, User_r
  843. _, User_is := Account.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  844. if !User_Admin_is && !User_is {
  845. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  846. c.ServeJSON()
  847. return
  848. }
  849. Time_start := c.GetString("Time_start")
  850. Time_end := c.GetString("Time_end")
  851. T_sn := c.GetString("T_sn")
  852. T_id := c.GetString("T_id")
  853. T_task_id := c.GetString("T_task_id")
  854. err := c.TaskData_Pdf(Time_start, Time_end, T_task_id, T_sn, T_id, Task.Temperature)
  855. if err != nil {
  856. c.Data["json"] = lib.JSONS{Code: 202, Msg: err.Error()}
  857. c.ServeJSON()
  858. return
  859. }
  860. }
  861. func (c *TaskDataController) TaskData_Humidity_Pdf() {
  862. // 验证登录管理员 User_is, User_r
  863. _, User_Admin_is := Account.Verification_Admin(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  864. // 验证登录用户 User_is, User_r
  865. _, User_is := Account.Verification(c.Ctx.GetCookie("User_tokey"), c.GetString("User_tokey"))
  866. if !User_Admin_is && !User_is {
  867. c.Data["json"] = lib.JSONS{Code: 201, Msg: "请重新登录!"}
  868. c.ServeJSON()
  869. return
  870. }
  871. Time_start := c.GetString("Time_start")
  872. Time_end := c.GetString("Time_end")
  873. T_sn := c.GetString("T_sn")
  874. T_id := c.GetString("T_id")
  875. T_task_id := c.GetString("T_task_id")
  876. err := c.TaskData_Pdf(Time_start, Time_end, T_task_id, T_sn, T_id, Task.Humidity)
  877. if err != nil {
  878. c.Data["json"] = lib.JSONS{Code: 202, Msg: err.Error()}
  879. c.ServeJSON()
  880. return
  881. }
  882. }
  883. func (c *TaskDataController) TaskData_Pdf(Time_start, Time_end, T_task_id, T_sn, T_id, T_type string) (err error) {
  884. Task_r, is := Task.Read_Task(T_task_id)
  885. if !is {
  886. return errors.New("T_task_id 错误!")
  887. }
  888. user, is := Account.Read_User(Task_r.T_uuid)
  889. if !is {
  890. return errors.New("Task uuid 错误!")
  891. }
  892. if Task_r.T_delivery_state == 2 {
  893. return errors.New("数据采集中,请稍后!")
  894. }
  895. pdf := &gopdf.GoPdf{}
  896. pdf.Start(gopdf.Config{PageSize: gopdf.Rect{W: 595.28, H: 841.89}}) //595.28, 841.89 = A4
  897. err = pdf.AddTTFFont("wts", "static/fonts/MiSans-Medium.ttf")
  898. if err != nil {
  899. return
  900. }
  901. err = pdf.SetFont("wts", "", 15)
  902. if err != nil {
  903. return
  904. }
  905. pdf.SetGrayFill(0.5)
  906. pdf.SetMargins(0, 20, 0, 20)
  907. pdf.AddPage()
  908. imgH, _ := gopdf.ImageHolderByPath("./static/logo.jpg")
  909. err = pdf.ImageByHolder(imgH, 10, 10, &gopdf.Rect{W: 93, H: 32})
  910. name := user.T_name
  911. var y float64 = 40
  912. textw, _ := pdf.MeasureTextWidth(name)
  913. pdf.SetX((595 / 2) - (textw / 2))
  914. pdf.SetY(y)
  915. pdf.Text(name)
  916. y += 20
  917. T_type_name := ""
  918. if T_type == Task.Temperature {
  919. T_type_name = "温度"
  920. }
  921. if T_type == Task.Humidity {
  922. T_type_name = "湿度"
  923. }
  924. title := Task_r.T_name + T_type_name + "验证数据"
  925. textw, _ = pdf.MeasureTextWidth(title)
  926. pdf.SetX((595 / 2) - (textw / 2))
  927. pdf.SetY(y)
  928. pdf.Text(title)
  929. // 查询设备列表
  930. //dcList, _ := Device.Read_DeviceClassList_OrderList(Task_r.T_class, T_sn, T_id, "", 0, 9999)
  931. dcList := []Device.DeviceClassList{}
  932. dataList, _ := Task.Read_TaskData_ById_List(Task_r.T_task_id, T_sn, T_id, Time_start, Time_end, 0, 9999)
  933. dataListMap := make(map[string][]Task.TaskData_)
  934. dcListMap := make(map[string]string)
  935. dataMap := make(map[string]string)
  936. timeMap := make(map[string]bool)
  937. for _, data := range dataList {
  938. dataListMap[data.T_id] = append(dataListMap[data.T_id], data)
  939. if _, ok := dcListMap[data.T_id]; !ok {
  940. dcListMap[data.T_id] = data.T_sn
  941. }
  942. k := fmt.Sprintf("%s,%s", data.T_time, data.T_id)
  943. if T_type == Task.Temperature {
  944. dataMap[k] = fmt.Sprintf("%.1f", data.T_t)
  945. }
  946. if T_type == Task.Humidity {
  947. dataMap[k] = fmt.Sprintf("%.1f", data.T_rh)
  948. }
  949. if _, ok := timeMap[data.T_time]; !ok {
  950. timeMap[data.T_time] = true
  951. }
  952. }
  953. for id, sn := range dcListMap {
  954. dcList = append(dcList, Device.DeviceClassList{T_id: id, T_sn: sn})
  955. }
  956. for id, list := range dataListMap {
  957. sort.Slice(list, func(i, j int) bool {
  958. return list[i].T_time < list[j].T_time
  959. })
  960. dataListMap[id] = list
  961. }
  962. var timeList []string
  963. for k, _ := range timeMap {
  964. timeList = append(timeList, k)
  965. }
  966. sort.Slice(timeList, func(i, j int) bool {
  967. return timeList[i] < timeList[j]
  968. })
  969. err = pdf.SetFont("wts", "", 12)
  970. if err != nil {
  971. return
  972. }
  973. if len(timeList) > 0 {
  974. y += 20
  975. timeStr := fmt.Sprintf("%s - %s", timeList[0], timeList[len(timeList)-1])
  976. textw, _ = pdf.MeasureTextWidth(timeStr)
  977. pdf.SetX((595 / 2) - (textw / 2))
  978. pdf.SetY(y)
  979. pdf.Text(timeStr)
  980. }
  981. //y += 20
  982. var x float64 = 10
  983. var w float64 = 120
  984. err = pdf.SetFont("wts", "", 10)
  985. if err != nil {
  986. return
  987. }
  988. idWidthMap := make(map[string]float64)
  989. dcList1 := make([]Device.DeviceClassList, 0)
  990. for _, list := range dcList {
  991. idw, _ := pdf.MeasureTextWidth(list.T_id)
  992. if !lib.IsNumeric(list.T_id) {
  993. dcList1 = append(dcList1, list)
  994. }
  995. if idw > 20 {
  996. idWidthMap[list.T_id] = idw + 12
  997. } else {
  998. idWidthMap[list.T_id] = 30.3
  999. }
  1000. }
  1001. sort.Slice(dcList1, func(i, j int) bool {
  1002. return dcList1[i].T_id < dcList1[j].T_id
  1003. })
  1004. dcList2 := listSubtract(dcList, dcList1)
  1005. // 获取探头时间
  1006. var timeList2 []string
  1007. timeMap2 := make(map[string]struct{})
  1008. for _, list := range dcList2 {
  1009. dataList2, ok := dataListMap[list.T_id]
  1010. if ok {
  1011. for _, data := range dataList2 {
  1012. if _, ok = timeMap2[data.T_time]; !ok {
  1013. timeMap2[data.T_time] = struct{}{}
  1014. }
  1015. }
  1016. }
  1017. }
  1018. for k, _ := range timeMap2 {
  1019. timeList2 = append(timeList2, k)
  1020. }
  1021. sort.Slice(timeList2, func(i, j int) bool {
  1022. return timeList2[i] < timeList2[j]
  1023. })
  1024. var temp int
  1025. for index, dc := range dcList1 {
  1026. dataList2, _ := dataListMap[dc.T_id]
  1027. err = pdf.SetFont("wts", "", 15)
  1028. if err != nil {
  1029. return
  1030. }
  1031. y += 30
  1032. var textH float64 = 20 // if text height is 25px.
  1033. if y >= 790 {
  1034. addStampImage(pdf, 455, y-150)
  1035. // AddWatermark(Task_r, pdf)
  1036. pdf.AddPage()
  1037. y = 30
  1038. } else {
  1039. if index > 0 {
  1040. y += 10
  1041. }
  1042. }
  1043. textw, _ = pdf.MeasureTextWidth(dc.T_id)
  1044. pdf.SetX((595 / 2) - (textw / 2))
  1045. pdf.SetY(y)
  1046. pdf.Text(dc.T_id)
  1047. y += 10
  1048. err = pdf.SetFont("wts", "", 10)
  1049. if err != nil {
  1050. return
  1051. }
  1052. if y < 790 {
  1053. x = 10
  1054. for i := 0; i < 4; i++ {
  1055. w = 113.8
  1056. lib.RectFillColor(pdf, "时间", 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1057. x += w
  1058. w = 30
  1059. lib.RectFillColor(pdf, T_type_name, 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1060. x += w
  1061. }
  1062. y += 20
  1063. } else {
  1064. addStampImage(pdf, 455, y-150)
  1065. // AddWatermark(Task_r, pdf)
  1066. pdf.AddPage()
  1067. y = 20
  1068. x = 10
  1069. for i := 0; i < 4; i++ {
  1070. w = 113.8
  1071. lib.RectFillColor(pdf, "时间", 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1072. x += w
  1073. w = 30
  1074. lib.RectFillColor(pdf, T_type_name, 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1075. x += w
  1076. }
  1077. y += 20
  1078. }
  1079. for i := 0; i < len(dataList2); i++ {
  1080. if y > 790 {
  1081. addStampImage(pdf, 455, y-130)
  1082. // AddWatermark(Task_r, pdf)
  1083. pdf.AddPage()
  1084. y = pdf.GetY()
  1085. x = 10
  1086. for k := 0; k < 4; k++ {
  1087. w = 113.8
  1088. lib.RectFillColor(pdf, "时间", 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1089. x += w
  1090. w = 30
  1091. lib.RectFillColor(pdf, T_type_name, 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1092. x += w
  1093. }
  1094. y += 20
  1095. }
  1096. temp = i % 4
  1097. // 每页添加表头
  1098. if y == 20 && temp == 0 {
  1099. x = 10
  1100. for k := 0; k < 4; k++ {
  1101. w = 113.8
  1102. lib.RectFillColor(pdf, "时间", 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1103. x += w
  1104. w = 30
  1105. lib.RectFillColor(pdf, T_type_name, 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1106. x += w
  1107. }
  1108. y += 20
  1109. }
  1110. x = 10 + 143.8*float64(temp)
  1111. w = 113.8
  1112. lib.RectFillColor(pdf, dataList2[i].T_time, 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1113. x += w
  1114. w = 30
  1115. if T_type == Task.Temperature {
  1116. lib.RectFillColor(pdf, fmt.Sprintf("%.1f", dataList2[i].T_t), 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1117. }
  1118. if T_type == Task.Humidity {
  1119. lib.RectFillColor(pdf, fmt.Sprintf("%.1f", dataList2[i].T_rh), 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1120. }
  1121. //if y > 760 {
  1122. // fmt.Println("=====y:", y)
  1123. //}
  1124. //if y > 790 {
  1125. // addStampImage(pdf, 455, y-130)
  1126. //}
  1127. if temp == 3 {
  1128. y += 20
  1129. pdf.SetNewY(y, textH)
  1130. y = pdf.GetY()
  1131. }
  1132. }
  1133. }
  1134. if len(dcList2) > 0 {
  1135. if temp != 3 {
  1136. y += 20
  1137. }
  1138. y += 40
  1139. }
  1140. sort.Slice(dcList2, func(i, j int) bool {
  1141. return dcList2[i].T_id < dcList2[j].T_id
  1142. })
  1143. chunks := splitData(dcList2, 454.5, idWidthMap)
  1144. for i, list := range chunks {
  1145. if i > 0 {
  1146. y += 40
  1147. }
  1148. if y < 790 {
  1149. x = 10
  1150. w = 120.7
  1151. lib.RectFillColor(pdf, "时间", 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1152. x += w
  1153. w = 30.3
  1154. for _, v2 := range list {
  1155. w = idWidthMap[v2.T_id]
  1156. lib.RectFillColor(pdf, v2.T_id, 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1157. x += w
  1158. }
  1159. y += 20
  1160. } else {
  1161. addStampImage(pdf, 455, y-150)
  1162. // AddWatermark(Task_r, pdf)
  1163. pdf.AddPage()
  1164. y = 20
  1165. }
  1166. StampImageX := x
  1167. for j, t := range timeList2 {
  1168. if j > 0 {
  1169. y += 20
  1170. }
  1171. if y >= 790 {
  1172. addStampImage(pdf, StampImageX-130, y-130)
  1173. // AddWatermark(Task_r, pdf)
  1174. pdf.AddPage()
  1175. y = pdf.GetY()
  1176. x = 10
  1177. w = 120.7
  1178. lib.RectFillColor(pdf, "时间", 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1179. x += w
  1180. w = 30.3
  1181. for _, v2 := range list {
  1182. w = idWidthMap[v2.T_id]
  1183. lib.RectFillColor(pdf, v2.T_id, 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1184. x += w
  1185. }
  1186. y += 20
  1187. x = 10
  1188. }
  1189. x = 10
  1190. if y == 20 {
  1191. w = 120.7
  1192. lib.RectFillColor(pdf, "时间", 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1193. x += w
  1194. w = 30.3
  1195. for _, v2 := range list {
  1196. w = idWidthMap[v2.T_id]
  1197. lib.RectFillColor(pdf, v2.T_id, 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1198. x += w
  1199. }
  1200. y += 20
  1201. x = 10
  1202. }
  1203. w = 120.7
  1204. lib.RectFillColor(pdf, t, 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1205. x += w
  1206. w = 30.3
  1207. for _, v := range list {
  1208. t_t := dataMap[fmt.Sprintf("%s,%s", t, v.T_id)]
  1209. w = idWidthMap[v.T_id]
  1210. lib.RectFillColor(pdf, t_t, 12, x, y, w, 20, 255, 255, 255, lib.AlignCenter, lib.ValignMiddle)
  1211. x += w
  1212. }
  1213. StampImageX = x
  1214. }
  1215. }
  1216. if y <= 790 {
  1217. if y < 150 {
  1218. y = 20
  1219. } else {
  1220. y = y - 145
  1221. }
  1222. x = x - 135
  1223. if len(dcList2) == 0 {
  1224. x = 455
  1225. }
  1226. addStampImage(pdf, x, y)
  1227. // AddWatermark(Task_r, pdf)
  1228. } else if y > 790 && len(dcList2) == 0 {
  1229. if y < 150 {
  1230. y = 20
  1231. } else {
  1232. y = y - 145
  1233. }
  1234. addStampImage(pdf, 455, y)
  1235. }
  1236. filename := time.Now().Format("20060102150405") + ".pdf"
  1237. timeStr := "ofile/" + T_type_name + filename
  1238. err = pdf.WritePdf(timeStr)
  1239. if err != nil {
  1240. c.Data["json"] = lib.JSONS{Code: 202, Msg: err.Error()}
  1241. c.ServeJSON()
  1242. return
  1243. }
  1244. // 上传 OSS
  1245. //url, is := NatsServer.Qiniu_UploadFile(lib.GetCurrentDirectory()+"/ofile/"+timeStr, timeStr)
  1246. //url, is := NatsServer.Qiniu_UploadFile("/Users/zoie/work/bzd_project/ColdVerify_server/"+timeStr, timeStr)
  1247. //if !is {
  1248. // err = errors.New("oss!")
  1249. // return
  1250. //}
  1251. defer func() {
  1252. //删除目录
  1253. os.Remove(timeStr)
  1254. }()
  1255. c.Ctx.Output.Download(timeStr)
  1256. //c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: url}
  1257. //c.ServeJSON()
  1258. //return
  1259. return nil
  1260. }
  1261. // 添加公章图片
  1262. func addStampImage(pdf *gopdf.GoPdf, x, y float64) {
  1263. imagePath := "./static/commonSeal.jpg"
  1264. if x < 10 {
  1265. x = 10
  1266. }
  1267. if y < 20 {
  1268. y = 20
  1269. }
  1270. err := pdf.Image(imagePath, x, y, &gopdf.Rect{W: 123, H: 128.6})
  1271. if err != nil {
  1272. log.Print(err.Error())
  1273. }
  1274. }
  1275. func listSubtract(sliceA, sliceB []Device.DeviceClassList) []Device.DeviceClassList {
  1276. // 创建一个集合用于存储 sliceB 的元素
  1277. elementsToRemove := make(map[string]Device.DeviceClassList)
  1278. for _, item := range sliceB {
  1279. elementsToRemove[item.T_id] = item
  1280. }
  1281. // 构建新切片,包含 sliceA 中不在 elementsToRemove 中的元素
  1282. var result []Device.DeviceClassList
  1283. for _, item := range sliceA {
  1284. if _, found := elementsToRemove[item.T_id]; !found {
  1285. result = append(result, item)
  1286. }
  1287. }
  1288. return result
  1289. }
  1290. func splitData(data []Device.DeviceClassList, threshold float64, idWidthMap map[string]float64) [][]Device.DeviceClassList {
  1291. var result [][]Device.DeviceClassList
  1292. var currentBatch []Device.DeviceClassList
  1293. var currentSum float64
  1294. for _, item := range data {
  1295. wd := idWidthMap[item.T_id]
  1296. if currentSum+wd > threshold+0.1 {
  1297. // 当前批次超过阈值,切分
  1298. result = append(result, currentBatch)
  1299. // 重置当前批次和当前总和
  1300. currentBatch = []Device.DeviceClassList{}
  1301. currentSum = 0
  1302. }
  1303. currentBatch = append(currentBatch, item)
  1304. currentSum += wd
  1305. }
  1306. // 添加最后一批
  1307. if len(currentBatch) > 0 {
  1308. result = append(result, currentBatch)
  1309. }
  1310. return result
  1311. }