apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-24 04:31:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Update kg core endpoint methods

Browse source
This commit is contained in:
Cyber MacGeddon 2025-06-02 13:41:25 +01:00
parent 2e577e900a
commit 7cfa40c85c
2 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
trustgraph-flow/trustgraph/gateway/endpoint/manager.py
View file

@ -55,11 +55,13 @@ class EndpointManager:
StreamEndpoint( StreamEndpoint(
endpoint_path = "/api/v1/import-core", endpoint_path = "/api/v1/import-core",
auth = auth, auth = auth,
method = "POST",
dispatcher = dispatcher_manager.dispatch_core_import(), dispatcher = dispatcher_manager.dispatch_core_import(),
), ),
StreamEndpoint( StreamEndpoint(
endpoint_path = "/api/v1/export-core", endpoint_path = "/api/v1/export-core",
auth = auth, auth = auth,
method = "GET",
dispatcher = dispatcher_manager.dispatch_core_export(), dispatcher = dispatcher_manager.dispatch_core_export(),
), ),
] ]

16
trustgraph-flow/trustgraph/gateway/endpoint/stream_endpoint.py
View file

@ -8,12 +8,13 @@ logger.setLevel(logging.INFO)
class StreamEndpoint: class StreamEndpoint:
def __init__(self, endpoint_path, auth, dispatcher): def __init__(self, endpoint_path, auth, dispatcher, method="POST"):
self.path = endpoint_path self.path = endpoint_path
self.auth = auth self.auth = auth
self.operation = "service" self.operation = "service"
self.method = method
self.dispatcher = dispatcher self.dispatcher = dispatcher
@ -22,9 +23,16 @@ class StreamEndpoint:
def add_routes(self, app): def add_routes(self, app):
app.add_routes([ if self.method == "POST":
web.post(self.path, self.handle), app.add_routes([
]) web.post(self.path, self.handle),
])
elif self.method == "GET":
app.add_routes([
web.get(self.path, self.handle),
])
else:
raise RuntimeError("Bad method" + self.method)
async def handle(self, request): async def handle(self, request):