Passed
Push — master ( dc6eee...fba62f )
by Vinicius
06:34 queued 04:27
created
1
"""Main module of kytos/topology Kytos Network Application.
2
3
Manage the network topology
4
"""
5
# pylint: disable=wrong-import-order
6 1
import pathlib
7 1
import time
8 1
from collections import defaultdict
9 1
from datetime import timezone
10 1
from threading import Lock
11 1
from typing import List, Optional
12
13 1
from kytos.core import KytosEvent, KytosNApp, log, rest
14 1
from kytos.core.common import EntityStatus
15 1
from kytos.core.exceptions import (KytosInvalidTagRanges,
16
                                   KytosLinkCreationError, KytosTagError)
17 1
from kytos.core.helpers import listen_to, load_spec, now, validate_openapi
18 1
from kytos.core.interface import Interface
19 1
from kytos.core.link import Link
20 1
from kytos.core.rest_api import (HTTPException, JSONResponse, Request,
21
                                 content_type_json_or_415, get_json_or_400)
22 1
from kytos.core.switch import Switch
23 1
from kytos.core.tag_ranges import get_tag_ranges
24 1
from napps.kytos.topology import settings
25
26 1
from .controllers import TopoController
27 1
from .exceptions import RestoreError
28 1
from .models import Topology
29
30 1
DEFAULT_LINK_UP_TIMER = 10
31
32
33 1
class Main(KytosNApp):  # pylint: disable=too-many-public-methods
34
    """Main class of kytos/topology NApp.
35
36
    This class is the entry point for this napp.
37
    """
38
39 1
    spec = load_spec(pathlib.Path(__file__).parent / "openapi.yml")
40
41 1
    def setup(self):
42
        """Initialize the NApp's links list."""
43 1
        self.links: dict[str, Link] = {}
44 1
        self.intf_available_tags = {}
45 1
        self.link_up_timer = getattr(settings, 'LINK_UP_TIMER',
46
                                     DEFAULT_LINK_UP_TIMER)
47
48 1
        self._links_lock = Lock()
49 1
        self._links_notify_lock = defaultdict(Lock)
50
        # to keep track of potential unorded scheduled interface events
51 1
        self._intfs_lock = defaultdict(Lock)
52 1
        self._intfs_updated_at = {}
53 1
        self._intfs_tags_updated_at = {}
54 1
        self.link_up = set()
55 1
        self.link_status_lock = Lock()
56 1
        self.topo_controller = self.get_topo_controller()
57 1
        Link.register_status_func(f"{self.napp_id}_link_up_timer",
58
                                  self.link_status_hook_link_up_timer)
59 1
        self.topo_controller.bootstrap_indexes()
60 1
        self.load_topology()
61
62 1
    @staticmethod
63 1
    def get_topo_controller() -> TopoController:
64
        """Get TopoController."""
65
        return TopoController()
66
67 1
    def execute(self):
68
        """Execute once when the napp is running."""
69
        pass
70
71 1
    def shutdown(self):
72
        """Do nothing."""
73
        log.info('NApp kytos/topology shutting down.')
74
75 1
    def _get_metadata(self, request: Request) -> dict:
76
        """Return a JSON with metadata."""
77 1
        content_type_json_or_415(request)
78 1
        metadata = get_json_or_400(request, self.controller.loop)
79 1
        if not isinstance(metadata, dict):
80 1
            raise HTTPException(400, "Invalid metadata value: {metadata}")
81 1
        return metadata
82
83 1
    def _get_link_or_create(self, endpoint_a, endpoint_b):
84
        """Get an existing link or create a new one.
85
86
        Returns:
87
            Tuple(Link, bool): Link and a boolean whether it has been created.
88
        """
89 1
        new_link = Link(endpoint_a, endpoint_b)
90
91 1
        for link in self.links.values():
92 1
            if new_link == link:
93 1
                return (link, False)
94
95 1
        self.links[new_link.id] = new_link
96 1
        return (new_link, True)
97
98 1
    def _get_switches_dict(self):
99
        """Return a dictionary with the known switches."""
100 1
        switches = {'switches': {}}
101 1
        for idx, switch in enumerate(self.controller.switches.copy().values()):
102 1
            switch_data = switch.as_dict()
103 1
            if not all(key in switch_data['metadata']
104
                       for key in ('lat', 'lng')):
105
                # Switches are initialized somewhere in the ocean
106
                switch_data['metadata']['lat'] = str(0.0)
107
                switch_data['metadata']['lng'] = str(-30.0+idx*10.0)
108 1
            switches['switches'][switch.id] = switch_data
109 1
        return switches
110
111 1
    def _get_links_dict(self):
112
        """Return a dictionary with the known links."""
113 1
        return {'links': {link.id: link.as_dict() for link in
114
                          self.links.copy().values()}}
115
116 1
    def _get_topology_dict(self):
117
        """Return a dictionary with the known topology."""
118 1
        return {'topology': {**self._get_switches_dict(),
119
                             **self._get_links_dict()}}
120
121 1
    def _get_topology(self):
122
        """Return an object representing the topology."""
123 1
        return Topology(self.controller.switches.copy(), self.links.copy())
124
125 1
    def _get_link_from_interface(self, interface):
126
        """Return the link of the interface, or None if it does not exist."""
127 1
        for link in list(self.links.values()):
128 1
            if interface in (link.endpoint_a, link.endpoint_b):
129 1
                return link
130 1
        return None
131
132 1
    def _load_link(self, link_att):
133 1
        endpoint_a = link_att['endpoint_a']['id']
134 1
        endpoint_b = link_att['endpoint_b']['id']
135 1
        link_str = link_att['id']
136 1
        log.info(f"Loading link: {link_str}")
137 1
        interface_a = self.controller.get_interface_by_id(endpoint_a)
138 1
        interface_b = self.controller.get_interface_by_id(endpoint_b)
139
140 1
        error = f"Fail to load endpoints for link {link_str}. "
141 1
        if not interface_a:
142 1
            raise RestoreError(f"{error}, endpoint_a {endpoint_a} not found")
143 1
        if not interface_b:
144
            raise RestoreError(f"{error}, endpoint_b {endpoint_b} not found")
145
146 1
        with self._links_lock:
147 1
            link, _ = self._get_link_or_create(interface_a, interface_b)
148
149 1
        if link_att['enabled']:
150 1
            link.enable()
151
        else:
152 1
            link.disable()
153
154
        # These ones are just runtime active southbound protocol data
155
        # It won't be stored in the future, only kept in the runtime.
156
        # Also network operators can follow logs to track this state changes
157 1
        for key in (
158
            "last_status_is_active", "last_status_change", "notified_up_at"
159
        ):
160 1
            link_att["metadata"].pop(key, None)
161
162 1
        link.extend_metadata(link_att["metadata"])
163 1
        interface_a.update_link(link)
164 1
        interface_b.update_link(link)
165 1
        interface_a.nni = True
