split daemon request and runtime setup

2026-08-22 16:07:12 +08:00
parent b163b48ae3
commit a37ee9d09d
3 changed files with 141 additions and 99 deletions
@@ -130,84 +130,23 @@ func (s *Server) Serve(ctx context.Context) error {
 // 最终无论成功失败都会写入一条 ResponseResult 响应并记录日志。
 func (s *Server) handle(ctx context.Context, connection net.Conn) {
 	defer connection.Close()
-	_ = connection.SetReadDeadline(time.Now().Add(10 * time.Second))
-	request, err := decodeRequest(connection)
-	_ = connection.SetReadDeadline(time.Time{})
+	request, err := s.readRequest(connection)
 	if err != nil {
-		_ = s.writeResponse(connection, daemonapi.Response{Kind: daemonapi.ResponseResult, Error: err.Error()})
+		s.writeError(connection, err.Error())
 		return
 	}
 	if request.Service != "backend" {
-		_ = s.writeResponse(connection, daemonapi.Response{Kind: daemonapi.ResponseResult, Error: "service must be backend"})
+		s.writeError(connection, "service must be backend")
 		return
 	}
-	progressWritable := true
-	report := func(progress backendupdate.Progress) {
-		if !progressWritable {
-			return
-		}
-		err := s.writeResponse(connection, daemonapi.Response{
-			Kind:          daemonapi.ResponseProgress,
-			TransactionID: progress.TransactionID,
-			State:         string(progress.State),
-			Message:       progress.Message,
-		})
-		if err != nil {
-			progressWritable = false
-			s.logger.WarnContext(ctx, "write daemon update progress", "error", err)
-		}
+	progress := s.progressReporter(ctx, connection)
+	if request.Operation == daemonapi.OperationStatus || request.Operation == daemonapi.OperationDoctor || request.Operation == daemonapi.OperationReconcile {
+		s.handleDiagnosis(ctx, connection, request)
+		return
 	}
-	var record transaction.Transaction
-	var updateErr error
-	switch request.Operation {
-	case daemonapi.OperationUpdate:
-		switch request.InputType {
-		case daemonapi.InputTypeRepackZIP:
-			if request.File == "" || request.ImageReference != "" {
-				_ = s.writeResponse(connection, daemonapi.Response{Kind: daemonapi.ResponseResult, Error: "repack-zip requires file and does not accept imageReference"})
-				return
-			}
-			record, updateErr = s.updater.UpdateRepack(ctx, request.File, report)
-		case daemonapi.InputTypeNativeJAR:
-			if request.File == "" || request.ImageReference != "" {
-				_ = s.writeResponse(connection, daemonapi.Response{Kind: daemonapi.ResponseResult, Error: "native-jar requires file and does not accept imageReference"})
-				return
-			}
-			record, updateErr = s.updater.UpdateNativeJAR(ctx, request.File, report)
-		case daemonapi.InputTypeContainerImage:
-			if request.File != "" || request.ImageReference == "" {
-				_ = s.writeResponse(connection, daemonapi.Response{Kind: daemonapi.ResponseResult, Error: "container-image requires imageReference and does not accept file"})
-				return
-			}
-			record, updateErr = s.updater.UpdateContainerImage(ctx, request.ImageReference, request.StartLog, report)
-		default:
-			_ = s.writeResponse(connection, daemonapi.Response{Kind: daemonapi.ResponseResult, Error: "inputType must be repack-zip, native-jar, or container-image"})
-			return
-		}
-	case daemonapi.OperationRestart:
-		if request.InputType != "" || request.File != "" || request.ImageReference != "" {
-			_ = s.writeResponse(connection, daemonapi.Response{Kind: daemonapi.ResponseResult, Error: "restart does not accept inputType or file"})
-			return
-		}
-		record, updateErr = s.updater.Restart(ctx, report)
-	case daemonapi.OperationStatus, daemonapi.OperationDoctor:
-		if request.InputType != "" || request.File != "" || request.ImageReference != "" || request.Apply {
-			_ = s.writeResponse(connection, daemonapi.Response{Kind: daemonapi.ResponseResult, Error: request.Operation + " does not accept inputType, file, imageReference, or apply"})
-			return
-		}
-		diagnosis, diagnosisErr := s.diagnoser.Diagnose(ctx)
-		s.writeDiagnosis(ctx, connection, request.Operation, diagnosis, diagnosisErr)
-		return
-	case daemonapi.OperationReconcile:
-		if request.InputType != "" || request.File != "" || request.ImageReference != "" {
-			_ = s.writeResponse(connection, daemonapi.Response{Kind: daemonapi.ResponseResult, Error: "reconcile does not accept inputType, file, or imageReference"})
-			return
-		}
-		diagnosis, diagnosisErr := s.diagnoser.Reconcile(ctx, request.Apply)
-		s.writeDiagnosis(ctx, connection, request.Operation, diagnosis, diagnosisErr)
-		return
-	default:
-		_ = s.writeResponse(connection, daemonapi.Response{Kind: daemonapi.ResponseResult, Error: "operation must be update, restart, status, doctor, or reconcile"})
+	record, updateErr := s.handleBackendOperation(ctx, request, progress)
+	if updateErr != nil && record.ID == "" {
+		s.writeError(connection, updateErr.Error())
 		return
 	}
 	response := daemonapi.Response{Kind: daemonapi.ResponseResult, TransactionID: record.ID, State: string(record.State)}
@@ -222,6 +161,82 @@ func (s *Server) handle(ctx context.Context, connection net.Conn) {
 	}
 }

+func (s *Server) readRequest(connection net.Conn) (daemonapi.Request, error) {
+	_ = connection.SetReadDeadline(time.Now().Add(10 * time.Second))
+	request, err := decodeRequest(connection)
+	_ = connection.SetReadDeadline(time.Time{})
+	return request, err
+}
+
+func (s *Server) writeError(connection net.Conn, message string) {
+	_ = s.writeResponse(connection, daemonapi.Response{Kind: daemonapi.ResponseResult, Error: message})
+}
+
+func (s *Server) progressReporter(ctx context.Context, connection net.Conn) backendupdate.ProgressReporter {
+	progressWritable := true
+	return func(progress backendupdate.Progress) {
+		if !progressWritable {
+			return
+		}
+		err := s.writeResponse(connection, daemonapi.Response{Kind: daemonapi.ResponseProgress, TransactionID: progress.TransactionID, State: string(progress.State), Message: progress.Message})
+		if err != nil {
+			progressWritable = false
+			s.logger.WarnContext(ctx, "write daemon update progress", "error", err)
+		}
+	}
+}
+
+func (s *Server) handleBackendOperation(ctx context.Context, request daemonapi.Request, report backendupdate.ProgressReporter) (transaction.Transaction, error) {
+	switch request.Operation {
+	case daemonapi.OperationUpdate:
+		return s.handleUpdate(ctx, request, report)
+	case daemonapi.OperationRestart:
+		if request.InputType != "" || request.File != "" || request.ImageReference != "" {
+			return transaction.Transaction{}, errors.New("restart does not accept inputType or file")
+		}
+		return s.updater.Restart(ctx, report)
+	default:
+		return transaction.Transaction{}, errors.New("operation must be update, restart, status, doctor, or reconcile")
+	}
+}
+
+func (s *Server) handleUpdate(ctx context.Context, request daemonapi.Request, report backendupdate.ProgressReporter) (transaction.Transaction, error) {
+	switch request.InputType {
+	case daemonapi.InputTypeRepackZIP:
+		if request.File == "" || request.ImageReference != "" {
+			return transaction.Transaction{}, errors.New("repack-zip requires file and does not accept imageReference")
+		}
+		return s.updater.UpdateRepack(ctx, request.File, report)
+	case daemonapi.InputTypeNativeJAR:
+		if request.File == "" || request.ImageReference != "" {
+			return transaction.Transaction{}, errors.New("native-jar requires file and does not accept imageReference")
+		}
+		return s.updater.UpdateNativeJAR(ctx, request.File, report)
+	case daemonapi.InputTypeContainerImage:
+		if request.File != "" || request.ImageReference == "" {
+			return transaction.Transaction{}, errors.New("container-image requires imageReference and does not accept file")
+		}
+		return s.updater.UpdateContainerImage(ctx, request.ImageReference, request.StartLog, report)
+	default:
+		return transaction.Transaction{}, errors.New("inputType must be repack-zip, native-jar, or container-image")
+	}
+}
+
+func (s *Server) handleDiagnosis(ctx context.Context, connection net.Conn, request daemonapi.Request) {
+	if request.InputType != "" || request.File != "" || request.ImageReference != "" || (request.Operation != daemonapi.OperationReconcile && request.Apply) {
+		s.writeError(connection, request.Operation+" does not accept inputType, file, imageReference, or apply")
+		return
+	}
+	var diagnosis daemonapi.Diagnosis
+	var err error
+	if request.Operation == daemonapi.OperationReconcile {
+		diagnosis, err = s.diagnoser.Reconcile(ctx, request.Apply)
+	} else {
+		diagnosis, err = s.diagnoser.Diagnose(ctx)
+	}
+	s.writeDiagnosis(ctx, connection, request.Operation, diagnosis, err)
+}
+
 // writeResponse 把一条响应以换行分隔的 JSON 编码写入连接，失败时返回底层写入错误。
 func (s *Server) writeResponse(connection net.Conn, response daemonapi.Response) error {
 	return json.NewEncoder(connection).Encode(response)