TaskData.go 31 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030
  1. package Task
  2. import (
  3. "ColdVerify_local/conf"
  4. "ColdVerify_local/lib"
  5. "ColdVerify_local/logs"
  6. "encoding/json"
  7. "fmt"
  8. "github.com/astaxie/beego/cache"
  9. _ "github.com/astaxie/beego/cache/redis"
  10. "github.com/beego/beego/v2/adapter/orm"
  11. orm2 "github.com/beego/beego/v2/client/orm"
  12. _ "github.com/go-sql-driver/mysql"
  13. "strconv"
  14. "strings"
  15. "time"
  16. )
  17. // 模板
  18. type TaskData struct {
  19. Id int `orm:"column(ID);size(11);auto;pk"`
  20. T_id string `orm:"size(256);null"` // 标题
  21. T_t float32 `orm:"size(10);null"` // 温度
  22. T_rh float32 `orm:"size(10);null"` // 湿度
  23. T_time time.Time `orm:"type(timestamp);null;"` // 采集时间
  24. CreateTime time.Time `orm:"column(create_time);type(timestamp);null;auto_now_add"` //auto_now_add 第一次保存时才设置时间 UpdateTime time.Time `orm:"column(update_time);type(timestamp);null;auto_now"` //auto_now 每次 model 保存时都会对时间自动更新
  25. }
  26. func (t *TaskData) TableName() string {
  27. return "task_data" // 数据库名称 // ************** 替换 FormulaList **************
  28. }
  29. var redisCache_TaskData cache.Cache
  30. func init() {
  31. //orm2.Debug = true
  32. config := fmt.Sprintf(`{"key":"%s","conn":"%s","dbNum":"%s","password":"%s"}`,
  33. "redis_"+"TaskData", conf.Redis_address, conf.Redis_dbNum, conf.Redis_password)
  34. logs.Println(config)
  35. var err error
  36. redisCache_TaskData, err = cache.NewCache("redis", config)
  37. if err != nil || redisCache_TaskData == nil {
  38. errMsg := "failed to init redis"
  39. logs.Println(errMsg, err)
  40. }
  41. }
  42. type TaskDataJPG struct {
  43. State int `json:"state"` //1:生成中 2:已完成 3:失败
  44. Msg string `json:"msg"`
  45. Url string `json:"url"` //url
  46. }
  47. // ---------------- Redis -------------------
  48. // Redis_Set(m.T_sn,m) // Redis 更新缓存
  49. func Redis_TaskDataJPG_Set(key string, r TaskDataJPG) (err error) {
  50. //json序列化
  51. str, err := json.Marshal(r)
  52. if err != nil {
  53. logs.Error(lib.FuncName(), err)
  54. return
  55. }
  56. err = redisCache_TaskData.Put(key, str, 5*time.Minute)
  57. if err != nil {
  58. logs.Println("set key:", key, ",value:", str, err)
  59. }
  60. return
  61. }
  62. // if r,is :=Redis_Get(T_sn);is{
  63. // return r,nil
  64. // }
  65. func Redis_TaskDataJPG_Get(key string) (r TaskDataJPG, is bool) {
  66. if redisCache_TaskData.IsExist(key) {
  67. logs.Println("找到key:", key)
  68. v := redisCache_TaskData.Get(key)
  69. json.Unmarshal(v.([]byte), &r)
  70. return r, true
  71. }
  72. logs.Println("没有 找到key:", key)
  73. return TaskDataJPG{}, false
  74. }
  75. func Redis_TaskDataJPG_Del(key string) (err error) {
  76. err = redisCache_TaskData.Delete(key)
  77. return
  78. }
  79. // 创建数据库 Device.CREATE_TaskData("")
  80. func CREATE_TaskData(alias_name, T_task_id string) bool {
  81. o := orm2.NewOrmUsingDB(alias_name)
  82. sql := "DROP TABLE IF EXISTS `z_task_data_" + T_task_id + "`"
  83. o.Raw(sql).Exec()
  84. sql = "CREATE TABLE IF NOT EXISTS `z_task_data_" + T_task_id + "` ( " +
  85. " `ID` int(11) NOT NULL AUTO_INCREMENT," +
  86. " `t_sn` varchar(256) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL," +
  87. " `t_id` varchar(256) DEFAULT NULL," +
  88. " `t_t` float(6, 1) NULL DEFAULT NULL," +
  89. " `t_rh` float(6, 1) NULL DEFAULT NULL," +
  90. " `t_time` datetime(0) NULL DEFAULT NULL," +
  91. " PRIMARY KEY (`ID`) USING BTREE," +
  92. " KEY `t_sn` (`t_sn`)," +
  93. " KEY `t_id` (`t_id`)," +
  94. " KEY `t_time` (`t_time`)" +
  95. ") ENGINE = InnoDB AUTO_INCREMENT = 5 CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci ROW_FORMAT = Dynamic;"
  96. _, err := o.Raw(sql).Exec()
  97. if err != nil {
  98. return false
  99. }
  100. //sqlIndex := "ALTER TABLE `z_task_data_" + T_task_id + "` add unique index(`t_sn`,`t_id`,`t_time`);"
  101. //logs.Debug(sqlIndex)
  102. //o.Raw(sqlIndex).Exec()
  103. return true
  104. }
  105. // ---------------- 特殊方法 -------------------
  106. // 清空
  107. func Truncate_TaskData(alias_name, T_task_id string) bool {
  108. o := orm2.NewOrmUsingDB(alias_name)
  109. sql := "truncate table z_task_data_" + T_task_id
  110. logs.Println(sql)
  111. _, err := o.Raw(sql).Exec()
  112. if err != nil {
  113. logs.Error(lib.FuncName(), err)
  114. return false
  115. }
  116. return true
  117. }
  118. type TaskData_ struct {
  119. ID int `orm:"column(ID);size(100);null"` // ID
  120. T_sn string `orm:"column(t_sn);size(256);null"` // sn
  121. T_id string `orm:"column(t_id);size(256);null"` // 标题
  122. T_t float32 `orm:"column(t_t);size(10);null"` // 温度
  123. T_rh float32 `orm:"column(t_rh);size(10);null"` // 湿度
  124. T_time string `orm:"column(t_times);null;"` // 采集时间
  125. }
  126. type TaskDataClass_ struct {
  127. T_sn string `orm:"column(t_sn);size(256);null"` // 标题
  128. T_id string `orm:"column(t_id);size(256);null"` // 名称
  129. }
  130. func Read_TaskData_ById_List(T_task_id string, SN string, T_id string, Time_start_ string, Time_end_ string, page int, page_z int) ([]TaskData_, int64) {
  131. o := orm.NewOrm()
  132. var maps []TaskData_
  133. var maps_z []orm2.ParamsList
  134. pagez := page_z
  135. var offset int
  136. if page <= 1 {
  137. offset = 0
  138. } else {
  139. page -= 1
  140. offset = page * pagez
  141. }
  142. sql_condition := ""
  143. if len(Time_start_) > 1 {
  144. sql_condition += " AND t_time >= '" + Time_start_ + "'"
  145. }
  146. if len(Time_end_) > 1 {
  147. sql_condition += " AND t_time <= '" + Time_end_ + "'"
  148. }
  149. if len(T_id) > 0 {
  150. sql_condition += " AND t_id = '" + T_id + "'"
  151. }
  152. if len(SN) > 0 {
  153. sql_condition += " AND t_sn = '" + SN + "'"
  154. }
  155. if len(sql_condition) > 0 {
  156. sql_condition = " WHERE " + strings.TrimLeft(sql_condition, " AND ")
  157. }
  158. sql := "SELECT COUNT(ID) FROM z_task_data_" + T_task_id + sql_condition
  159. fmt.Println(sql)
  160. _, err := o.Raw(sql).ValuesList(&maps_z)
  161. if err != nil {
  162. return maps, 0
  163. }
  164. if len(maps_z) == 0 {
  165. return maps, 0
  166. }
  167. //fmt.Println("maps_z;",maps_z[0][0])
  168. sql = "SELECT ID,t_sn,t_id,t_t,t_rh,DATE_FORMAT(t_time,'%Y-%m-%d %H:%i:%s') AS t_times,t_time FROM z_task_data_" + T_task_id + sql_condition + " ORDER BY t_time DESC"
  169. if page_z != 9999 {
  170. sql = sql + " LIMIT " + strconv.Itoa(offset) + "," + strconv.Itoa(pagez)
  171. }
  172. fmt.Println(sql)
  173. _, err = o.Raw(sql).QueryRows(&maps)
  174. if err != nil {
  175. logs.Error(lib.FuncName(), err)
  176. }
  177. //value, _ := strconv.ParseFloat(fmt.Sprintf("%.2f", cnt), 64)
  178. key, _ := strconv.Atoi(maps_z[0][0].(string))
  179. return maps, int64(key)
  180. }
  181. func Read_TaskData_ById_List_AES(T_task_id string, SN string, T_id string, Time_start_ string, Time_end_ string, page int, page_z int) ([]TaskData_, int64) {
  182. o := orm.NewOrm()
  183. var maps []TaskData_
  184. var maps_z []orm2.ParamsList
  185. pagez := page_z
  186. var offset int
  187. if page <= 1 {
  188. offset = 0
  189. } else {
  190. page -= 1
  191. offset = page * pagez
  192. }
  193. sql_condition := ""
  194. if len(Time_start_) > 1 {
  195. sql_condition += " AND t_time >= '" + Time_start_ + "'"
  196. }
  197. if len(Time_end_) > 1 {
  198. sql_condition += " AND t_time <= '" + Time_end_ + "'"
  199. }
  200. if len(T_id) > 0 {
  201. sql_condition += " AND t_id = '" + T_id + "'"
  202. }
  203. if len(SN) > 0 {
  204. sql_condition += " AND t_sn = '" + SN + "'"
  205. }
  206. if len(sql_condition) > 0 {
  207. sql_condition = " WHERE " + strings.TrimLeft(sql_condition, " AND ")
  208. }
  209. sql := "SELECT COUNT(ID) FROM z_task_data_" + T_task_id + sql_condition
  210. fmt.Println(sql)
  211. _, err := o.Raw(sql).ValuesList(&maps_z)
  212. if err != nil {
  213. return maps, 0
  214. }
  215. if len(maps_z) == 0 {
  216. return maps, 0
  217. }
  218. //fmt.Println("maps_z;",maps_z[0][0])
  219. sql = "SELECT ID,t_sn,t_id,t_t,t_rh,DATE_FORMAT(t_time,'%Y-%m-%d %H:%i:%s') AS t_times,t_time FROM z_task_data_" + T_task_id + sql_condition + " ORDER BY t_time"
  220. if page_z != 9999 {
  221. sql = sql + " LIMIT " + strconv.Itoa(offset) + "," + strconv.Itoa(pagez)
  222. }
  223. fmt.Println(sql)
  224. _, err = o.Raw(sql).QueryRows(&maps)
  225. if err != nil {
  226. logs.Error(lib.FuncName(), err)
  227. }
  228. //value, _ := strconv.ParseFloat(fmt.Sprintf("%.2f", cnt), 64)
  229. key, _ := strconv.Atoi(maps_z[0][0].(string))
  230. return maps, int64(key)
  231. }
  232. func Read_TaskData_ById_List_(T_task_id string, SN string) []TaskData_ {
  233. o := orm.NewOrm()
  234. var maps []TaskData_
  235. sql_condition := ""
  236. sql_condition += " t_sn ='" + SN + "'"
  237. //fmt.Println("maps_z;",maps_z[0][0])
  238. sql := "SELECT ID,t_sn,t_id,t_t,t_rh,DATE_FORMAT(t_time,'%Y-%c-%d %H:%i') AS t_times,t_time FROM z_task_data_" + T_task_id + " WHERE " + sql_condition + " ORDER BY t_time "
  239. fmt.Println(sql)
  240. _, err := o.Raw(sql).QueryRows(&maps)
  241. if err != nil {
  242. logs.Error(lib.FuncName(), err)
  243. return maps
  244. }
  245. return maps
  246. }
  247. func Read_TaskData_ById_ClassList(T_task_id string) []TaskDataClass_ {
  248. o := orm.NewOrm()
  249. var maps []TaskDataClass_
  250. //fmt.Println("maps_z;",maps_z[0][0])
  251. //sql := "SELECT DISTINCT t_sn,t_id FROM z_task_data_" + T_task_id + " ORDER BY t_id "
  252. //sql := "SELECT DISTINCT t_sn FROM z_task_data_" + T_task_id + " ORDER BY t_id "
  253. sql := "SELECT t_sn,t_id FROM z_task_data_" + T_task_id + " GROUP BY t_sn,t_id ORDER BY t_id+0 "
  254. fmt.Println(sql)
  255. _, err := o.Raw(sql).QueryRows(&maps)
  256. if err != nil {
  257. logs.Error(lib.FuncName(), err)
  258. return maps
  259. }
  260. return maps
  261. }
  262. func Read_TaskData_sn(T_task_id string) []TaskDataClass_ {
  263. o := orm.NewOrm()
  264. var maps []TaskDataClass_
  265. sql := "SELECT DISTINCT t_sn FROM z_task_data_" + T_task_id + " ORDER BY t_sn "
  266. fmt.Println(sql)
  267. _, err := o.Raw(sql).QueryRows(&maps)
  268. if err != nil {
  269. logs.Error(lib.FuncName(), err)
  270. return maps
  271. }
  272. return maps
  273. }
  274. func Update_TaskData_ByT_sn(T_task_id, T_sn, T_id string) error {
  275. o := orm.NewOrm()
  276. // 修改id
  277. sql := "UPDATE z_task_data_" + T_task_id + " SET `t_id` = '" + T_id + "' WHERE `t_sn` = '" + T_sn + "'"
  278. logs.Println(sql)
  279. _, err := o.Raw(sql).Exec()
  280. if err != nil {
  281. logs.Error(lib.FuncName(), err)
  282. return err
  283. }
  284. return nil
  285. }
  286. func Delete_TaskData_ByT_sn(T_task_id, T_sn string) error {
  287. o := orm.NewOrm()
  288. // 修改id
  289. sql := "DELETE FROM z_task_data_" + T_task_id + " WHERE `t_sn` = '" + T_sn + "'"
  290. logs.Println(sql)
  291. _, err := o.Raw(sql).Exec()
  292. if err != nil {
  293. logs.Error(lib.FuncName(), err)
  294. return err
  295. }
  296. return nil
  297. }
  298. // 添加
  299. func Add_TaskData(T_task_id string, T_sn string, T_id string, T_t string, T_rh string, T_time string) bool {
  300. o := orm2.NewOrm()
  301. tx, err := o.Begin()
  302. if err != nil {
  303. logs.Error("start the transaction failed")
  304. return false
  305. }
  306. // 开始插入数据
  307. //
  308. //sql := "INSERT INTO z_task_data_" + T_task_id + " (`t_sn`, `t_id`, `t_time`) " +
  309. // "VALUES ('" + T_sn + "', " + T_id + ", '" + T_time + "') " +
  310. // "ON DUPLICATE KEY UPDATE t_t=" + T_t + ", t_rh="+ T_rh +";"
  311. // 去重复数据
  312. sql := "DELETE FROM z_task_data_" + T_task_id + " WHERE " + " t_id = '" + T_id + "' AND " + " t_sn = '" + T_sn + "' " + "AND t_time = '" + T_time + "' "
  313. // 这里有时间优化 用于一次 prepare 多次 exec,以提高批量执行的速度
  314. //fmt.Println(sql)
  315. res, err := tx.Raw(sql).Exec()
  316. if err != nil {
  317. tx.Rollback()
  318. logs.Error(lib.FuncName(), err)
  319. return false
  320. }
  321. res.RowsAffected()
  322. sql = "INSERT INTO z_task_data_" + T_task_id + " (`t_sn`, `t_id`, `t_t`, `t_rh`, `t_time`) " +
  323. "VALUES ('" + T_sn + "', '" + T_id + "', " + T_t + "," + T_rh + ", '" + T_time + "')"
  324. // 更新数据
  325. //sql := "INSERT INTO z_task_data_" + T_task_id + " (`t_sn`, `t_id`, `t_t`, `t_rh`, `t_time`) " +
  326. // "VALUES ('" + T_sn + "', " + T_id + ", " + T_t + "," + T_rh + ", '" + T_time + "')" +
  327. // "on duplicate key update `t_t`=" + T_t + ",`t_rh`=" + T_rh
  328. //fmt.Println(sql)
  329. res, err = tx.Raw(sql).Exec()
  330. if err != nil {
  331. tx.Rollback()
  332. logs.Error(lib.FuncName(), err)
  333. return false
  334. }
  335. tx.Commit()
  336. //fmt.Println("mysql row affected nums: ", num)
  337. return true
  338. }
  339. // 创建唯一索引
  340. func Create_Unique_Index(T_task_id string) error {
  341. // 创建索引
  342. o := orm.NewOrm()
  343. createIndexSql := "ALTER TABLE z_task_data_" + T_task_id + " ADD CONSTRAINT unique_index_t_sn_t_id_t_time UNIQUE (t_sn, t_id, t_time);"
  344. //fmt.Println(sql)
  345. _, err := o.Raw(createIndexSql).Exec()
  346. if err != nil && !strings.Contains(err.Error(), "Duplicate key name 'unique_index_t_sn_t_id_t_time'") {
  347. logs.Error(lib.FuncName(), err)
  348. return err
  349. }
  350. return nil
  351. }
  352. // 删除唯一索引
  353. func Delete_Unique_Index(T_task_id string) error {
  354. // 创建索引
  355. o := orm.NewOrm()
  356. // 删除索引
  357. deleteIndexSql := "ALTER TABLE z_task_data_" + T_task_id + " DROP INDEX unique_index_t_sn_t_id_t_time;"
  358. //fmt.Println(sql)
  359. _, err := o.Raw(deleteIndexSql).Exec()
  360. if err != nil {
  361. logs.Error(lib.FuncName(), err)
  362. return err
  363. }
  364. return nil
  365. }
  366. func Adds_TaskData(T_task_id string, valueStrings []string) error {
  367. o := orm.NewOrm()
  368. // 插入数据
  369. sql := "INSERT INTO z_task_data_" + T_task_id + " (`t_sn`, `t_id`, `t_t`, `t_rh`, `t_time`) VALUES"
  370. sql += strings.Join(valueStrings, ",")
  371. sql += " ON DUPLICATE KEY UPDATE t_t = VALUES(t_t), t_rh = VALUES(t_rh)"
  372. //fmt.Println(sql)
  373. _, err := o.Raw(sql).Exec()
  374. if err != nil {
  375. logs.Error(lib.FuncName(), err)
  376. return err
  377. }
  378. //fmt.Println("mysql row affected nums: ", num)
  379. return nil
  380. }
  381. // 修改
  382. func Up_TaskData(T_task_id string, Id string, T_t string, T_rh string, T_time string) bool {
  383. o := orm.NewOrm()
  384. SET_str := " SET "
  385. if len(T_t) > 0 {
  386. SET_str = SET_str + " t_t = " + T_t + " ,"
  387. }
  388. if len(T_rh) > 0 {
  389. SET_str = SET_str + " t_rh = " + T_rh + " ,"
  390. }
  391. if len(T_time) > 0 {
  392. SET_str = SET_str + " t_time = '" + T_time + "' ,"
  393. }
  394. if len(SET_str) > 7 {
  395. SET_str = SET_str[:(len(SET_str) - 1)]
  396. }
  397. // 开始插入数据 UPDATE `cold_verify`.`Z_TaskData_d8qMyeXLzIxn` SET `t_t` = 20.2 WHERE `ID` = 69
  398. sql := "UPDATE z_task_data_" + T_task_id + " " + SET_str + " WHERE ID = " + Id
  399. // 这里有时间优化 用于一次 prepare 多次 exec,以提高批量执行的速度
  400. //fmt.Println(sql)
  401. res, err := o.Raw(sql).Exec()
  402. if err != nil {
  403. logs.Error(lib.FuncName(), err)
  404. return false
  405. }
  406. res.RowsAffected()
  407. //fmt.Println("mysql row affected nums: ", num)
  408. return true
  409. }
  410. // 添加
  411. func Del_TaskData(T_task_id string, Id string) bool {
  412. o := orm.NewOrm()
  413. // 开始插入数据 UPDATE `cold_verify`.`Z_TaskData_d8qMyeXLzIxn` SET `t_t` = 20.2 WHERE `ID` = 69
  414. sql := "DELETE FROM z_task_data_" + T_task_id + " WHERE ID = " + Id
  415. // 这里有时间优化 用于一次 prepare 多次 exec,以提高批量执行的速度
  416. logs.Println(sql)
  417. res, err := o.Raw(sql).Exec()
  418. if err != nil {
  419. logs.Error(lib.FuncName(), err)
  420. return false
  421. }
  422. res.RowsAffected()
  423. //fmt.Println("mysql row affected nums: ", num)
  424. return true
  425. }
  426. func Read_TaskData_ByT_id(T_task_id string, Id string) (t_sn string, err error) {
  427. o := orm.NewOrm()
  428. // 开始插入数据 UPDATE `cold_verify`.`Z_TaskData_d8qMyeXLzIxn` SET `t_t` = 20.2 WHERE `ID` = 69
  429. sql := "SELECT t_sn FROM z_task_data_" + T_task_id + " WHERE t_id = '" + Id + "'"
  430. // 这里有时间优化 用于一次 prepare 多次 exec,以提高批量执行的速度
  431. logs.Println(sql)
  432. err = o.Raw(sql).QueryRow(&t_sn)
  433. if err != nil {
  434. logs.Error(lib.FuncName(), err)
  435. return t_sn, err
  436. }
  437. //fmt.Println("mysql row affected nums: ", num)
  438. return t_sn, err
  439. }
  440. func Read_TaskData_ByT_sn(T_task_id string, sn string) (t_id string, err error) {
  441. o := orm.NewOrm()
  442. // 开始插入数据 UPDATE `cold_verify`.`Z_TaskData_d8qMyeXLzIxn` SET `t_t` = 20.2 WHERE `ID` = 69
  443. sql := "SELECT t_id FROM z_task_data_" + T_task_id + " WHERE t_sn = '" + sn + "'"
  444. // 这里有时间优化 用于一次 prepare 多次 exec,以提高批量执行的速度
  445. logs.Println(sql)
  446. err = o.Raw(sql).QueryRow(&t_id)
  447. if err != nil {
  448. logs.Error(lib.FuncName(), err)
  449. return t_id, err
  450. }
  451. //fmt.Println("mysql row affected nums: ", num)
  452. return t_id, err
  453. }
  454. // 添加
  455. func Del_TaskData_t_id(T_task_id string, Id string) bool {
  456. o := orm.NewOrm()
  457. // 开始插入数据 UPDATE `cold_verify`.`Z_TaskData_d8qMyeXLzIxn` SET `t_t` = 20.2 WHERE `ID` = 69
  458. sql := "DELETE FROM z_task_data_" + T_task_id + " WHERE t_id = '" + Id + "'"
  459. // 这里有时间优化 用于一次 prepare 多次 exec,以提高批量执行的速度
  460. logs.Println(sql)
  461. res, err := o.Raw(sql).Exec()
  462. if err != nil {
  463. logs.Error(lib.FuncName(), err)
  464. return false
  465. }
  466. res.RowsAffected()
  467. //fmt.Println("mysql row affected nums: ", num)
  468. return true
  469. }
  470. // 检查导出表总数
  471. func Check_TaskData_Num(T_task_id string) bool {
  472. o1 := orm2.NewOrmUsingDB(conf.Local_AliasName)
  473. o2 := orm2.NewOrmUsingDB(conf.Server_AliasName)
  474. var maps_z1, maps_z2 []orm2.ParamsList
  475. sql := "SELECT COUNT(ID) FROM z_task_data_" + T_task_id
  476. fmt.Println(sql)
  477. _, err := o1.Raw(sql).ValuesList(&maps_z1)
  478. if err != nil {
  479. return false
  480. }
  481. _, err = o2.Raw(sql).ValuesList(&maps_z2)
  482. if err != nil {
  483. return false
  484. }
  485. if len(maps_z1) != len(maps_z2) {
  486. return false
  487. }
  488. return true
  489. }
  490. // 检查导出表总数
  491. func Check_TaskData_Num_Verify1(T_task_id string) bool {
  492. o1 := orm2.NewOrmUsingDB(conf.Local_AliasName)
  493. o2 := orm2.NewOrmUsingDB(conf.Verify1_AliasName)
  494. var maps_z1, maps_z2 []orm2.ParamsList
  495. sql := "SELECT COUNT(ID) FROM z_task_data_" + strings.ToLower(T_task_id)
  496. fmt.Println(sql)
  497. _, err := o1.Raw(sql).ValuesList(&maps_z1)
  498. if err != nil {
  499. return false
  500. }
  501. sql2 := "SELECT COUNT(ID) FROM Z_TaskData_" + T_task_id
  502. _, err = o2.Raw(sql2).ValuesList(&maps_z2)
  503. if err != nil {
  504. return false
  505. }
  506. if len(maps_z1) != len(maps_z2) {
  507. return false
  508. }
  509. return true
  510. }
  511. func Read_TaskData_Num(T_task_id string) int {
  512. o1 := orm2.NewOrmUsingDB(conf.Local_AliasName)
  513. var maps_z []orm2.ParamsList
  514. sql := "SELECT COUNT(ID) FROM z_task_data_" + T_task_id
  515. fmt.Println(sql)
  516. _, err := o1.Raw(sql).ValuesList(&maps_z)
  517. if err != nil {
  518. logs.Error(lib.FuncName(), err)
  519. return 0
  520. }
  521. key, _ := strconv.Atoi(maps_z[0][0].(string))
  522. return key
  523. }
  524. func Dump_TaskData(T_task_id, root, password, url_port, database, sql_file string) (string, error) {
  525. // url_port 127.0.0.1:3306
  526. // mysql8.0 以上加 --column-statistics=0
  527. // mysqldump --column-statistics=0 -uroot -proot -h127.0.0.1 -P3306 cold_verify Z_TaskData_ixEfo5zk2Oeb > /Data/Z_TaskData_ixEfo5zk2Oeb.sql
  528. v := Read_Local_Mysql_Version()
  529. host_port := strings.Split(url_port, ":")
  530. table_name := "z_task_data_" + T_task_id
  531. org := "mysqldump "
  532. if v >= 8 {
  533. org += "--column-statistics=0 "
  534. }
  535. //--no-create-info 只导出数据,而不添加 CREATE TABLE 语句。
  536. //--single-transaction 在备份库的时候并不锁定数据表
  537. //--add-locks:在每个表导出之前增加LOCK TABLES并且之后UNLOCK TABLE。(默认为打开状态,使用--skip-add-locks取消选项)
  538. //--compact:压缩模式,产生更少的输出;
  539. org = org + fmt.Sprintf("-u%s -p%s -h%s -P%s --no-create-info --set-gtid-purged=OFF --skip-add-locks --compact --quick %s %s > %s ",
  540. root, password, host_port[0], host_port[1], database, table_name, sql_file)
  541. logs.Println(org)
  542. _, err := lib.Command(org)
  543. if err != nil {
  544. logs.Error(lib.FuncName(), err)
  545. }
  546. return org, err
  547. }
  548. func Dump_TaskData_Verify(T_task_id, root, password, url_port, database, sql_file string) (string, error) {
  549. // url_port 127.0.0.1:3306
  550. // mysql8.0 以上加 --column-statistics=0
  551. // mysqldump --column-statistics=0 -uroot -proot -h127.0.0.1 -P3306 cold_verify Z_TaskData_ixEfo5zk2Oeb > /Data/Z_TaskData_ixEfo5zk2Oeb.sql
  552. v := Read_Local_Mysql_Version()
  553. host_port := strings.Split(url_port, ":")
  554. table_name := "Z_TaskData_" + T_task_id
  555. org := "mysqldump "
  556. if v >= 8 {
  557. org += "--column-statistics=0 "
  558. }
  559. //--no-create-info 只导出数据,而不添加 CREATE TABLE 语句。
  560. //--single-transaction 在备份库的时候并不锁定数据表
  561. //--add-locks:在每个表导出之前增加LOCK TABLES并且之后UNLOCK TABLE。(默认为打开状态,使用--skip-add-locks取消选项)
  562. //--compact:压缩模式,产生更少的输出;
  563. org = org + fmt.Sprintf("-u%s -p%s -h%s -P%s --no-create-info --set-gtid-purged=OFF --skip-add-locks --compact --quick %s %s > %s ",
  564. root, password, host_port[0], host_port[1], database, table_name, sql_file)
  565. logs.Println(org)
  566. _, err := lib.Command(org)
  567. if err != nil {
  568. logs.Error(lib.FuncName(), err)
  569. }
  570. return org, err
  571. }
  572. func Insert_TaskData(root, password, url_port, database, sql_file string) (string, error) {
  573. // url_port 127.0.0.1:3306
  574. // mysql -u root -p root -h127.0.0.1 -P3306 cold_verify_local < /data/Z_TaskData_ixEfo5zk2Oeb.sql
  575. host_port := strings.Split(url_port, ":")
  576. org := fmt.Sprintf("mysql -u%s -p%s -h%s -P%s %s < %s",
  577. root, password, host_port[0], host_port[1], database, sql_file)
  578. logs.Println(org)
  579. _, err := lib.Command(org)
  580. if err != nil {
  581. logs.Error(lib.FuncName(), err)
  582. }
  583. return org, err
  584. }
  585. func Read_Local_Mysql_Version() int {
  586. o := orm2.NewOrmUsingDB(conf.Local_AliasName)
  587. var params []orm2.Params
  588. o.Raw("select version();").Values(&params)
  589. version, _ := strconv.Atoi(params[0]["version()"].(string)[0:1])
  590. return version
  591. }
  592. func Import_TaskData(T_task_id string, offset, pagez int) bool {
  593. var maps []TaskData_
  594. serverOrm := orm2.NewOrmUsingDB(conf.Server_AliasName)
  595. localOrm := orm2.NewOrmUsingDB(conf.Local_AliasName)
  596. tb_name := "z_task_data_" + T_task_id
  597. // 开从先上版查询数据
  598. sql := "select `ID`, `t_sn`, `t_id`, `t_t`, `t_rh`, DATE_FORMAT(t_time,'%Y-%m-%d %H:%i:%s') AS t_times from " + tb_name + " LIMIT " + strconv.Itoa(offset) + "," + strconv.Itoa(pagez)
  599. logs.Println(sql)
  600. _, err := serverOrm.Raw(sql).QueryRows(&maps)
  601. if err != nil {
  602. logs.Error(lib.FuncName(), err)
  603. return false
  604. }
  605. values := []string{}
  606. for _, row := range maps {
  607. // 处理数据
  608. values = append(values, fmt.Sprintf("('%s',%d,%v,%v,'%s')", row.T_sn, row.T_id, row.T_t, row.T_rh, row.T_time))
  609. }
  610. // 向本地版插入数据
  611. sql = "INSERT IGNORE INTO " + tb_name + "( `t_sn`, `t_id`, `t_t`, `t_rh`, `t_time`) VALUES " + strings.Join(values, ",")
  612. //logs.Println(sql)
  613. _, err = localOrm.Raw(sql).Exec()
  614. if err != nil {
  615. logs.Error(lib.FuncName(), err)
  616. return false
  617. }
  618. return true
  619. }
  620. // InsertTaskData 添加设备数据
  621. func InsertTaskData(T_task_id string, d TaskData_) {
  622. o := orm.NewOrm()
  623. var maps TaskData_
  624. sql := fmt.Sprintf("SELECT * FROM z_task_data_%s where t_time='%s' and t_id = '%s' and t_sn = '%s'", T_task_id, d.T_time, d.T_id, d.T_sn)
  625. err := o.Raw(sql).QueryRow(&maps)
  626. if err != nil && err.Error() != orm.ErrNoRows.Error() {
  627. logs.Error(lib.FuncName(), err)
  628. }
  629. // 添加
  630. if maps.ID == 0 {
  631. sqlStatement := fmt.Sprintf("insert into z_task_data_%s(t_sn, t_id, t_t, t_rh, t_time) value('%s','%s',%f,%f,'%s')", T_task_id, d.T_sn, d.T_id, d.T_t, d.T_rh, d.T_time)
  632. _, err = o.Raw(sqlStatement).Exec()
  633. if err != nil {
  634. logs.Error(lib.FuncName(), err)
  635. }
  636. return
  637. }
  638. //已经存在该条目,更新
  639. sqlStatement := fmt.Sprintf("update z_task_data_%s set t_t = %f,t_rh = %f where t_time= '%s' and t_id = '%s' and t_sn = '%s'", T_task_id, d.T_t, d.T_rh, d.T_time, d.T_id, d.T_sn)
  640. _, err = o.Raw(sqlStatement).Exec()
  641. if err != nil {
  642. logs.Error(lib.FuncName(), err)
  643. }
  644. }
  645. // DeleteTaskDataByTimeRange 删除时间范围内的数据
  646. func DeleteTaskDataByTimeRange(T_task_id, sn string, id string, start, end string) {
  647. sqlStatement := fmt.Sprintf("delete from z_task_data_%s where t_time between '%s' and '%s' and t_id = '%s' and t_sn = '%s'", T_task_id, start, end, id, sn)
  648. o := orm.NewOrm()
  649. exec, err := o.Raw(sqlStatement).Exec()
  650. if err != nil {
  651. fmt.Println(err.Error())
  652. }
  653. affected, _ := exec.RowsAffected()
  654. fmt.Printf("从%s~%s时间段删除了%d条数据", start, end, affected)
  655. }
  656. // UpdateTaskDataTemperatureAndHumidityRandom 随机更新
  657. func UpdateTaskDataTemperatureAndHumidityRandom(T_task_id, sn, id, startTime, endTime string, ttMax, ttMin, trhMax, trhMin int) {
  658. //sql语句
  659. sqlStatement := fmt.Sprintf("update z_task_data_%s set t_t = t_t + FLOOR(%d + RAND() * (%d - %d + 1)) /100.0, t_rh = t_rh + FLOOR(%d + RAND() * (%d - %d + 1)) /100.0 where t_time BETWEEN '%s' AND '%s' and t_id = '%s' and t_sn = '%s'", T_task_id, ttMin, ttMax, ttMin, trhMin, trhMax, trhMin, startTime, endTime, id, sn)
  660. o := orm.NewOrm()
  661. exec, err := o.Raw(sqlStatement).Exec()
  662. if err != nil {
  663. fmt.Println("执行错误:", sqlStatement)
  664. }
  665. affected, _ := exec.RowsAffected()
  666. fmt.Println("执行更新行数:", affected)
  667. }
  668. // UpdateTaskDataTemperatureAndHumidity 更新设备探头数据温湿度
  669. func UpdateTaskDataTemperatureAndHumidity(T_task_id, sn, id, startTime, endTime string, temperature, humidity float64) {
  670. sqlStatement := fmt.Sprintf("update z_task_data_%s set t_t = t_t + %f , t_rh = t_rh + %f where t_id = '%s' and t_sn = '%s' and t_time BETWEEN '%s' and '%s'", T_task_id, temperature, humidity, id, sn, startTime, endTime)
  671. o := orm.NewOrm()
  672. exec, err := o.Raw(sqlStatement).Exec()
  673. if err != nil {
  674. fmt.Println(err.Error())
  675. }
  676. affected, err := exec.RowsAffected()
  677. if err != nil {
  678. fmt.Println(err.Error())
  679. }
  680. fmt.Printf("影响了%d行\n", affected)
  681. }
  682. // UpdateTaskData 更新设备探头数据
  683. func UpdateTaskData(T_task_id, sn, id string, old, newO TaskData_) {
  684. sqlStatement := fmt.Sprintf("update z_task_data_%s set t_t = %f , t_rh = %f where t_sn = '%s' and t_id = '%s' and t_t = %f and t_rh = %f and t_time= '%s'", T_task_id, newO.T_t, newO.T_rh, sn, id, old.T_t, old.T_rh, old.T_time)
  685. fmt.Println("执行SQL:", sqlStatement)
  686. o := orm.NewOrm()
  687. exec, err := o.Raw(sqlStatement).Exec()
  688. if err != nil {
  689. fmt.Println("执行错误:", sqlStatement, err.Error())
  690. }
  691. affected, _ := exec.RowsAffected()
  692. fmt.Println("执行更新行数:", affected)
  693. }
  694. // 查询温度最小值 最大值
  695. func Read_TaskData_T_Min_Max(T_task_id string, SN string, T_id string, Time_start_ string, Time_end_ string) (minT, maxT float64) {
  696. o := orm.NewOrm()
  697. sql_condition := ""
  698. if len(Time_start_) > 1 {
  699. sql_condition += " AND t_time >= '" + Time_start_ + "'"
  700. }
  701. if len(Time_end_) > 1 {
  702. sql_condition += " AND t_time <= '" + Time_end_ + "'"
  703. }
  704. if len(T_id) > 0 {
  705. sql_condition += " AND t_id = '" + T_id + "'"
  706. }
  707. if len(SN) > 0 {
  708. sql_condition += " AND t_sn = '" + SN + "'"
  709. }
  710. if len(sql_condition) > 0 {
  711. sql_condition = " WHERE " + strings.TrimLeft(sql_condition, " AND ")
  712. }
  713. //fmt.Println("maps_z;",maps_z[0][0])
  714. sql := "SELECT MIN(t_t) AS min_t, MAX(t_t) AS max_t FROM z_task_data_" + T_task_id + sql_condition
  715. fmt.Println(sql)
  716. err := o.Raw(sql).QueryRow(&minT, &maxT)
  717. if err != nil {
  718. logs.Error(lib.FuncName(), err)
  719. }
  720. return
  721. }
  722. // UpdateTaskDataTemperatureAndHumidityByGeometric 更新设备探头数据温湿度 等比缩放
  723. func UpdateTaskDataTemperatureAndHumidityByGeometric(T_task_id, sn, id, startTime, endTime string, temperature, humidity float64) {
  724. sqlStatement := fmt.Sprintf("update z_task_data_%s set t_t = t_t * %f , t_rh = t_rh * %f where t_id = '%s' and t_sn = '%s' and t_time BETWEEN '%s' and '%s'", T_task_id, temperature, humidity, id, sn, startTime, endTime)
  725. o := orm.NewOrm()
  726. exec, err := o.Raw(sqlStatement).Exec()
  727. if err != nil {
  728. fmt.Println(err.Error())
  729. }
  730. affected, err := exec.RowsAffected()
  731. if err != nil {
  732. fmt.Println(err.Error())
  733. }
  734. fmt.Printf("影响了%d行\n", affected)
  735. }
  736. func Read_TaskData_T_Min_Max_Time_Min_Max(T_task_id string, SN []string, T_id []string, Time_start_ string, Time_end_ string) (minT, maxT float64, minTime, maxTime time.Time) {
  737. o := orm.NewOrm()
  738. sql_condition := ""
  739. if len(Time_start_) > 1 {
  740. sql_condition += " AND t_time >= '" + Time_start_ + "'"
  741. }
  742. if len(Time_end_) > 1 {
  743. sql_condition += " AND t_time <= '" + Time_end_ + "'"
  744. }
  745. if len(T_id) > 0 || len(SN) > 0 {
  746. sql_condition += " AND t_id in (" + strings.Join(T_id, ",") + ") OR t_sn in (" + strings.Join(SN, ",") + ")"
  747. }
  748. if len(sql_condition) > 0 {
  749. sql_condition = " WHERE " + strings.TrimLeft(sql_condition, " AND ")
  750. }
  751. //fmt.Println("maps_z;",maps_z[0][0])
  752. sql := "SELECT MIN(t_t) AS min_t, MAX(t_t) AS max_t,MIN(t_time) AS min_time, MAX(t_time) AS max_time FROM z_task_data_" + T_task_id + sql_condition
  753. fmt.Println(sql)
  754. err := o.Raw(sql).QueryRow(&minT, &maxT, &minTime, &maxTime)
  755. if err != nil {
  756. logs.Error(lib.FuncName(), err)
  757. }
  758. return
  759. }
  760. // 获取线上设备数据
  761. func Read_DeviceData_T_Min_Max_Time_Min_Max(T_sn, T_id string, Time_start_ string, Time_end_ string) (minT, maxT float64, minTime, maxTime time.Time) {
  762. o := orm2.NewOrmUsingDB(conf.Server_AliasName)
  763. sql_condition := ""
  764. if len(Time_start_) > 1 {
  765. sql_condition += " AND t_time >= '" + Time_start_ + "'"
  766. }
  767. if len(Time_end_) > 1 {
  768. sql_condition += " AND t_time <= '" + Time_end_ + "'"
  769. }
  770. sql_condition += " AND t_id = '" + T_id + "'"
  771. if len(sql_condition) > 0 {
  772. sql_condition = " WHERE " + strings.TrimLeft(sql_condition, " AND ")
  773. }
  774. //fmt.Println("maps_z;",maps_z[0][0])
  775. sql := "SELECT MIN(t_t) AS min_t, MAX(t_t) AS max_t,MIN(t_time) AS min_time, MAX(t_time) AS max_time FROM z_devicedata_" + T_sn + sql_condition
  776. fmt.Println(sql)
  777. err := o.Raw(sql).QueryRow(&minT, &maxT, &minTime, &maxTime)
  778. if err != nil {
  779. logs.Error(lib.FuncName(), err)
  780. }
  781. return
  782. }
  783. // 获取线上设备数据
  784. func Read_DeviceData_ById_List(T_sn, T_id string, Time_start_ string, Time_end_ string) []TaskData_ {
  785. o := orm2.NewOrmUsingDB(conf.Server_AliasName)
  786. var maps []TaskData_
  787. sql_condition := ""
  788. if len(Time_start_) > 1 {
  789. sql_condition += " AND t_time >= '" + Time_start_ + "'"
  790. }
  791. if len(Time_end_) > 1 {
  792. sql_condition += " AND t_time <= '" + Time_end_ + "'"
  793. }
  794. sql_condition += " AND t_id = '" + T_id + "'"
  795. if len(sql_condition) > 0 {
  796. sql_condition = " WHERE " + strings.TrimLeft(sql_condition, " AND ")
  797. }
  798. //fmt.Println("maps_z;",maps_z[0][0])
  799. sql := "SELECT ID,t_id,t_t,t_rh,DATE_FORMAT(t_time,'%Y-%m-%d %H:%i:%s') AS t_times,t_time FROM z_devicedata_" + T_sn + sql_condition + " ORDER BY t_time"
  800. fmt.Println(sql)
  801. _, err := o.Raw(sql).QueryRows(&maps)
  802. if err != nil {
  803. logs.Error(lib.FuncName(), err)
  804. }
  805. return maps
  806. }
  807. func Read_TaskData_ByIds_List(T_task_id string, SN []string, T_id []string, Time_start_ string, Time_end_ string) []TaskData {
  808. o := orm.NewOrm()
  809. var maps []TaskData
  810. sql_condition := ""
  811. if len(Time_start_) > 1 {
  812. sql_condition += " AND t_time >= '" + Time_start_ + "'"
  813. }
  814. if len(Time_end_) > 1 {
  815. sql_condition += " AND t_time <= '" + Time_end_ + "'"
  816. }
  817. if len(T_id) > 0 || len(SN) > 0 {
  818. sql_condition += " AND (t_id in (" + strings.Join(T_id, ",") + ") OR t_sn in (" + strings.Join(SN, ",") + "))"
  819. }
  820. if len(sql_condition) > 0 {
  821. sql_condition = " WHERE " + strings.TrimLeft(sql_condition, " AND ")
  822. }
  823. //fmt.Println("maps_z;",maps_z[0][0])
  824. sql := "SELECT ID,t_sn,t_id,t_t,t_rh,t_time FROM z_task_data_" + T_task_id + sql_condition + " ORDER BY t_time"
  825. fmt.Println(sql)
  826. _, err := o.Raw(sql).QueryRows(&maps)
  827. if err != nil {
  828. logs.Error(lib.FuncName(), err)
  829. }
  830. return maps
  831. }
  832. // 按照 t_id 分组数据
  833. func GroupDataByTID(data []TaskData) map[string][]TaskData {
  834. groupedData := make(map[string][]TaskData)
  835. for _, d := range data {
  836. groupedData[d.T_id] = append(groupedData[d.T_id], d)
  837. }
  838. return groupedData
  839. }
  840. // 删除的重复数据
  841. func DeleteDeduplicate(T_task_id string) error {
  842. localOrm := orm2.NewOrmUsingDB(conf.Local_AliasName)
  843. tb_name := "z_task_data_" + T_task_id
  844. // 创建临时表
  845. sqlCreate := "CREATE TABLE `tmp_table` AS (SELECT MAX(`ID`) AS `max_id` FROM " + tb_name + " GROUP BY `t_sn`,`t_id`,`t_time`);"
  846. _, err := localOrm.Raw(sqlCreate).Exec()
  847. if err != nil {
  848. logs.Error(lib.FuncName(), err)
  849. return err
  850. }
  851. sqlDelete := "DELETE FROM " + tb_name + " WHERE `ID` NOT IN (SELECT `max_id` FROM `tmp_table`);"
  852. _, err = localOrm.Raw(sqlDelete).Exec()
  853. if err != nil {
  854. logs.Error(lib.FuncName(), err)
  855. return err
  856. }
  857. sqlDrop := "DROP TABLE `tmp_table`;"
  858. _, err = localOrm.Raw(sqlDrop).Exec()
  859. if err != nil {
  860. logs.Error(lib.FuncName(), err)
  861. return err
  862. }
  863. return nil
  864. }