166 1
        interface_b.nni = True
167
168 1
    def _load_switch(self, switch_id, switch_att):
169 1
        log.info(f'Loading switch dpid: {switch_id}')
170 1
        switch = self.controller.get_switch_or_create(switch_id)
171 1
        if switch_att['enabled']:
172 1
            switch.enable()
173
        else:
174 1
            switch.disable()
175 1
        switch.description['manufacturer'] = switch_att.get('manufacturer', '')
176 1
        switch.description['hardware'] = switch_att.get('hardware', '')
177 1
        switch.description['software'] = switch_att.get('software')
178 1
        switch.description['serial'] = switch_att.get('serial', '')
179 1
        switch.description['data_path'] = switch_att.get('data_path', '')
180 1
        switch.extend_metadata(switch_att["metadata"])
181
182 1
        for iface_id, iface_att in switch_att.get('interfaces', {}).items():
183 1
            log.info(f'Loading interface iface_id={iface_id}')
184 1
            interface = switch.update_or_create_interface(
185
                            port_no=iface_att['port_number'],
186
                            name=iface_att['name'],
187
                            address=iface_att.get('mac', None),
188
                            speed=iface_att.get('speed', None))
189 1
            if iface_att['enabled']:
190 1
                interface.enable()
191
            else:
192 1
                interface.disable()
193 1
            interface.lldp = iface_att['lldp']
194 1
            interface.extend_metadata(iface_att["metadata"])
195 1
            interface.deactivate()
196 1
            name = 'kytos/topology.port.created'
197 1
            event = KytosEvent(name=name, content={
198
                                              'switch': switch_id,
199
                                              'port': interface.port_number,
200
                                              'port_description': {
201
                                                  'alias': interface.name,
202
                                                  'mac': interface.address,
203
                                                  'state': interface.state
204
                                                  }
205
                                              })
206 1
            self.controller.buffers.app.put(event, timeout=1)
207
208 1
        intf_ids = [v["id"] for v in switch_att.get("interfaces", {}).values()]
209 1
        intf_details = self.topo_controller.get_interfaces_details(intf_ids)
210 1
        with self._links_lock:
211 1
            self.load_interfaces_tags_values(switch, intf_details)
212
213
    # pylint: disable=attribute-defined-outside-init
214 1
    def load_topology(self):
215
        """Load network topology from DB."""
216 1
        topology = self.topo_controller.get_topology()
217 1
        switches = topology["topology"]["switches"]
218 1
        links = topology["topology"]["links"]
219
220 1
        failed_switches = {}
221 1
        log.debug(f"_load_network_status switches={switches}")
222 1
        for switch_id, switch_att in switches.items():
223 1
            try:
224 1
                self._load_switch(switch_id, switch_att)
225 1
            except (KeyError, AttributeError, TypeError) as err:
226 1
                failed_switches[switch_id] = err
227 1
                log.error(f'Error loading switch: {err}')
228
229 1
        failed_links = {}
230 1
        log.debug(f"_load_network_status links={links}")
231 1
        for link_id, link_att in links.items():
232 1
            try:
233 1
                self._load_link(link_att)
234 1
            except (KeyError, AttributeError, TypeError) as err:
235 1
                failed_links[link_id] = err
236 1
                log.error(f'Error loading link {link_id}: {err}')
237
238 1
        name = 'kytos/topology.topology_loaded'
239 1
        event = KytosEvent(
240
            name=name,
241
            content={
242
                'topology': self._get_topology(),
243
                'failed_switches': failed_switches,
244
                'failed_links': failed_links
245
            })
246 1
        self.controller.buffers.app.put(event, timeout=1)
247
248 1
    @rest('v3/')
249 1
    def get_topology(self, _request: Request) -> JSONResponse:
250
        """Return the latest known topology.
251
252
        This topology is updated when there are network events.
253
        """
254 1
        return JSONResponse(self._get_topology_dict())
255
256
    # Switch related methods
257 1
    @rest('v3/switches')
258 1
    def get_switches(self, _request: Request) -> JSONResponse:
259
        """Return a json with all the switches in the topology."""
260
        return JSONResponse(self._get_switches_dict())
261
262 1
    @rest('v3/switches/{dpid}/enable', methods=['POST'])
263 1
    def enable_switch(self, request: Request) -> JSONResponse:
264
        """Administratively enable a switch in the topology."""
265 1
        dpid = request.path_params["dpid"]
266 1
        try:
267 1
            switch = self.controller.switches[dpid]
268 1
            self.topo_controller.enable_switch(dpid)
269 1
            switch.enable()
270 1
        except KeyError:
271 1
            raise HTTPException(404, detail="Switch not found")
272
273 1
        self.notify_switch_enabled(dpid)
274 1
        self.notify_topology_update()
275 1
        self.notify_switch_links_status(switch, "link enabled")
276 1
        return JSONResponse("Operation successful", status_code=201)
277
278 1
    @rest('v3/switches/{dpid}/disable', methods=['POST'])
279 1
    def disable_switch(self, request: Request) -> JSONResponse:
280
        """Administratively disable a switch in the topology."""
281 1
        dpid = request.path_params["dpid"]
282 1
        try:
283 1
            switch = self.controller.switches[dpid]
284 1
            self.topo_controller.disable_switch(dpid)
285 1
            switch.disable()
286 1
        except KeyError:
287 1
            raise HTTPException(404, detail="Switch not found")
288
289 1
        self.notify_switch_disabled(dpid)
290 1
        self.notify_topology_update()
291 1
        self.notify_switch_links_status(switch, "link disabled")
292 1
        return JSONResponse("Operation successful", status_code=201)
293
294 1
    @rest('v3/switches/{dpid}/metadata')
295 1
    def get_switch_metadata(self, request: Request) -> JSONResponse:
296
        """Get metadata from a switch."""
297 1
        dpid = request.path_params["dpid"]
298 1
        try:
299 1
            metadata = self.controller.switches[dpid].metadata
300 1
            return JSONResponse({"metadata": metadata})
301 1
        except KeyError:
302 1
            raise HTTPException(404, detail="Switch not found")
303
304 1
    @rest('v3/switches/{dpid}/metadata', methods=['POST'])
305 1
    def add_switch_metadata(self, request: Request) -> JSONResponse:
306
        """Add metadata to a switch."""
307 1
        dpid = request.path_params["dpid"]
308 1
        metadata = self._get_metadata(request)
309 1
        try:
310 1
            switch = self.controller.switches[dpid]
311 1
        except KeyError:
312 1
            raise HTTPException(404, detail="Switch not found")
313
314 1
        self.topo_controller.add_switch_metadata(dpid, metadata)
315 1
        switch.extend_metadata(metadata)
316 1
        self.notify_metadata_changes(switch, 'added')
317 1
        return JSONResponse("Operation successful", status_code=201)
318
319 1
    @rest('v3/switches/{dpid}/metadata/{key}', methods=['DELETE'])
