add running model info

server/workerstatu.go

@@ -85,11 +85,26 @@ func (wm *WorkerManager) AddWorkerFirst(worker *Worker) error {
 		Hardware: types.PbToHardwareInfo(worker.info.nodeInfo.Hardware),
 	})
 	for _, installed := range worker.info.nodeInfo.Models.InstalledModels {
-		wm.workerInstalledOperator.Insert(context.Background(), &operator.WorkerInstalledInfo{
+		id, _ := strconv.Atoi(installed.ModelId)
+		iInfo := &operator.WorkerInstalledInfo{
 			WorkerId: worker.WorkerAccount().String(),
-			ModelId:  installed.ModelId,
-		})
+			ModelId:  id,
+		}
+		if len(worker.info.nodeInfo.Hardware.GPU) > 0 {
+			iInfo.GpuFree = worker.info.nodeInfo.Hardware.GPU[0].MemFree
+		}
+		wm.workerInstalledOperator.Insert(context.Background(), iInfo)
+	}
+	for _, running := range worker.info.nodeInfo.Models.RunningModels {
+		id, _ := strconv.Atoi(running.ModelId)
+		iInfo := &operator.WorkerRunningInfo{
+			WorkerId: worker.WorkerAccount().String(),
+			ModelId:  id,
+			ExecTime: int(running.ExecTime),
+		}
+		wm.workerRunningOperator.Insert(context.Background(), iInfo)
 	}
+
 	if err != nil {
 		log.WithError(err).Error("insert worker info failed")
 		return err