Merge pull request #30 from snuhcs-course/ml/modify-stub-gen-path

Ml/modify stub gen path & Add dockerfile for ml

.gitignore

@@ -1 +1,2 @@
 .idea
+__pycache__

idl/buf.gen.yaml

@@ -5,6 +5,6 @@ plugins:
   - plugin: buf.build/protocolbuffers/ruby
     out: ../backend/gen
   - plugin: buf.build/grpc/python:v1.59.2
-    out: ../ml/gen/
+    out: ../ml/
   - plugin: buf.build/protocolbuffers/python
-    out: ../ml/gen/
+    out: ../ml/

ml/Dockerfile

@@ -0,0 +1,16 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+RUN pip install poetry
+
+COPY pyproject.toml poetry.lock ./
+
+RUN poetry config virtualenvs.create false && \
+    poetry install --without dev
+
+COPY . .
+
+CMD python main.py
+
+

ml/main.py

@@ -1,9 +1,10 @@
+import os
 from concurrent import futures
 
 import grpc
 
-from gen.swpp.v1 import place_pb2
-from gen.swpp.v1.place_pb2_grpc import PlaceServiceServicer, add_PlaceServiceServicer_to_server
+from swpp.v1 import place_pb2
+from swpp.v1.place_pb2_grpc import PlaceServiceServicer, add_PlaceServiceServicer_to_server
 
 
 class PlaceService(PlaceServiceServicer):
@@ -17,9 +18,11 @@ def ListPlaces(self, request, context):
 
 
 if __name__ == '__main__':
-    # Create Unary Server
+    port = os.getenv("PORT", "50051")
+
     server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
     add_PlaceServiceServicer_to_server(PlaceService(), server)
-    server.add_insecure_port('[::]:50051')
+    server.add_insecure_port(f"[::]:{port}")
     server.start()
+    print(f"Server started on port {port}")
     server.wait_for_termination()