320 1
    def delete_switch_metadata(self, request: Request) -> JSONResponse:
321
        """Delete metadata from a switch."""
322 1
        dpid = request.path_params["dpid"]
323 1
        key = request.path_params["key"]
324 1
        try:
325 1
            switch = self.controller.switches[dpid]
326 1
        except KeyError:
327 1
            raise HTTPException(404, detail="Switch not found")
328
329 1
        try:
330 1
            _ = switch.metadata[key]
331
        except KeyError:
332
            raise HTTPException(404, "Metadata not found")
333
334 1
        self.topo_controller.delete_switch_metadata_key(dpid, key)
335 1
        switch.remove_metadata(key)
336 1
        self.notify_metadata_changes(switch, 'removed')
337 1
        return JSONResponse("Operation successful")
338
339
    # Interface related methods
340 1
    @rest('v3/interfaces')
341 1
    def get_interfaces(self, _request: Request) -> JSONResponse:
342
        """Return a json with all the interfaces in the topology."""
343 1
        interfaces = {}
344 1
        switches = self._get_switches_dict()
345 1
        for switch in switches['switches'].values():
346 1
            for interface_id, interface in switch['interfaces'].items():
347 1
                interfaces[interface_id] = interface
348
349 1
        return JSONResponse({'interfaces': interfaces})
350
351 1 View Code Duplication
    @rest('v3/interfaces/switch/{dpid}/enable', methods=['POST'])
352 1
    @rest('v3/interfaces/{interface_enable_id}/enable', methods=['POST'])
353 1
    def enable_interface(self, request: Request) -> JSONResponse:
354
        """Administratively enable interfaces in the topology."""
355 1
        interface_enable_id = request.path_params.get("interface_enable_id")
356 1
        dpid = request.path_params.get("dpid")
357 1
        if dpid is None:
358 1
            dpid = ":".join(interface_enable_id.split(":")[:-1])
359 1
        try:
360 1
            switch = self.controller.switches[dpid]
361
        except KeyError:
362
            raise HTTPException(404, detail="Switch not found")
363
364 1
        if interface_enable_id:
365 1
            interface_number = int(interface_enable_id.split(":")[-1])
366
367 1
            try:
368 1
                interface = switch.interfaces[interface_number]
369 1
                self.topo_controller.enable_interface(interface.id)
370 1
                interface.enable()
371 1
                self.notify_interface_link_status(interface, "link enabled")
372 1
            except KeyError:
373 1
                msg = f"Switch {dpid} interface {interface_number} not found"
374 1
                raise HTTPException(404, detail=msg)
375
        else:
376 1
            for interface in switch.interfaces.copy().values():
377 1
                interface.enable()
378 1
                self.notify_interface_link_status(interface, "link enabled")
379 1
            self.topo_controller.upsert_switch(switch.id, switch.as_dict())
380 1
        self.notify_topology_update()
381 1
        return JSONResponse("Operation successful")
382
383 1 View Code Duplication
    @rest('v3/interfaces/switch/{dpid}/disable', methods=['POST'])
384 1
    @rest('v3/interfaces/{interface_disable_id}/disable', methods=['POST'])
385 1
    def disable_interface(self, request: Request) -> JSONResponse:
386
        """Administratively disable interfaces in the topology."""
387 1
        interface_disable_id = request.path_params.get("interface_disable_id")
388 1
        dpid = request.path_params.get("dpid")
389 1
        if dpid is None:
390 1
            dpid = ":".join(interface_disable_id.split(":")[:-1])
391 1
        try:
392 1
            switch = self.controller.switches[dpid]
393 1
        except KeyError:
394 1
            raise HTTPException(404, detail="Switch not found")
395
396 1
        if interface_disable_id:
397 1
            interface_number = int(interface_disable_id.split(":")[-1])
398
399 1
            try:
400 1
                interface = switch.interfaces[interface_number]
401 1
                self.topo_controller.disable_interface(interface.id)
402 1
                interface.disable()
403 1
                self.notify_interface_link_status(interface, "link disabled")
404 1
            except KeyError:
405 1
                msg = f"Switch {dpid} interface {interface_number} not found"
406 1
                raise HTTPException(404, detail=msg)
407
        else:
408 1
            for interface in switch.interfaces.copy().values():
409 1
                interface.disable()
410 1
                self.notify_interface_link_status(interface, "link disabled")
411 1
            self.topo_controller.upsert_switch(switch.id, switch.as_dict())
412 1
        self.notify_topology_update()
413 1
        return JSONResponse("Operation successful")
414
415 1
    @rest('v3/interfaces/{interface_id}/metadata')
416 1
    def get_interface_metadata(self, request: Request) -> JSONResponse:
417
        """Get metadata from an interface."""
418 1
        interface_id = request.path_params["interface_id"]
419 1
        switch_id = ":".join(interface_id.split(":")[:-1])
420 1
        interface_number = int(interface_id.split(":")[-1])
421 1
        try:
422 1
            switch = self.controller.switches[switch_id]
423 1
        except KeyError:
424 1
            raise HTTPException(404, detail="Switch not found")
425
426 1
        try:
427 1
            interface = switch.interfaces[interface_number]
428 1
        except KeyError:
429 1
            raise HTTPException(404, detail="Interface not found")
430
431 1
        return JSONResponse({"metadata": interface.metadata})
432
433 1
    @rest('v3/interfaces/{interface_id}/metadata', methods=['POST'])
434 1
    def add_interface_metadata(self, request: Request) -> JSONResponse:
435
        """Add metadata to an interface."""
436 1
        interface_id = request.path_params["interface_id"]
437 1
        metadata = self._get_metadata(request)
438 1
        switch_id = ":".join(interface_id.split(":")[:-1])
439 1
        interface_number = int(interface_id.split(":")[-1])
440 1
        try:
441 1
            switch = self.controller.switches[switch_id]
442 1
        except KeyError:
443 1
            raise HTTPException(404, detail="Switch not found")
444
445 1
        try:
446 1
            interface = switch.interfaces[interface_number]
447 1
            self.topo_controller.add_interface_metadata(interface.id, metadata)
448 1
        except KeyError:
449 1
            raise HTTPException(404, detail="Interface not found")
450
451 1
        interface.extend_metadata(metadata)
452 1
        self.notify_metadata_changes(interface, 'added')
453 1
        return JSONResponse("Operation successful", status_code=201)
454
455 1
    @rest('v3/interfaces/{interface_id}/metadata/{key}', methods=['DELETE'])
456 1
    def delete_interface_metadata(self, request: Request) -> JSONResponse:
457
        """Delete metadata from an interface."""
458 1
        interface_id = request.path_params["interface_id"]
459 1
        key = request.path_params["key"]
460 1
        switch_id = ":".join(interface_id.split(":")[:-1])
461 1
        try:
462 1
            interface_number = int(interface_id.split(":")[-1])
463
        except ValueError:
