diff --git a/internal/store/postgres/device_store.go b/internal/store/postgres/device_store.go index 3f6001d..c911414 100644 --- a/internal/store/postgres/device_store.go +++ b/internal/store/postgres/device_store.go @@ -250,3 +250,10 @@ func (s *DeviceStore) UpdatePollingStatus(ctx context.Context, deviceID uint, en Where("id = ?", deviceID). Update("enable_polling", enablePolling).Error } + +// UpdateRealnamePolicy 更新设备的实名认证策略 +func (s *DeviceStore) UpdateRealnamePolicy(ctx context.Context, deviceID uint, realnamePolicy string) error { + return s.db.WithContext(ctx).Model(&model.Device{}). + Where("id = ?", deviceID). + Update("realname_policy", realnamePolicy).Error +} diff --git a/internal/store/postgres/iot_card_store.go b/internal/store/postgres/iot_card_store.go index 7967052..06bd027 100644 --- a/internal/store/postgres/iot_card_store.go +++ b/internal/store/postgres/iot_card_store.go @@ -124,6 +124,13 @@ func (s *IotCardStore) UpdatePollingStatus(ctx context.Context, cardID uint, ena Update("enable_polling", enablePolling).Error } +// UpdateRealnamePolicy 更新卡的实名认证策略 +func (s *IotCardStore) UpdateRealnamePolicy(ctx context.Context, cardID uint, realnamePolicy string) error { + return s.db.WithContext(ctx).Model(&model.IotCard{}). + Where("id = ?", cardID). + Update("realname_policy", realnamePolicy).Error +} + func (s *IotCardStore) Delete(ctx context.Context, id uint) error { return s.db.WithContext(ctx).Delete(&model.IotCard{}, id).Error }