250 lines
5.8 KiB
Go
250 lines
5.8 KiB
Go
package main
|
||
|
||
import (
|
||
"database/sql"
|
||
"fmt"
|
||
"time"
|
||
|
||
_ "github.com/go-sql-driver/mysql"
|
||
)
|
||
|
||
var db *sql.DB
|
||
|
||
const SCALING_FACTOR = 1000 // 浮点数到整数的转换因子
|
||
|
||
// 初始化数据库连接
|
||
func InitDB() error {
|
||
username := "remote"
|
||
password := "root"
|
||
host := "8.134.185.53"
|
||
port := "3306"
|
||
dbName := "probe_db"
|
||
|
||
// 连接字符串
|
||
dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?parseTime=true", username, password, host, port, dbName)
|
||
|
||
var err error
|
||
db, err = sql.Open("mysql", dsn)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
err = db.Ping()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
db.SetMaxOpenConns(10)
|
||
db.SetMaxIdleConns(5)
|
||
db.SetConnMaxLifetime(time.Minute * 5)
|
||
|
||
return nil
|
||
}
|
||
|
||
// 关闭数据库连接
|
||
func CloseDB() {
|
||
if db != nil {
|
||
db.Close()
|
||
}
|
||
}
|
||
|
||
// EnsureDeviceExists 确保设备存在，不存在则创建为默认配置
|
||
func EnsureDeviceExists(deviceID int) error {
|
||
if deviceID <= 0 {
|
||
return nil
|
||
}
|
||
_, err := db.Exec(
|
||
"INSERT INTO devices (id, forward_enable, host, port) VALUES (?, 0, NULL, NULL) ON DUPLICATE KEY UPDATE id = id",
|
||
deviceID,
|
||
)
|
||
return err
|
||
}
|
||
|
||
// 保存传感器数据 - 将浮点值转换为整数存储，添加温度与设备ID支持
|
||
func SaveSensorData(sensorID int, x, y, z, temperature float64, deviceID int) error {
|
||
xInt := int(x * SCALING_FACTOR)
|
||
yInt := int(y * SCALING_FACTOR)
|
||
zInt := int(z * SCALING_FACTOR)
|
||
tempInt := int(temperature * SCALING_FACTOR)
|
||
|
||
if deviceID > 0 {
|
||
query := `INSERT INTO sensor_data (sensor_id, x_value, y_value, z_value, temperature, device_id) VALUES (?, ?, ?, ?, ?, ?)`
|
||
_, err := db.Exec(query, sensorID, xInt, yInt, zInt, tempInt, deviceID)
|
||
return err
|
||
}
|
||
|
||
query := `INSERT INTO sensor_data (sensor_id, x_value, y_value, z_value, temperature, device_id) VALUES (?, ?, ?, ?, ?, NULL)`
|
||
_, err := db.Exec(query, sensorID, xInt, yInt, zInt, tempInt)
|
||
return err
|
||
}
|
||
|
||
// 获取传感器数据 - 添加时间范围，包含温度字段
|
||
func GetSensorData(sensorID int, limit int, startDate time.Time, endDate time.Time) ([]SensorData, error) {
|
||
query := `SELECT id, sensor_id, x_value, y_value, z_value,
|
||
COALESCE(temperature, 0) as temperature,
|
||
timestamp as timestamp
|
||
FROM sensor_data
|
||
WHERE sensor_id = ?`
|
||
|
||
var args []interface{}
|
||
args = append(args, sensorID)
|
||
|
||
if !startDate.IsZero() {
|
||
query += " AND timestamp >= ?"
|
||
args = append(args, startDate)
|
||
}
|
||
|
||
if !endDate.IsZero() {
|
||
query += " AND timestamp <= ?"
|
||
args = append(args, endDate)
|
||
}
|
||
|
||
query += " ORDER BY timestamp DESC"
|
||
|
||
// 只有当limit > 0时才添加LIMIT子句
|
||
if limit > 0 {
|
||
query += " LIMIT ?"
|
||
args = append(args, limit)
|
||
}
|
||
|
||
rows, err := db.Query(query, args...)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
defer rows.Close()
|
||
|
||
var result []SensorData
|
||
|
||
for rows.Next() {
|
||
var data SensorData
|
||
var xInt, yInt, zInt, tempInt int
|
||
|
||
err := rows.Scan(&data.ID, &data.SensorID, &xInt, &yInt, &zInt, &tempInt, &data.Timestamp)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
data.X = float64(xInt) / SCALING_FACTOR
|
||
data.Y = float64(yInt) / SCALING_FACTOR
|
||
data.Z = float64(zInt) / SCALING_FACTOR
|
||
data.Temperature = float64(tempInt) / SCALING_FACTOR
|
||
|
||
result = append(result, data)
|
||
}
|
||
|
||
return result, nil
|
||
}
|
||
|
||
// 获取所有传感器数据，包含温度字段
|
||
func GetAllSensorData(limit int, startDate time.Time, endDate time.Time) ([]SensorData, error) {
|
||
query := `SELECT id, sensor_id, x_value, y_value, z_value,
|
||
COALESCE(temperature, 0) as temperature,
|
||
timestamp as timestamp
|
||
FROM sensor_data
|
||
WHERE 1=1`
|
||
|
||
var args []interface{}
|
||
|
||
if !startDate.IsZero() {
|
||
query += " AND timestamp >= ?"
|
||
args = append(args, startDate)
|
||
}
|
||
|
||
if !endDate.IsZero() {
|
||
query += " AND timestamp <= ?"
|
||
args = append(args, endDate)
|
||
}
|
||
|
||
query += " ORDER BY timestamp DESC"
|
||
|
||
// 只有当limit > 0时才添加LIMIT子句
|
||
if limit > 0 {
|
||
query += " LIMIT ?"
|
||
args = append(args, limit)
|
||
}
|
||
|
||
rows, err := db.Query(query, args...)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
defer rows.Close()
|
||
|
||
var result []SensorData
|
||
|
||
for rows.Next() {
|
||
var data SensorData
|
||
var xInt, yInt, zInt, tempInt int
|
||
|
||
err := rows.Scan(&data.ID, &data.SensorID, &xInt, &yInt, &zInt, &tempInt, &data.Timestamp)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
data.X = float64(xInt) / SCALING_FACTOR
|
||
data.Y = float64(yInt) / SCALING_FACTOR
|
||
data.Z = float64(zInt) / SCALING_FACTOR
|
||
data.Temperature = float64(tempInt) / SCALING_FACTOR
|
||
|
||
result = append(result, data)
|
||
}
|
||
|
||
return result, nil
|
||
}
|
||
|
||
// 获取所有传感器ID
|
||
func GetAllSensorIDs() ([]int, error) {
|
||
query := `SELECT DISTINCT sensor_id FROM sensor_data ORDER BY sensor_id`
|
||
|
||
rows, err := db.Query(query)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
defer rows.Close()
|
||
|
||
var ids []int
|
||
for rows.Next() {
|
||
var id int
|
||
if err := rows.Scan(&id); err != nil {
|
||
return nil, err
|
||
}
|
||
ids = append(ids, id)
|
||
}
|
||
|
||
return ids, nil
|
||
}
|
||
|
||
// SensorData 结构用于存储传感器数据，添加温度字段
|
||
type SensorData struct {
|
||
ID int `json:"id"`
|
||
SensorID int `json:"sensor_id"`
|
||
X float64 `json:"x"`
|
||
Y float64 `json:"y"`
|
||
Z float64 `json:"z"`
|
||
Temperature float64 `json:"temperature"`
|
||
Timestamp time.Time `json:"timestamp"`
|
||
}
|
||
|
||
// Device 表映射
|
||
type Device struct {
|
||
ID int
|
||
ForwardEnable bool
|
||
Host sql.NullString
|
||
Port sql.NullInt64
|
||
RegCodeHex sql.NullString
|
||
}
|
||
|
||
// GetDevice 获取设备配置
|
||
func GetDevice(deviceID int) (*Device, error) {
|
||
row := db.QueryRow(`SELECT id, COALESCE(forward_enable, 0) as forward_enable, host, port, reg_code_hex FROM devices WHERE id = ?`, deviceID)
|
||
var d Device
|
||
var fe int
|
||
if err := row.Scan(&d.ID, &fe, &d.Host, &d.Port, &d.RegCodeHex); err != nil {
|
||
if err == sql.ErrNoRows {
|
||
return nil, nil
|
||
}
|
||
return nil, err
|
||
}
|
||
d.ForwardEnable = fe != 0
|
||
return &d, nil
|
||
}
|