464
            detail = f"Invalid interface_id {interface_id}"
465
            raise HTTPException(400, detail=detail)
466
467 1
        try:
468 1
            switch = self.controller.switches[switch_id]
469 1
        except KeyError:
470 1
            raise HTTPException(404, detail="Switch not found")
471
472 1
        try:
473 1
            interface = switch.interfaces[interface_number]
474 1
        except KeyError:
475 1
            raise HTTPException(404, detail="Interface not found")
476
477 1
        try:
478 1
            _ = interface.metadata[key]
479 1
        except KeyError:
480 1
            raise HTTPException(404, detail="Metadata not found")
481
482 1
        self.topo_controller.delete_interface_metadata_key(interface.id, key)
483 1
        interface.remove_metadata(key)
484 1
        self.notify_metadata_changes(interface, 'removed')
485 1
        return JSONResponse("Operation successful")
486
487 1
    @rest('v3/interfaces/{interface_id}/tag_ranges', methods=['POST'])
488 1
    @validate_openapi(spec)
489 1
    def set_tag_range(self, request: Request) -> JSONResponse:
490
        """Set tag range"""
491 1
        content_type_json_or_415(request)
492 1
        content = get_json_or_400(request, self.controller.loop)
493 1
        tag_type = content.get("tag_type")
494 1
        try:
495 1
            ranges = get_tag_ranges(content["tag_ranges"])
496
        except KytosInvalidTagRanges as err:
497
            raise HTTPException(400, detail=str(err))
498 1
        interface_id = request.path_params["interface_id"]
499 1
        interface = self.controller.get_interface_by_id(interface_id)
500 1
        if not interface:
501 1
            raise HTTPException(404, detail="Interface not found")
502 1
        try:
503 1
            interface.set_tag_ranges(ranges, tag_type)
504 1
            self.handle_on_interface_tags(interface)
505 1
        except KytosTagError as err:
506 1
            raise HTTPException(400, detail=str(err))
507 1
        return JSONResponse("Operation Successful", status_code=200)
508
509 1
    @rest('v3/interfaces/{interface_id}/tag_ranges', methods=['DELETE'])
510 1
    @validate_openapi(spec)
511 1
    def delete_tag_range(self, request: Request) -> JSONResponse:
512
        """Set tag_range from tag_type to default value [1, 4095]"""
513 1
        interface_id = request.path_params["interface_id"]
514 1
        params = request.query_params
515 1
        tag_type = params.get("tag_type", 'vlan')
516 1
        interface = self.controller.get_interface_by_id(interface_id)
517 1
        if not interface:
518 1
            raise HTTPException(404, detail="Interface not found")
519 1
        try:
520 1
            interface.remove_tag_ranges(tag_type)
521 1
            self.handle_on_interface_tags(interface)
522 1
        except KytosTagError as err:
523 1
            raise HTTPException(400, detail=str(err))
524 1
        return JSONResponse("Operation Successful", status_code=200)
525
526 1
    @rest('v3/interfaces/{interface_id}/special_tags', methods=['POST'])
527 1
    @validate_openapi(spec)
528 1
    def set_special_tags(self, request: Request) -> JSONResponse:
529
        """Set special_tags"""
530 1
        content_type_json_or_415(request)
531 1
        content = get_json_or_400(request, self.controller.loop)
532 1
        tag_type = content.get("tag_type")
533 1
        special_tags = content["special_tags"]
534 1
        interface_id = request.path_params["interface_id"]
535 1
        interface = self.controller.get_interface_by_id(interface_id)
536 1
        if not interface:
537 1
            raise HTTPException(404, detail="Interface not found")
538 1
        try:
539 1
            interface.set_special_tags(tag_type, special_tags)
540 1
            self.handle_on_interface_tags(interface)
541 1
        except KytosTagError as err:
542 1
            raise HTTPException(400, detail=str(err))
543 1
        return JSONResponse("Operation Successful", status_code=200)
544
545 1
    @rest('v3/interfaces/tag_ranges', methods=['GET'])
546 1
    @validate_openapi(spec)
547 1
    def get_all_tag_ranges(self, _: Request) -> JSONResponse:
548
        """Get all tag_ranges, available_tags, special_tags
549
         and special_available_tags from interfaces"""
550 1
        result = {}
551 1
        for switch in self.controller.switches.copy().values():
552 1
            for interface in switch.interfaces.copy().values():
553 1
                result[interface.id] = {
554
                    "available_tags": interface.available_tags,
555
                    "tag_ranges": interface.tag_ranges,
556
                    "special_tags": interface.special_tags,
557
                    "special_available_tags": interface.special_available_tags
558
                }
559 1
        return JSONResponse(result, status_code=200)
560
561 1
    @rest('v3/interfaces/{interface_id}/tag_ranges', methods=['GET'])
562 1
    @validate_openapi(spec)
563 1
    def get_tag_ranges_by_intf(self, request: Request) -> JSONResponse:
564
        """Get tag_ranges, available_tags, special_tags
565
         and special_available_tags from an interface"""
566 1
        interface_id = request.path_params["interface_id"]
567 1
        interface = self.controller.get_interface_by_id(interface_id)
568 1
        if not interface:
569 1
            raise HTTPException(404, detail="Interface not found")
570 1
        result = {
571
            interface_id: {
572
                "available_tags": interface.available_tags,
573
                "tag_ranges": interface.tag_ranges,
574
                "special_tags": interface.special_tags,
575
                "special_available_tags": interface.special_available_tags
576
            }
577
        }
578 1
        return JSONResponse(result, status_code=200)
579
580
    # Link related methods
581 1
    @rest('v3/links')
582 1
    def get_links(self, _request: Request) -> JSONResponse:
583
        """Return a json with all the links in the topology.
584
585
        Links are connections between interfaces.
586
        """
587
        return JSONResponse(self._get_links_dict())
588
589 1
    @rest('v3/links/{link_id}/enable', methods=['POST'])
590 1
    def enable_link(self, request: Request) -> JSONResponse:
591
        """Administratively enable a link in the topology."""
592 1
        link_id = request.path_params["link_id"]
593 1
        try:
594 1
            with self._links_lock:
595 1
                link = self.links[link_id]
596 1
                self.topo_controller.enable_link(link_id)
597 1
                link.enable()
598 1
        except KeyError:
599 1
            raise HTTPException(404, detail="Link not found")
600 1
        self.notify_link_status_change(
601
            self.links[link_id],
602
            reason='link enabled'
603
        )
604 1
        self.notify_topology_update()
605 1
        return JSONResponse("Operation successful", status_code=201)
606
607 1
    @rest('v3/links/{link_id}/disable', methods=['POST'])
608 1
    def disable_link(self, request: Request) -> JSONResponse:
609
        """Administratively disable a link in the topology."""
610 1
        link_id = request.path_params["link_id"]
611 1
        try:
612 1
            with self._links_lock:
613 1
                link = self.links[link_id]
