smegmesh/pkg/sync/syncererror.go

package sync

import (
	"github.com/tim-beatham/wgmesh/pkg/conn"
	logging "github.com/tim-beatham/wgmesh/pkg/log"
	"github.com/tim-beatham/wgmesh/pkg/mesh"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
)

// SyncErrorHandler: Handles errors when attempting to sync
type SyncErrorHandler interface {
	Handle(meshId string, endpoint string, err error) bool
}

// SyncErrorHandlerImpl Is an implementation of the SyncErrorHandler
type SyncErrorHandlerImpl struct {
	meshManager mesh.MeshManager
	connManager conn.ConnectionManager
}

func (s *SyncErrorHandlerImpl) handleFailed(meshId string, nodeId string) bool {
	mesh := s.meshManager.GetMesh(meshId)
	mesh.Mark(nodeId)
	node, err := mesh.GetNode(nodeId)

	if err != nil {
		s.connManager.RemoveConnection(node.GetHostEndpoint())
	}
	return true
}

func (s *SyncErrorHandlerImpl) handleDeadlineExceeded(meshId string, nodeId string) bool {
	mesh := s.meshManager.GetMesh(meshId)

	if mesh == nil {
		return true
	}

	node, err := mesh.GetNode(nodeId)

	if err != nil {
		return false
	}

	s.connManager.RemoveConnection(node.GetHostEndpoint())
	return true
}

func (s *SyncErrorHandlerImpl) Handle(meshId string, nodeId string, err error) bool {
	errStatus, _ := status.FromError(err)

	logging.Log.WriteInfof("Handled gRPC error: %s", errStatus.Message())

	switch errStatus.Code() {
	case codes.Unavailable, codes.Unknown, codes.Internal, codes.NotFound:
		return s.handleFailed(meshId, nodeId)
	case codes.DeadlineExceeded:
		return s.handleDeadlineExceeded(meshId, nodeId)
	}

	return false
}

func NewSyncErrorHandler(m mesh.MeshManager, conn conn.ConnectionManager) SyncErrorHandler {
	return &SyncErrorHandlerImpl{meshManager: m, connManager: conn}
}