runtime/v2: Fix defer cleanup

wzshiming · wzshiming · commit 30e1e66e5cb1 · 2021-03-20T18:40:36.000+08:00
Signed-off-by: Shiming Zhang &lt;wzshiming@foxmail.com&gt;
diff --git a/runtime/v2/manager.go b/runtime/v2/manager.go
@@ -24,6 +24,7 @@ import (
 	"path/filepath"
 
 	"github.com/containerd/containerd/containers"
+	"github.com/containerd/containerd/errdefs"
 	"github.com/containerd/containerd/events/exchange"
 	"github.com/containerd/containerd/log"
 	"github.com/containerd/containerd/metadata"
@@ -155,6 +156,10 @@ func (m *TaskManager) Create(ctx context.Context, id string, opts runtime.Create
 			defer cancel()
 			_, errShim := shim.Delete(dctx)
 			if errShim != nil {
+				if errdefs.IsDeadlineExceeded(errShim) {
+					dctx, cancel = timeout.WithContext(context.Background(), cleanupTimeout)
+					defer cancel()
+				}
 				shim.Shutdown(dctx)
 				shim.Close()
 			}
