additional metrics

Author: pepeiborra

ghcide/src/Development/IDE/Core/Shake.hs

@@ -118,6 +118,7 @@ import           Development.IDE.GHC.Orphans            ()
 import           Development.IDE.Graph                  hiding (ShakeValue)
 import qualified Development.IDE.Graph                  as Shake
 import           Development.IDE.Graph.Database         (ShakeDatabase,
+                                                         shakeDatabaseDatabase,
                                                          shakeOpenDatabase,
                                                          shakeProfileDatabase,
                                                          shakeRunDatabaseForKeys)
@@ -154,12 +155,16 @@ import qualified Data.HashSet                           as HSet
 import           Data.IORef.Extra                       (atomicModifyIORef'_,
                                                          atomicModifyIORef_)
 import           Data.Text                              (pack)
+import           Development.IDE.Graph.Database         (shakeGetKeys)
+import           Development.IDE.Graph.Internal.Types   (Step (Step),
+                                                         databaseStep)
 import qualified Development.IDE.Types.Exports          as ExportsMap
 import           HieDb.Types
 import           Ide.Plugin.Config
 import qualified Ide.PluginUtils                        as HLS
 import           Ide.Types                              (PluginId)
-import           System.Metrics                         (Store, registerGauge)
+import           System.Metrics                         (Store, registerCounter,
+                                                         registerGauge)
 
 -- | Maximum age (in # builds) of a cached value after which it's considered garbage
 garbageAge :: Int
@@ -327,7 +332,7 @@ lastValueIO s@ShakeExtras{positionMapping,persistentKeys,state} k file = do
           | otherwise = do
           pmap <- readVar persistentKeys
           mv <- runMaybeT $ do
-            liftIO $ Logger.logDebug (logger s) $ T.pack $ "LOOKUP UP PERSISTENT FOR: " ++ show k
+            liftIO $ Logger.logDebug (logger s) $ T.pack $ "LOOKUP PERSISTENT FOR: " ++ show k
             f <- MaybeT $ pure $ HMap.lookup (Key k) pmap
             (dv,del,ver) <- MaybeT $ runIdeAction "lastValueIO" s $ f file
             MaybeT $ pure $ (,del,ver) <$> fromDynamic dv
@@ -548,11 +553,23 @@ shakeOpen lspEnv defaultConfig logger debouncer
         { optOTMemoryProfiling = IdeOTMemoryProfiling otProfilingEnabled
         , optProgressStyle
         } <- getIdeOptionsIO shakeExtras
+
     startTelemetry otProfilingEnabled logger $ state shakeExtras
 
     for_ metrics $ \store -> do
         let readValuesCounter = fromIntegral . HMap.size <$> readVar (state shakeExtras)
+            readDirtyKeys = fromIntegral . Prelude.length <$> readIORef (dirtyKeys shakeExtras)
+            readIndexPending = fromIntegral . HMap.size <$> readTVarIO (indexPending $ hiedbWriter shakeExtras)
+            readExportsMap = fromIntegral . HMap.size . getExportsMap <$> readVar (exportsMap shakeExtras)
+            readDatabaseCount = fromIntegral . Prelude.length <$> shakeGetKeys shakeDb
+            readDatabaseStep = (\(Step s) -> fromIntegral s) <$> readIORef (databaseStep $ shakeDatabaseDatabase shakeDb)
+
         registerGauge "ghcide.values_count" readValuesCounter store
+        registerGauge "ghcide.dirty_keys_count" readDirtyKeys store
+        registerGauge "ghcide.indexing_pending_count" readIndexPending store
+        registerCounter "ghcide.exports_map_count" readExportsMap store
+        registerCounter "ghcide.database_count" readDatabaseCount store
+        registerCounter "ghcide.build_count" readDatabaseStep store
 
     return ideState