Update

Procfile

@@ -1,2 +1,2 @@
-web: gunicorn web.app:app --bind 0.0.0.0:8000 --workers 2
+web: gunicorn web.app:app --bind 0.0.0.0:8000 --workers 4
 firehose: python start_stream.py

README.md

@@ -59,7 +59,7 @@ Open `.env` in your preferred text editor and fill in the necessary variables.
 
 Build and run Docker image:
 ```shell
-docker build -t myfeed .
+docker build --rm -t myfeed .
 docker run --rm -it --env-file .env -p 8000:8000 -v feeddata:/var/data/ myfeed
 ```
 

firehose/data_stream.py

@@ -123,7 +123,7 @@ def _run(name, operations_callback, stream_stop_event=None):
         SubscriptionState.create(service=name, cursor=0)
 
     # Initialize the firehose client w/o a cursor for now
-    client = FirehoseSubscribeReposClient()
+    client = FirehoseSubscribeReposClient(params)
 
     def on_message_handler(message: firehose_models.MessageFrame) -> None:
         """
@@ -156,7 +156,10 @@ def on_message_handler(message: firehose_models.MessageFrame) -> None:
             # Update the client's parameters with the new cursor
             client.update_params(models.ComAtprotoSyncSubscribeRepos.Params(cursor=commit.seq))
             # Persist the new cursor in the database
-            #SubscriptionState.update(cursor=commit.seq).where(SubscriptionState.service == name).execute()
+            try:
+                SubscriptionState.update(cursor=commit.seq).where(SubscriptionState.service == name).execute()
+            except Exception as e:
+                logger.error(f"Failed to update cursor in database: {e}")
 
         if not commit.blocks:
             # Skip if there are no blocks to process

web/app.py

@@ -41,9 +41,7 @@ def describe_feed_generator():
 @app.route('/xrpc/app.bsky.feed.getFeedSkeleton', methods=['GET'])
 def get_feed_skeleton():
     feed = request.args.get('feed', default=None, type=str)
-    logger.info(f'Feed: {feed}')
     algo = algos.get(feed)
-    logger.info(f'Algos: {algos}')
     if not algo:
         return 'Unsupported algorithm', 400