Refactor. Move parts

Author: sleipnir

example/spawn_example.py

@@ -2,7 +2,7 @@
 Copyright 2022 Eigr.
 Licensed under the Apache License, Version 2.0.
 """
-from spawn.sdk import Spawn
+from spawn.eigr.functions.actors.api.sdk import Spawn
 from example.joe import JoeActor
 from example.domain.domain_pb2 import Reply, Request
 

spawn/eigr/functions/actors/api/context.py

@@ -1,10 +1,10 @@
-from dataclasses import dataclass
-from typing import Dict
+from dataclasses import dataclass, field
+from typing import MutableMapping
 
 
 @dataclass
 class Context:
     state: object
     caller: str = None
-    metadata: Dict[str, str] = None
-    tags: Dict[str, str] = None
+    metadata: MutableMapping[str, str] = field(default_factory=dict)
+    tags: MutableMapping[str, str] = field(default_factory=dict)

spawn/eigr/functions/actors/api/decorators/action.py

@@ -3,12 +3,13 @@
 
 from spawn.eigr.functions.actors.api.context import Context
 from spawn.eigr.functions.actors.api.metadata import Metadata
+from spawn.eigr.functions.actors.api.settings import ActorSettings
 from spawn.eigr.functions.actors.api.value import Value
 from spawn.eigr.functions.actors.api.workflows.broadcast import Broadcast
 from spawn.eigr.functions.actors.api.workflows.effect import Effect
 from spawn.eigr.functions.actors.api.decorators.actor import Actor
 
-from spawn.eigr.functions.actors.core import ActionInfo, ActorSettings, Actors
+from spawn.eigr.functions.actors.internal.core import ActionInfo, Actors
 
 logging.basicConfig(level=logging.DEBUG)
 _logger = logging.getLogger(__name__)

spawn/eigr/functions/actors/api/decorators/actor.py

@@ -1,6 +1,6 @@
 import functools
 
-from spawn.eigr.functions.actors.core import ActorInfo, Actors
+from spawn.eigr.functions.actors.internal.core import ActorInfo, Actors
 
 
 def Actor(cls):

spawn/sdk.py spawn/eigr/functions/actors/api/sdk.pyspawn/sdk.py renamed to spawn/eigr/functions/actors/api/sdk.py

@@ -6,7 +6,6 @@
 
 from dataclasses import dataclass, field
 
-from spawn.entity import ActorEntity
 from spawn.handler import action_handler
 from spawn.controller import SpawnActorController as ActorController
 
@@ -30,9 +29,9 @@ class Spawn:
 
     __host = os.environ.get("HOST", "0.0.0.0")
     __port = os.environ.get("PORT", "8091")
-    __actors: List[ActorEntity] = field(default_factory=list)
     __app = Flask(__name__)
-    __is_debug_enable = json.loads(os.environ.get("SDK_DEBUG_ENABLE", "false").lower())
+    __is_debug_enable = json.loads(
+        os.environ.get("SDK_DEBUG_ENABLE", "false").lower())
     __actorController = ActorController(
         os.environ.get("PROXY_HOST", "localhost"),
         os.environ.get("PROXY_PORT", "9002"),
@@ -65,7 +64,8 @@ def start(self):
         """Start the user function and HTTP Server."""
         address = "{}:{}".format(self.__host, self.__port)
 
-        server = threading.Thread(target=lambda: self.__start_server(action_handler))
+        server = threading.Thread(
+            target=lambda: self.__start_server(action_handler))
         logging.info("Starting Spawn on address %s", address)
         try:
             server.start()

spawn/eigr/functions/actors/api/settings.py

@@ -0,0 +1,29 @@
+from enum import Enum
+from typing import Generic, TypeVar
+
+T = TypeVar('T')
+
+
+class Kind(str, Enum):
+    NAMED = 'NAMED'
+    UNNAMED = 'UNNAMED'
+    POOLED = 'POOLED'
+
+
+class ActorSettings:
+    def __init__(
+            self,
+            name: str,
+            kind: Kind = Kind.NAMED,
+            stateful: bool = True,
+            state_type: Generic[T] = None,
+            channel: str = None,
+            deactivate_timeout: int = 30,
+            snapshot_timeout: int = 10):
+        self.name = name
+        self.kind = kind
+        self.stateful = stateful
+        self.state_type = state_type
+        self.channel = channel
+        self.deactivate_timeout = deactivate_timeout
+        self.snapshot_timeout = snapshot_timeout

spawn/eigr/functions/actors/core.py

@@ -0,0 +1,4 @@
+"""
+Copyright 2022 Eigr.
+Licensed under the Apache License, Version 2.0.
+"""

spawn/eigr/functions/actors/internal/__init__.py

@@ -0,0 +1,32 @@
+from typing import List, MutableMapping
+
+from spawn.eigr.functions.actors.api.settings import ActorSettings
+
+
+class ActionInfo:
+    def __init__(self, name, input=None, output=None):
+        self.name = name
+        self.input = input
+        self.output = output
+
+
+class ActorInfo:
+    def __init__(self, name: str, settings: ActorSettings, actions: List[ActionInfo] = None) -> None:
+        self.name = name
+        self.settings = settings
+        self.actions = actions
+
+
+class Actors:
+    _instance = None
+    actors = MutableMapping[str, ActorInfo]
+
+    def __init__(self):
+        if not self.actors:
+            self.actors = {}
+
+    def __new__(cls, *args, **kwargs):
+        if not isinstance(cls._instance, cls):
+            cls._instance = object.__new__(cls)
+
+        return cls._instance