serverless-benchmarks/sebs/faas/queue.py at 86540afa7f72f9dfcee87b89d3faf3e0082885d1 · spcl/serverless-benchmarks · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
from abc import ABC
from abc import abstractmethod
from enum import Enum

from sebs.utils import LoggingBase


class QueueType(str, Enum):
    TRIGGER = "trigger"
    RESULT = "result"


class Queue(ABC, LoggingBase):

    @staticmethod
    @abstractmethod
    def deployment_name() -> str:
        pass

    @property
    def region(self):
        return self._region

    @property
    def queue_type(self):
        return self._queue_type

    @property
    def name(self):
        return self._name

    def __init__(
        self,
        benchmark: str,
        queue_type: QueueType,
        region: str
    ):
        super().__init__()
        self._name = benchmark

        # Convention: the trigger queue carries the name of the function. The
        # result queue carries the name of the function + "-result".
        if (queue_type == QueueType.RESULT and not benchmark.endswith("-result")):
            self._name = "{}-{}".format(benchmark, queue_type)

        self._queue_type = queue_type
        self._region = region

    @abstractmethod
    def create_queue(self):
        pass

    @abstractmethod
    def remove_queue(self):
        pass

    @abstractmethod
    def send_message(self, serialized_message: str):
        pass

    @abstractmethod
    def receive_message(self) -> str:
        pass