chore: rewrite MachineStatus to use QController

This rewrite should improve performance of the controller drastically.
Fixes: #281

Signed-off-by: Artem Chernyshev <artem.chernyshev@talos-systems.com>

cmd/integration-test/pkg/tests/talos.go

@@ -60,6 +60,10 @@ func clearConnectionRefused(ctx context.Context, t *testing.T, c *talosclient.Cl
 				return retry.ExpectedError(err)
 			}
 
+			if strings.Contains(err.Error(), "connection reset by peer") {
+				return retry.ExpectedError(err)
+			}
+
 			return err
 		}
 

internal/backend/runtime/omni/controllers/omni/internal/task/machine/helpers.go

@@ -8,6 +8,7 @@ package machine
 import (
 	"context"
 	"fmt"
+	"time"
 
 	"github.com/cosi-project/runtime/pkg/resource"
 	"github.com/cosi-project/runtime/pkg/resource/meta"
@@ -41,6 +42,9 @@ func forEachResource[T resource.Resource](
 
 // QueryRegisteredTypes gets all registered types from the meta namespace.
 func QueryRegisteredTypes(ctx context.Context, st state.State) (map[resource.Type]struct{}, error) {
+	ctx, cancel := context.WithTimeout(ctx, time.Second*5)
+	defer cancel()
+
 	// query all resources to start watching only resources that are defined for running version of talos
 	resources, err := safe.StateList[*meta.ResourceDefinition](ctx, st, resource.NewMetadata(meta.NamespaceName, meta.ResourceDefinitionType, "", resource.VersionUndefined))
 	if err != nil {

internal/backend/runtime/omni/controllers/omni/internal/task/machine/machine.go

@@ -192,12 +192,6 @@ func (spec CollectTaskSpec) RunTask(ctx context.Context, logger *zap.Logger, not
 
 	registeredTypes, err := QueryRegisteredTypes(ctx, c.COSI)
 	if err != nil {
-		// this is the first request to the Talos API
-		// if it fails we handle it and update the machine status with the request error
-		if !spec.sendInfo(ctx, Info{}, notifyCh, err) {
-			return nil
-		}
-
 		return err
 	}
 

internal/backend/runtime/omni/controllers/omni/internal/task/task.go

@@ -73,6 +73,8 @@ func (task *Task[T, S]) runWithRestarts(ctx context.Context) {
 
 		// finished without an error
 		if err == nil {
+			task.logger.Info("task finished")
+
 			return
 		}