fix(usbdevice): close closed channel exception

Signed-off-by: Jack Yu <jack.yu@suse.com>
harvester · Jul 31, 2024 · cfe1f77 · cfe1f77
1 parent 6a90910
commit cfe1f77
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pkg/deviceplugins/usb_device_plugin.go b/pkg/deviceplugins/usb_device_plugin.go
@@ -114,7 +114,7 @@ func (plugin *USBDevicePlugin) stopDevicePlugin() error {
 }
 
 func (plugin *USBDevicePlugin) startDevicePlugin() error {
-	plugin.deregistered = make(chan struct{})
+	plugin.deregistered = make(chan struct{}, 1)
 
 	err := plugin.cleanup()
 	if err != nil {
@@ -310,7 +310,12 @@ func (plugin *USBDevicePlugin) ListAndWatch(_ *pluginapi.Empty, lws pluginapi.De
 		plugin.logger.Reason(err).Warningf("Failed to deregister device plugin %s",
 			plugin.resourceName)
 	}
-	close(plugin.deregistered)
+
+	select {
+	case plugin.deregistered <- struct{}{}:
+	default:
+	}
+
 	return nil
 }