614 1
                self.topo_controller.disable_link(link_id)
615 1
                link.disable()
616 1
        except KeyError:
617 1
            raise HTTPException(404, detail="Link not found")
618 1
        self.notify_link_status_change(
619
            self.links[link_id],
620
            reason='link disabled'
621
        )
622 1
        self.notify_topology_update()
623 1
        return JSONResponse("Operation successful", status_code=201)
624
625 1
    @rest('v3/links/{link_id}/metadata')
626 1
    def get_link_metadata(self, request: Request) -> JSONResponse:
627
        """Get metadata from a link."""
628 1
        link_id = request.path_params["link_id"]
629 1
        try:
630 1
            return JSONResponse({"metadata": self.links[link_id].metadata})
631 1
        except KeyError:
632 1
            raise HTTPException(404, detail="Link not found")
633
634 1
    @rest('v3/links/{link_id}/metadata', methods=['POST'])
635 1
    def add_link_metadata(self, request: Request) -> JSONResponse:
636
        """Add metadata to a link."""
637 1
        link_id = request.path_params["link_id"]
638 1
        metadata = self._get_metadata(request)
639 1
        try:
640 1
            link = self.links[link_id]
641 1
        except KeyError:
642 1
            raise HTTPException(404, detail="Link not found")
643
644 1
        self.topo_controller.add_link_metadata(link_id, metadata)
645 1
        link.extend_metadata(metadata)
646 1
        self.notify_metadata_changes(link, 'added')
647 1
        self.notify_topology_update()
648 1
        return JSONResponse("Operation successful", status_code=201)
649
650 1
    @rest('v3/links/{link_id}/metadata/{key}', methods=['DELETE'])
651 1
    def delete_link_metadata(self, request: Request) -> JSONResponse:
652
        """Delete metadata from a link."""
653 1
        link_id = request.path_params["link_id"]
654 1
        key = request.path_params["key"]
655 1
        try:
656 1
            link = self.links[link_id]
657 1
        except KeyError:
658 1
            raise HTTPException(404, detail="Link not found")
659
660 1
        try:
661 1
            _ = link.metadata[key]
662 1
        except KeyError:
663 1
            raise HTTPException(404, detail="Metadata not found")
664
665 1
        self.topo_controller.delete_link_metadata_key(link.id, key)
666 1
        link.remove_metadata(key)
667 1
        self.notify_metadata_changes(link, 'removed')
668 1
        self.notify_topology_update()
669 1
        return JSONResponse("Operation successful")
670
671 1
    @rest('v3/links/{link_id}', methods=['DELETE'])
672 1
    def delete_link(self, request: Request) -> JSONResponse:
673
        """Delete a disabled link from topology.
674
         It won't work for link with other statuses.
675
        """
676 1
        link_id = request.path_params["link_id"]
677 1
        try:
678 1
            with self._links_lock:
679 1
                link = self.links[link_id]
680 1
                if link.status != EntityStatus.DISABLED:
681 1
                    raise HTTPException(409, detail="Link is not disabled.")
682 1
                if link.endpoint_a.link and link == link.endpoint_a.link:
683 1
                    switch = link.endpoint_a.switch
684 1
                    link.endpoint_a.link = None
685 1
                    link.endpoint_a.nni = False
686 1
                    self.topo_controller.upsert_switch(
687
                        switch.id, switch.as_dict()
688
                    )
689 1
                if link.endpoint_b.link and link == link.endpoint_b.link:
690 1
                    switch = link.endpoint_b.switch
691 1
                    link.endpoint_b.link = None
692 1
                    link.endpoint_b.nni = False
693 1
                    self.topo_controller.upsert_switch(
694
                        switch.id, switch.as_dict()
695
                    )
696 1
                self.topo_controller.delete_link(link_id)
697 1
                link = self.links.pop(link_id)
698 1
        except KeyError:
699 1
            raise HTTPException(404, detail="Link not found.")
700 1
        self.notify_topology_update()
701 1
        name = 'kytos/topology.link.deleted'
702 1
        event = KytosEvent(name=name, content={'link': link})
703 1
        self.controller.buffers.app.put(event)
704 1
        return JSONResponse("Operation successful")
705
706 1
    @listen_to("kytos/.*.liveness.(up|down)")
707 1
    def on_link_liveness_status(self, event) -> None:
708
        """Handle link liveness up|down status event."""
709
        link = Link(event.content["interface_a"], event.content["interface_b"])
710
        try:
711
            link = self.links[link.id]
712
        except KeyError:
713
            log.error(f"Link id {link.id} not found, {link}")
714
            return
715
        liveness_status = event.name.split(".")[-1]
716
        self.handle_link_liveness_status(self.links[link.id], liveness_status)
717
718 1
    def handle_link_liveness_status(self, link, liveness_status) -> None:
719
        """Handle link liveness."""
720 1
        metadata = {"liveness_status": liveness_status}
721 1
        log.info(f"Link liveness {liveness_status}: {link}")
722 1
        self.topo_controller.add_link_metadata(link.id, metadata)
723 1
        link.extend_metadata(metadata)
724 1
        self.notify_topology_update()
725 1
        if link.status == EntityStatus.UP and liveness_status == "up":
726 1
            self.notify_link_status_change(link, reason="liveness_up")
727 1
        if link.status == EntityStatus.DOWN and liveness_status == "down":
728 1
            self.notify_link_status_change(link, reason="liveness_down")
729
730 1
    @listen_to("kytos/.*.liveness.disabled")
731 1
    def on_link_liveness_disabled(self, event) -> None:
732
        """Handle link liveness disabled event."""
733
        interfaces = event.content["interfaces"]
734
        self.handle_link_liveness_disabled(interfaces)
735
736 1
    def get_links_from_interfaces(self, interfaces) -> dict:
737
        """Get links from interfaces."""
738 1
        links_found = {}
739 1
        with self._links_lock:
740 1
            for interface in interfaces:
741 1
                for link in self.links.values():
742 1
                    if any((
743
                        interface.id == link.endpoint_a.id,
744
                        interface.id == link.endpoint_b.id,
745
                    )):
746 1
                        links_found[link.id] = link
747 1
        return links_found
748
749 1
    def handle_link_liveness_disabled(self, interfaces) -> None:
750
        """Handle link liveness disabled."""
751 1
        log.info(f"Link liveness disabled interfaces: {interfaces}")
752
753 1
        key = "liveness_status"
754 1
        links = self.get_links_from_interfaces(interfaces)
755 1
        for link in links.values():
756 1
            link.remove_metadata(key)
757 1
        link_ids = list(links.keys())
758 1
        self.topo_controller.bulk_delete_link_metadata_key(link_ids, key)
759 1
        self.notify_topology_update()
760 1
        for link in links.values():
761 1
            self.notify_link_status_change(link, reason="liveness_disabled")
762
763 1
    @listen_to("kytos/core.interface_tags")
764 1
    def on_interface_tags(self, event):
