.gitignore

@@ -120,5 +120,4 @@ GitHub.sublime-settings
 
 local_platform.json
 
-*pb2*
+*pb2*
-schemas

Makefile

@@ -1,6 +1,2 @@
 gen:
-	curl https://platform.sprinthub.ru/generator >> generator.py
+	python -m grpc_tools.protoc --proto_path schemas --python_out=. --pyi_out=. --grpc_python_out=. ./schemas/tasks.proto
-	python generator.py
-	rm generator.py
-run:
-	python ./server.py

schemas/tasks.proto

@@ -0,0 +1,40 @@
+syntax = "proto3";
+
+package queues;
+
+import "google/protobuf/struct.proto";
+
+service Tasks {
+    rpc Put (PutRequest) returns (EmptyResponse) {}
+
+    rpc Take (TakeRequest) returns (TakeResponse) {}
+
+    rpc Finish (FinishRequest) returns (EmptyResponse) {}
+}
+
+message Task {
+    string id = 1;
+    int64 attempt = 2;
+    google.protobuf.Struct payload = 3;
+}
+
+message PutRequest {
+    string queue = 1;
+    int64 seconds_to_execute = 2;
+    optional int64 delay = 3;
+    google.protobuf.Struct payload = 4;
+}
+
+message TakeRequest {
+    string queue = 1;
+}
+
+message FinishRequest {
+    string id = 1;
+}
+
+message EmptyResponse {}
+
+message TakeResponse {
+    optional Task task = 1;
+}

utils/queues.py

@@ -1,6 +1,6 @@
 import time
-from queues import tasks_pb2_grpc
+import tasks_pb2_grpc
-from queues import tasks_pb2
+import tasks_pb2
 
 from google.protobuf import json_format