Graceful shutdown

Author: lukasfrank

claimutils/claim/claimer.go

@@ -110,13 +110,25 @@ func (c *claimer) Start(ctx context.Context) error {
 	}
 
 	go c.run(ctx)
-
 	<-ctx.Done()
-	c.running.Store(false)
+
 	return nil
 }
 
 func (c *claimer) run(ctx context.Context) {
+	defer func() {
+		c.running.Store(false)
+		close(c.toClaim)
+		close(c.toRelease)
+
+		for req := range c.toClaim {
+			req.resultChan <- claimRes{err: ctx.Err()}
+		}
+		for req := range c.toRelease {
+			req.resultChan <- ctx.Err()
+		}
+	}()
+
 	for {
 		select {
 		case <-ctx.Done():

claimutils/gpu/gpu_test.go

@@ -32,7 +32,7 @@ var _ = Describe("GPU Claimer", func() {
 		plugin := gpu.NewGPUClaimPlugin(log.FromContext(ctx), "test-plugin", nil, nil)
 		Expect(plugin.Init()).Should(HaveOccurred())
 
-		By("init plugin without reader")
+		By("init plugin reader")
 		plugin = gpu.NewGPUClaimPlugin(log.FromContext(ctx), "test-plugin", &MockReader{}, nil)
 		Expect(plugin.Init()).ShouldNot(HaveOccurred())