765
        """Handle on_interface_tags."""
766
        interface = event.content['interface']
767
        with self._intfs_lock[interface.id]:
768
            if (
769
                interface.id in self._intfs_tags_updated_at
770
                and self._intfs_tags_updated_at[interface.id] > event.timestamp
771
            ):
772
                return
773
            self._intfs_tags_updated_at[interface.id] = event.timestamp
774
        self.handle_on_interface_tags(interface)
775
776 1
    def handle_on_interface_tags(self, interface):
777
        """Update interface details"""
778 1
        intf_id = interface.id
779 1
        self.topo_controller.upsert_interface_details(
780
            intf_id, interface.available_tags, interface.tag_ranges,
781
            interface.special_available_tags,
782
            interface.special_tags
783
        )
784
785 1
    @listen_to('.*.switch.(new|reconnected)')
786 1
    def on_new_switch(self, event):
787
        """Create a new Device on the Topology.
788
789
        Handle the event of a new created switch and update the topology with
790
        this new device. Also notify if the switch is enabled.
791
        """
792
        self.handle_new_switch(event)
793
794 1
    def handle_new_switch(self, event):
795
        """Create a new Device on the Topology."""
796 1
        switch = event.content['switch']
797 1
        switch.activate()
798 1
        self.topo_controller.upsert_switch(switch.id, switch.as_dict())
799 1
        log.debug('Switch %s added to the Topology.', switch.id)
800 1
        self.notify_topology_update()
801 1
        if switch.is_enabled():
802 1
            self.notify_switch_enabled(switch.id)
803
804 1
    @listen_to('.*.connection.lost')
805 1
    def on_connection_lost(self, event):
806
        """Remove a Device from the topology.
807
808
        Remove the disconnected Device and every link that has one of its
809
        interfaces.
810
        """
811
        self.handle_connection_lost(event)
812
813 1
    def handle_connection_lost(self, event):
814
        """Remove a Device from the topology."""
815 1
        switch = event.content['source'].switch
816 1
        if switch:
817 1
            switch.deactivate()
818 1
            log.debug('Switch %s removed from the Topology.', switch.id)
819 1
            self.notify_topology_update()
820
821 1
    def handle_interfaces_created(self, event):
822
        """Update the topology based on the interfaces created."""
823 1
        interfaces = event.content["interfaces"]
824 1
        if not interfaces:
825
            return
826 1
        switch = interfaces[0].switch
827 1
        self.topo_controller.upsert_switch(switch.id, switch.as_dict())
828 1
        name = "kytos/topology.switch.interface.created"
829 1
        for interface in interfaces:
830 1
            event = KytosEvent(name=name, content={'interface': interface})
831 1
            self.controller.buffers.app.put(event)
832
833 1
    def handle_interface_created(self, event):
834
        """Update the topology based on an interface created event.
835
836
        It's handled as a link_up in case a switch send a
837
        created event again and it can be belong to a link.
838
        """
839 1
        interface = event.content['interface']
840 1
        if not interface.is_active():
841 1
            return
842 1
        self.handle_interface_link_up(interface, event)
843
844 1
    @listen_to('.*.topology.switch.interface.created')
845 1
    def on_interface_created(self, event):
846
        """Handle individual interface create event.
847
848
        It's handled as a link_up in case a switch send a
849
        created event it can belong to an existign link.
850
        """
851
        self.handle_interface_created(event)
852
853 1
    @listen_to('.*.switch.interfaces.created')
854 1
    def on_interfaces_created(self, event):
855
        """Update the topology based on a list of created interfaces."""
856
        self.handle_interfaces_created(event)
857
858 1
    def handle_interface_down(self, event):
859
        """Update the topology based on a Port Modify event.
860
861
        The event notifies that an interface was changed to 'down'.
862
        """
863 1
        interface = event.content['interface']
864 1
        with self._intfs_lock[interface.id]:
865 1
            if (
866
                interface.id in self._intfs_updated_at
867
                and self._intfs_updated_at[interface.id] > event.timestamp
868
            ):
869
                return
870 1
            self._intfs_updated_at[interface.id] = event.timestamp
871 1
        interface.deactivate()
872 1
        self.handle_interface_link_down(interface, event)
873
874 1
    @listen_to('.*.switch.interface.deleted')
875 1
    def on_interface_deleted(self, event):
876
        """Update the topology based on a Port Delete event."""
877
        self.handle_interface_deleted(event)
878
879 1
    def handle_interface_deleted(self, event):
880
        """Update the topology based on a Port Delete event."""
881 1
        self.handle_interface_down(event)
882
883 1
    @listen_to('.*.switch.interface.link_up')
884 1
    def on_interface_link_up(self, event):
885
        """Update the topology based on a Port Modify event.
886
887
        The event notifies that an interface's link was changed to 'up'.
888
        """
889
        interface = event.content['interface']
890
        self.handle_interface_link_up(interface, event)
891
892 1
    def handle_interface_link_up(self, interface, event):
893
        """Update the topology based on a Port Modify event."""
894 1
        with self._intfs_lock[interface.id]:
895 1
            if (
896
                interface.id in self._intfs_updated_at
897
                and self._intfs_updated_at[interface.id] > event.timestamp
898
            ):
899 1
                return
900 1
            self._intfs_updated_at[interface.id] = event.timestamp
901 1
        self.handle_link_up(interface)
902
903 1
    def link_status_hook_link_up_timer(self, link) -> Optional[EntityStatus]:
904
        """Link status hook link up timer."""
905 1
        tnow = time.time()
906 1
        if (
907
            link.is_active()
908
            and link.is_enabled()
909
            and "last_status_change" in link.metadata
910
            and tnow - link.metadata['last_status_change'] < self.link_up_timer
911
        ):
912 1
            return EntityStatus.DOWN
913 1
        return None
914
915 1
    def notify_link_up_if_status(self, link, reason="link up") -> None:
916
        """Tries to notify link up and topology changes based on its status
917
918
        Currently, it needs to wait up to a timer."""
919 1
        time.sleep(self.link_up_timer)
920 1
        if link.status != EntityStatus.UP:
921
            return
922 1
        with self._links_notify_lock[link.id]:
923 1
            notified_at = link.get_metadata("notified_up_at")
924 1
            if (
925
                notified_at
926
                and (now() - notified_at.replace(tzinfo=timezone.utc)).seconds
927
                < self.link_up_timer
928
            ):
929 1
                return
930 1
            key, notified_at = "notified_up_at", now()
931 1
            link.update_metadata(key, now())
932 1
            self.notify_topology_update()
933 1
            self.notify_link_status_change(link, reason)
934
935 1
    def handle_link_up(self, interface):
936
        """Handle link up for an interface."""
937 1
        with self._links_lock:
938 1
            link = self._get_link_from_interface(interface)
939 1
            if not link:
940
                self.notify_topology_update()
941
                return
942 1
            other_interface = (
943
                link.endpoint_b if link.endpoint_a == interface
944
                else link.endpoint_a
945
            )
