return NotFound if response error NotFound

whywaita · Dec 9, 2021 · ba52677 · ba52677
1 parent 942494f
commit ba52677
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/shoes-lxd-multi/main.go b/shoes-lxd-multi/main.go
@@ -152,7 +152,12 @@ func (l Client) DeleteInstance(ctx context.Context, req *pb.DeleteInstanceReques
 	}
 
 	if _, err := slClient.DeleteInstance(ctx, slReq); err != nil {
-		return nil, status.Errorf(codes.Internal, "failed to DeleteInstance: %+v", err)
+		switch status.Code(err) {
+		case codes.NotFound:
+			return nil, status.Errorf(codes.NotFound, "failed to DeleteInstance: %+v", err)
+		default:
+			return nil, status.Errorf(codes.Internal, "failed to DeleteInstance: %+v", err)
+		}
 	}
 	return &pb.DeleteInstanceResponse{}, nil
 }