report number of subscriptions

src/Hash2Pub/PostService.hs

@@ -36,7 +36,7 @@ import           System.IO
 import           System.Random
 import           Text.Read                 (readEither)
 
-import           Formatting                (fixed, float, format, (%))
+import           Formatting                (fixed, format, int, (%))
 import qualified Network.Wai.Handler.Warp  as Warp
 import           Servant
 import           Servant.Client
@@ -67,6 +67,7 @@ data PostService d = PostService
     , httpMan              :: HTTP.Manager
     , statsQueue           :: TQueue StatsEvent
     , loadStats            :: TVar RelayStats
+    -- ^ current load stats, replaced periodically
     , logFileHandle        :: Handle
     }
     deriving (Typeable)
@@ -120,7 +121,7 @@ instance DHT d => Service PostService d where
         -- log a start message, this also truncates existing files
         TxtI.hPutStrLn loggingFile $ Txt.unlines
             [ "# Starting mock relay implementation"
-            , "#relay receive rate   ;relay delivery rate   ;instance publish rate ;instance fetch rate"
+            , "#relay receive rate   ;relay delivery rate   ;instance publish rate ;instance fetch rate   ;total subscriptions"
             ]
         -- Run 'concurrently_' from another thread to be able to return the
         -- 'PostService'.
@@ -681,7 +682,6 @@ fetchTagPosts serv = forever $ do
           --if HTTPT.statusCode (HTTP.responseStatus response) == 200
           --   then
           --   -- success, TODO: statistics
-          --   putStrLn "post fetch success"
           --   else
           pure ()
       Left _ ->
@@ -723,6 +723,7 @@ relayWorker serv = forever $ do
         runningJobs <- mapM async jobset
         -- so far just dropping failed attempts, TODO: retry mechanism
         successfulResults <- rights <$> mapM waitCatch runningJobs
+        putStrLn $ "successfully relayed " <> show (length successfulResults)
         pure ()
 
 
@@ -818,16 +819,26 @@ evaluateStatsThread serv statsAcc = getPOSIXTime >>= loop
           let rateStats = evaluateStats timePassed summedStats
           atomically $ writeTVar (loadStats serv) rateStats
           -- and now what? write a log to file
-          -- format: total relayReceiveRates;total relayDeliveryRates;postFetchRate;postPublishRate
+          -- format: total relayReceiveRates;total relayDeliveryRates;postFetchRate;postPublishRate; subscriberSum
           -- later: current (reported) load, target load
+          subscriberSum <- sumSubscribers
           TxtI.hPutStrLn (logFileHandle serv) $
-              format (fixed 20 % ";" % fixed 20 % ";" % fixed 20 % ";" % fixed 20)
+              format (fixed 20 % ";" % fixed 20 % ";" % fixed 20 % ";" % fixed 20 % ";" % int )
                   (sum . relayReceiveRates $ rateStats)
                   (sum . relayDeliveryRates $ rateStats)
                   (postPublishRate rateStats)
                   (postFetchRate rateStats)
+                  subscriberSum
           loop now
 
+      sumSubscribers = do
+          tagMap <- readTVarIO $ subscribers serv
+          foldM (\subscriberSum (subscriberMapSTM, _, _) -> do
+              subscriberMap <- readTVarIO subscriberMapSTM
+              pure $ subscriberSum + HMap.size subscriberMap
+                )
+                0 tagMap
+
 
 -- | Evaluate the accumulated statistic events: Currently mostly calculates the event
 -- rates by dividing through the collection time frame