946 1
            if other_interface.is_active() is False:
947 1
                self.notify_topology_update()
948 1
                return
949 1
            metadata = {
950
                'last_status_change': time.time(),
951
                'last_status_is_active': True
952
            }
953 1
            link.extend_metadata(metadata)
954 1
            link.activate()
955 1
            self.notify_topology_update()
956 1
        self.notify_link_up_if_status(link, "link up")
957
958 1
    @listen_to('.*.switch.interface.link_down')
959 1
    def on_interface_link_down(self, event):
960
        """Update the topology based on a Port Modify event.
961
962
        The event notifies that an interface's link was changed to 'down'.
963
        """
964
        interface = event.content['interface']
965
        self.handle_interface_link_down(interface, event)
966
967 1
    def handle_interface_link_down(self, interface, event):
968
        """Update the topology based on an interface."""
969 1
        with self._intfs_lock[interface.id]:
970 1
            if (
971
                interface.id in self._intfs_updated_at
972
                and self._intfs_updated_at[interface.id] > event.timestamp
973
            ):
974 1
                return
975 1
            self._intfs_updated_at[interface.id] = event.timestamp
976 1
        self.handle_link_down(interface)
977
978 1
    def handle_link_down(self, interface):
979
        """Notify a link is down."""
980 1
        with self._links_lock:
981 1
            link = self._get_link_from_interface(interface)
982 1
            if not link or not link.get_metadata("last_status_is_active"):
983 1
                self.notify_topology_update()
984 1
                return
985 1
            link.deactivate()
986 1
            metadata = {
987
                "last_status_change": time.time(),
988
                "last_status_is_active": False,
989
            }
990 1
            link.extend_metadata(metadata)
991 1
            self.notify_link_status_change(link, reason="link down")
992 1
            self.notify_topology_update()
993
994 1
    @listen_to('.*.interface.is.nni')
995 1
    def on_add_links(self, event):
996
        """Update the topology with links related to the NNI interfaces."""
997
        self.add_links(event)
998
999 1
    def add_links(self, event):
1000
        """Update the topology with links related to the NNI interfaces."""
1001 1
        interface_a = event.content['interface_a']
1002 1
        interface_b = event.content['interface_b']
1003
1004 1
        try:
1005 1
            with self._links_lock:
1006 1
                link, created = self._get_link_or_create(interface_a,
1007
                                                         interface_b)
1008 1
                interface_a.update_link(link)
1009 1
                interface_b.update_link(link)
1010
1011 1
                link.endpoint_a = interface_a
1012 1
                link.endpoint_b = interface_b
1013
1014 1
                interface_a.nni = True
1015 1
                interface_b.nni = True
1016
1017
        except KytosLinkCreationError as err:
1018
            log.error(f'Error creating link: {err}.')
1019
            return
1020
1021 1
        if not created:
1022
            return
1023
1024 1
        self.notify_topology_update()
1025 1
        if not link.is_active():
1026
            return
1027
1028 1
        metadata = {
1029
            'last_status_change': time.time(),
1030
            'last_status_is_active': True
1031
        }
1032 1
        link.extend_metadata(metadata)
1033 1
        self.topo_controller.upsert_link(link.id, link.as_dict())
1034 1
        self.notify_link_up_if_status(link, "link up")
1035
1036 1
    @listen_to('.*.of_lldp.network_status.updated')
1037 1
    def on_lldp_status_updated(self, event):
1038
        """Handle of_lldp.network_status.updated from of_lldp."""
1039
        self.handle_lldp_status_updated(event)
1040
1041 1
    @listen_to(".*.topo_controller.upsert_switch")
1042 1
    def on_topo_controller_upsert_switch(self, event) -> None:
1043
        """Listen to topo_controller_upsert_switch."""
1044
        self.handle_topo_controller_upsert_switch(event.content["switch"])
1045
1046 1
    def handle_topo_controller_upsert_switch(self, switch) -> Optional[dict]:
1047
        """Handle topo_controller_upsert_switch."""
1048 1
        return self.topo_controller.upsert_switch(switch.id, switch.as_dict())
1049
1050 1
    def handle_lldp_status_updated(self, event) -> None:
1051
        """Handle .*.network_status.updated events from of_lldp."""
1052 1
        content = event.content
1053 1
        interface_ids = content["interface_ids"]
1054 1
        switches = set()
1055 1
        for interface_id in interface_ids:
1056 1
            dpid = ":".join(interface_id.split(":")[:-1])
1057 1
            switch = self.controller.get_switch_by_dpid(dpid)
1058 1
            if switch:
1059 1
                switches.add(switch)
1060
1061 1
        name = "kytos/topology.topo_controller.upsert_switch"
1062 1
        for switch in switches:
1063 1
            event = KytosEvent(name=name, content={"switch": switch})
1064 1
            self.controller.buffers.app.put(event)
1065
1066 1
    def notify_switch_enabled(self, dpid):
1067
        """Send an event to notify that a switch is enabled."""
1068 1
        name = 'kytos/topology.switch.enabled'
1069 1
        event = KytosEvent(name=name, content={'dpid': dpid})
1070 1
        self.controller.buffers.app.put(event)
1071
1072 1
    def notify_switch_links_status(self, switch, reason):
1073
        """Send an event to notify the status of a link in a switch"""
1074 1
        with self._links_lock:
1075 1
            for link in self.links.values():
1076 1
                if switch in (link.endpoint_a.switch, link.endpoint_b.switch):
1077 1
                    if reason == "link enabled":
1078 1
                        name = 'kytos/topology.notify_link_up_if_status'
1079 1
                        content = {'reason': reason, "link": link}
1080 1
                        event = KytosEvent(name=name, content=content)
1081 1
                        self.controller.buffers.app.put(event)
1082
                    else:
1083 1
                        self.notify_link_status_change(link, reason)
1084
1085 1
    def notify_switch_disabled(self, dpid):
1086
        """Send an event to notify that a switch is disabled."""
1087 1
        name = 'kytos/topology.switch.disabled'
1088 1
        event = KytosEvent(name=name, content={'dpid': dpid})
1089 1
        self.controller.buffers.app.put(event)
1090
1091 1
    def notify_topology_update(self):
1092
        """Send an event to notify about updates on the topology."""
1093 1
        name = 'kytos/topology.updated'
1094 1
        event = KytosEvent(name=name, content={'topology':
1095
                                               self._get_topology()})
1096 1
        self.controller.buffers.app.put(event)
1097
1098 1
    def notify_interface_link_status(self, interface, reason):
1099
        """Send an event to notify the status of a link from
1100
        an interface."""
1101 1
        link = self._get_link_from_interface(interface)
1102 1
        if link:
1103 1
            if reason == "link enabled":
1104 1
                name = 'kytos/topology.notify_link_up_if_status'
1105 1
                content = {'reason': reason, "link": link}
