Skip to content

Commit 5a64a92

Browse files
committed
restructuring and renaming, queues and buffer use 'push' instead of redundant push_kobj/push_event_to
1 parent 7edec90 commit 5a64a92

30 files changed

+88
-110
lines changed

examples/coordinator.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
NodeProvides
1010
)
1111
from koi_net.core import FullNode
12-
from koi_net.context import HandlerContext
12+
from koi_net.processor.context import HandlerContext
1313
from koi_net.processor.handler import HandlerType, KnowledgeHandler
1414
from koi_net.processor.knowledge_object import KnowledgeObject
1515
from koi_net.protocol.event import Event, EventType
@@ -51,7 +51,7 @@ def handshake_handler(ctx: HandlerContext, kobj: KnowledgeObject):
5151

5252
logger.info("Sharing this node's bundle with peer")
5353
identity_bundle = ctx.cache.read(ctx.identity.rid)
54-
ctx.event_queue.push_event_to(
54+
ctx.event_queue.push(
5555
event=Event.from_bundle(EventType.NEW, identity_bundle),
5656
target=kobj.rid
5757
)
@@ -66,8 +66,8 @@ def handshake_handler(ctx: HandlerContext, kobj: KnowledgeObject):
6666
rid_types=[KoiNetNode, KoiNetEdge]
6767
)
6868

69-
ctx.kobj_queue.put_kobj(rid=edge_bundle.rid, event_type=EventType.FORGET)
70-
ctx.kobj_queue.put_kobj(bundle=edge_bundle)
69+
ctx.kobj_queue.push(rid=edge_bundle.rid, event_type=EventType.FORGET)
70+
ctx.kobj_queue.push(bundle=edge_bundle)
7171

7272
class CoordinatorNode(FullNode):
7373
config = CoordinatorConfig

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
44

55
[project]
66
name = "koi-net"
7-
version = "1.2.0b2"
7+
version = "1.2.0b3"
88
description = "Implementation of KOI-net protocol in Python"
99
authors = [
1010
{name = "Luke Miller", email = "[email protected]"}

src/koi_net/assembler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from pydantic import BaseModel
66
import structlog
77

8-
from .interfaces.entrypoint import EntryPoint
8+
from .entrypoints.base import EntryPoint
99

1010
log = structlog.stdlib.get_logger()
1111

src/koi_net/config/loader.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -16,12 +16,7 @@ def __getattr__(self, name):
1616
return getattr(self._config, name)
1717

1818
def load_from_yaml(self):
19-
"""Loads config state from YAML file.
20-
21-
Defaults to `config.yaml`. If `generate_missing` is set to
22-
`True`, a private key and RID will be generated if not already
23-
present in the config.
24-
"""
19+
"""Loads config state from YAML file."""
2520
yaml = YAML()
2621

2722
try:

src/koi_net/core.py

Lines changed: 9 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,11 @@
33
from .config.loader import ConfigLoader
44
from .assembler import NodeAssembler
55
from .config.core import NodeConfig
6-
from .context import HandlerContext
6+
from .processor.context import HandlerContext
77
from .effector import Effector
88
from .handshaker import Handshaker
99
from .identity import NodeIdentity
10-
from .processor.kobj_worker import KnowledgeProcessingWorker
10+
from .workers import KnowledgeProcessingWorker, EventProcessingWorker
1111
from .lifecycle import NodeLifecycle
1212
from .network.error_handler import ErrorHandler
1313
from .network.event_queue import EventQueue
@@ -16,8 +16,11 @@
1616
from .network.resolver import NetworkResolver
1717
from .network.response_handler import ResponseHandler
1818
from .network.poll_event_buffer import PollEventBuffer
19-
from .poller import NodePoller
20-
from .processor.handlers import (
19+
from .processor.pipeline import KnowledgePipeline
20+
from .processor.kobj_queue import KobjQueue
21+
from .secure import Secure
22+
from .entrypoints import NodeServer, NodePoller
23+
from .processor.knowledge_handlers import (
2124
basic_manifest_handler,
2225
basic_network_output_filter,
2326
basic_rid_handler,
@@ -26,18 +29,6 @@
2629
forget_edge_on_node_deletion,
2730
secure_profile_handler
2831
)
29-
from .processor.event_worker import EventProcessingWorker
30-
from .processor.pipeline import KnowledgePipeline
31-
from .processor.kobj_queue import KobjQueue
32-
from .secure import Secure
33-
from .server import NodeServer
34-
35-
36-
# factory functions for components with non standard initializiation
37-
38-
39-
def make_cache(config: NodeConfig) -> Cache:
40-
return Cache(directory_path=config.koi_net.cache_directory_path)
4132

4233

4334
class BaseNode(NodeAssembler):
@@ -55,7 +46,8 @@ class BaseNode(NodeAssembler):
5546
basic_network_output_filter,
5647
forget_edge_on_node_deletion
5748
]
58-
cache = make_cache
49+
cache = lambda config: Cache(
50+
directory_path=config.koi_net.cache_directory_path)
5951
identity = NodeIdentity
6052
graph = NetworkGraph
6153
secure = Secure

src/koi_net/default_actions.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,8 @@
11
"""Implementations of default dereference actions."""
22

3-
from .context import ActionContext
43
from rid_lib.types import KoiNetNode
54
from rid_lib.ext import Bundle
6-
from .effector import Effector
5+
from .effector import Effector, ActionContext
76

87

98
@Effector.register_default_action(KoiNetNode)

src/koi_net/effector.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,7 @@ def deref(
155155
and bundle is not None
156156
and source != BundleSource.CACHE
157157
):
158-
self.kobj_queue.put_kobj(
158+
self.kobj_queue.push(
159159
bundle=bundle,
160160
source=source if type(source) is KoiNetNode else None
161161
)

src/koi_net/entrypoint.py

Lines changed: 0 additions & 11 deletions
This file was deleted.
Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
from .poller import NodePoller
2+
from .server import NodeServer

src/koi_net/entrypoints/base.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
from koi_net.config.core import NodeConfig
2+
3+
4+
class EntryPoint:
5+
def run(self): ...

0 commit comments

Comments
 (0)