TaskData.go 30 KB

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