fix: creating multiple pods at the same time will be assigned to the same NPU.

Author: lomtom

internal/server/server.go

@@ -20,9 +20,11 @@ import (
 	"context"
 	"flag"
 	"fmt"
+	"math"
 	"net"
 	"os"
 	"path"
+	"strconv"
 	"strings"
 	"time"
 
@@ -350,6 +352,15 @@ func (ps *PluginServer) Allocate(ctx context.Context, reqs *v1beta1.AllocateRequ
 	success := false
 	var pod *v1.Pod
 	defer func() {
+		if success {
+			annotations := make(map[string]string)
+			annotations[util.DeviceBindPhase] = util.DeviceBindSuccess
+			annotations[util.AssignedTimeAnnotations] = strconv.FormatUint(math.MaxUint64, 10)
+			err := util.PatchPodAnnotations(pod, annotations)
+			if err != nil {
+				klog.Errorf("patch pod %s annotations error: %v", pod.Name, err)
+			}
+		}
 		lockerr := nodelock.ReleaseNodeLock(ps.nodeName, NodeLockAscend, pod, success)
 		if lockerr != nil {
 			klog.Errorf("failed to release lock:%s", lockerr.Error())