.gitignore

@@ -118,5 +118,4 @@ GitHub.sublime-settings
 !.vscode/extensions.json
 .history
 
-*pb2*
-schemas
+*pb2*

Makefile

@@ -1,6 +1,4 @@
 gen:
-	curl https://platform.sprinthub.ru/generator >> generator.py
-	python generator.py
-	rm generator.py
+	python -m grpc_tools.protoc --proto_path schemas --python_out=. --pyi_out=. --grpc_python_out=. ./schemas/tasks.proto
 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;
+}

server.py

@@ -3,8 +3,8 @@ import datetime
 import grpc
 import bson
 
-from queues import tasks_pb2
-from queues import tasks_pb2_grpc
+import tasks_pb2
+import tasks_pb2_grpc
 
 from utils import time
 from storage.mongo import tasks