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} }