修复存储的token失效的问题

This commit is contained in:
Alex Yang
2025-12-03 00:13:57 +08:00
parent 8bad1544cd
commit a7d794405c
2 changed files with 14 additions and 3 deletions

View File

@@ -50,9 +50,12 @@ func (s *MySQLStorage) GetDevices() []Device {
devices := make([]Device, 0)
for rows.Next() {
var device Device
if err := rows.Scan(&device.ID, &device.Name, &device.IP, &device.Token, &device.Status, &device.CreatedAt, &device.UpdatedAt); err != nil {
var createdAt, updatedAt int64
if err := rows.Scan(&device.ID, &device.Name, &device.IP, &device.Token, &device.Status, &createdAt, &updatedAt); err != nil {
continue
}
device.CreatedAt = createdAt
device.UpdatedAt = updatedAt
devices = append(devices, device)
}
@@ -65,10 +68,13 @@ func (s *MySQLStorage) GetDevice(id string) (Device, bool) {
row := s.db.QueryRow(query, id)
var device Device
err := row.Scan(&device.ID, &device.Name, &device.IP, &device.Token, &device.Status, &device.CreatedAt, &device.UpdatedAt)
var createdAt, updatedAt int64
err := row.Scan(&device.ID, &device.Name, &device.IP, &device.Token, &device.Status, &createdAt, &updatedAt)
if err != nil {
return Device{}, false
}
device.CreatedAt = createdAt
device.UpdatedAt = updatedAt
return device, true
}
@@ -79,11 +85,16 @@ func (s *MySQLStorage) GetDeviceByToken(token string) (Device, bool) {
row := s.db.QueryRow(query, token)
var device Device
err := row.Scan(&device.ID, &device.Name, &device.IP, &device.Token, &device.Status, &device.CreatedAt, &device.UpdatedAt)
var createdAt, updatedAt int64
err := row.Scan(&device.ID, &device.Name, &device.IP, &device.Token, &device.Status, &createdAt, &updatedAt)
if err != nil {
return Device{}, false
}
device.CreatedAt = createdAt
device.UpdatedAt = updatedAt
return device, true
}

Binary file not shown.