Add entity counts (#5583)

Author: Isaac Solo

discovery-provider/src/queries/get_entities_count_check.py

@@ -0,0 +1,34 @@
+
+from src.models.playlists.playlist import Playlist
+from src.models.social.follow import Follow
+from src.models.social.repost import Repost
+from src.models.social.save import Save
+from src.models.tracks.track import Track
+from src.models.users.user import User
+from src.utils import db_session
+from src.utils.structured_logger import StructuredLogger
+
+logger = StructuredLogger(__name__)
+
+
+def get_entities_count_check():
+    """
+    Gets counts for each of these entities.
+    """
+    db = db_session.get_db_read_replica()
+    with db.scoped_session() as session:
+        user_counts = session.query(User).filter(User.is_current).count()
+        track_counts = session.query(Track).filter(Track.is_current).count()
+        playlist_counts = session.query(Playlist).filter(Playlist.is_current).count()
+        repost_counts = session.query(Repost).filter(Repost.is_current).count()
+        save_counts = session.query(Save).filter(Save.is_current).count()
+        follow_counts = session.query(Follow).filter(Follow.is_current).count()
+        res = {
+            "user_counts": user_counts,
+            "track_counts": track_counts,
+            "playlist_counts": playlist_counts,
+            "repost_counts": repost_counts,
+            "save_counts": save_counts,
+            "follow_counts": follow_counts
+        }
+        return res

discovery-provider/src/queries/health_check.py

@@ -10,6 +10,7 @@
 from src.queries.get_latest_play import get_latest_play
 from src.queries.get_sol_plays import get_latest_sol_play_check_info
 from src.queries.get_trusted_notifier_discrepancies import get_trusted_notifier_discrepancies
+from src.queries.get_entities_count_check import get_entities_count_check
 from src.queries.queries import parse_bool_param
 from src.tasks.index_profile_challenge_backfill import (
     index_profile_challenge_backfill_tablename,
@@ -71,6 +72,12 @@ def trusted_notifier_discrepancies_check():
     return success_response(health_results, 500 if error else 200, sign_response=False)
 
 
+@bp.route("/entities_count_check", methods=["GET"])
+def entities_count_check():
+    res_count = get_entities_count_check()
+    return success_response(res_count)
+
+
 # Health check for block diff between DB and chain.
 @bp.route("/block_check", methods=["GET"])
 def block_check():