1106 1
                event = KytosEvent(name=name, content=content)
1107 1
                self.controller.buffers.app.put(event)
1108
            else:
1109 1
                self.notify_link_status_change(link, reason)
1110
1111 1
    def notify_link_status_change(self, link, reason='not given'):
1112
        """Send an event to notify about a status change on a link."""
1113 1
        link_id = link.id
1114 1
        with self.link_status_lock:
1115 1
            if (
1116
                (not link.status_reason and link.status == EntityStatus.UP)
1117
                and link_id not in self.link_up
1118
            ):
1119 1
                self.link_up.add(link_id)
1120 1
                event = KytosEvent(
1121
                    name='kytos/topology.link_up',
1122
                    content={
1123
                        'link': link,
1124
                        'reason': reason
1125
                    },
1126
                )
1127 1
            elif (
1128
                (link.status_reason or link.status != EntityStatus.UP)
1129
                and link_id in self.link_up
1130
            ):
1131 1
                self.link_up.remove(link_id)
1132 1
                event = KytosEvent(
1133
                    name='kytos/topology.link_down',
1134
                    content={
1135
                        'link': link,
1136
                        'reason': reason
1137
                    },
1138
                )
1139
            else:
1140 1
                return
1141 1
        self.controller.buffers.app.put(event)
1142
1143 1
    def notify_metadata_changes(self, obj, action):
1144
        """Send an event to notify about metadata changes."""
1145 1
        if isinstance(obj, Switch):
1146 1
            entity = 'switch'
1147 1
            entities = 'switches'
1148 1
        elif isinstance(obj, Interface):
1149 1
            entity = 'interface'
1150 1
            entities = 'interfaces'
1151 1
        elif isinstance(obj, Link):
1152 1
            entity = 'link'
1153 1
            entities = 'links'
1154
        else:
1155 1
            raise ValueError(
1156
                'Invalid object, supported: Switch, Interface, Link'
1157
            )
1158
1159 1
        name = f'kytos/topology.{entities}.metadata.{action}'
1160 1
        content = {entity: obj, 'metadata': obj.metadata.copy()}
1161 1
        event = KytosEvent(name=name, content=content)
1162 1
        self.controller.buffers.app.put(event)
1163 1
        log.debug(f'Metadata from {obj.id} was {action}.')
1164
1165 1
    @listen_to('kytos/topology.notify_link_up_if_status')
1166 1
    def on_notify_link_up_if_status(self, event):
1167
        """Tries to notify link up and topology changes"""
1168
        link = event.content["link"]
1169
        reason = event.content["reason"]
1170
        self.notify_link_up_if_status(link, reason)
1171
1172 1
    @listen_to('.*.switch.port.created')
1173 1
    def on_notify_port_created(self, event):
1174
        """Notify when a port is created."""
1175
        self.notify_port_created(event)
1176
1177 1
    def notify_port_created(self, event):
1178
        """Notify when a port is created."""
1179 1
        name = 'kytos/topology.port.created'
1180 1
        event = KytosEvent(name=name, content=event.content)
1181 1
        self.controller.buffers.app.put(event)
1182
1183 1
    @staticmethod
1184 1
    def load_interfaces_tags_values(switch: Switch,
1185
                                    interfaces_details: List[dict]) -> None:
1186
        """Load interfaces available tags (vlans)."""
1187 1
        if not interfaces_details:
1188
            return
1189 1
        for interface_details in interfaces_details:
1190 1
            available_tags = interface_details['available_tags']
1191 1
            if not available_tags:
1192
                continue
1193 1
            log.debug(f"Interface id {interface_details['id']} loading "
1194
                      f"{len(available_tags)} "
1195
                      "available tags")
1196 1
            port_number = int(interface_details["id"].split(":")[-1])
1197 1
            interface = switch.interfaces[port_number]
1198 1
            interface.set_available_tags_tag_ranges(
1199
                available_tags,
1200
                interface_details['tag_ranges'],
1201
                interface_details['special_available_tags'],
1202
                interface_details['special_tags'],
1203
            )
1204
1205 1
    @listen_to('topology.interruption.start')
1206 1
    def on_interruption_start(self, event: KytosEvent):
1207
        """Deals with the start of service interruption."""
1208
        with self._links_lock:
1209
            self.handle_interruption_start(event)
1210
1211 1 View Code Duplication
    def handle_interruption_start(self, event: KytosEvent):
0 ignored issues
show
This code seems to be duplicated in your project.
Loading history...
1212
        """Deals with the start of service interruption."""
1213 1
        interrupt_type = event.content['type']
1214 1
        switches = event.content.get('switches', [])
1215 1
        interfaces = event.content.get('interfaces', [])
1216 1
        links = event.content.get('links', [])
1217 1
        log.info(
1218
            'Received interruption start of type \'%s\' '
1219
            'affecting switches %s, interfaces %s, links %s',
1220
            interrupt_type,
1221
            switches,
1222
            interfaces,
1223
            links
1224
        )
1225
        # for switch_id in switches:
1226
        #     pass
1227
        # for interface_id in interfaces:
1228
        #     pass
1229 1
        for link_id in links:
1230 1
            link = self.links.get(link_id)
1231 1
            if link is None:
1232
                log.error(
1233
                    "Invalid link id '%s' for interruption of type '%s;",
1234
                    link_id,
1235
                    interrupt_type
1236
                )
1237
            else:
1238 1
                self.notify_link_status_change(link, interrupt_type)
1239 1
        self.notify_topology_update()
1240
1241 1
    @listen_to('topology.interruption.end')
1242 1
    def on_interruption_end(self, event: KytosEvent):
1243
        """Deals with the end of service interruption."""
1244
        with self._links_lock:
1245
            self.handle_interruption_end(event)
1246
1247 1 View Code Duplication
    def handle_interruption_end(self, event: KytosEvent):
0 ignored issues
show
This code seems to be duplicated in your project.
Loading history...
1248
        """Deals with the end of service interruption."""
1249 1
        interrupt_type = event.content['type']
1250 1
        switches = event.content.get('switches', [])
1251 1
        interfaces = event.content.get('interfaces', [])
1252 1
        links = event.content.get('links', [])
1253 1
        log.info(
1254
            'Received interruption end of type \'%s\' '
1255
            'affecting switches %s, interfaces %s, links %s',
1256
            interrupt_type,
1257
            switches,
1258
            interfaces,
1259
            links
1260
        )
1261
        # for switch_id in switches:
1262
        #     pass
1263
        # for interface_id in interfaces:
1264
        #     pass
1265 1
        for link_id in links:
1266 1
            link = self.links.get(link_id)
1267 1
            if link is None:
1268
                log.error(
1269
                    "Invalid link id '%s' for interruption of type '%s;",
1270
                    link_id,
1271
                    interrupt_type
1272
                )
1273
            else:
1274 1
                self.notify_link_status_change(link, interrupt_type)
1275
        self.notify_topology_update()
1276