confine CPP to the EKG module

Author: pepeiborra

ghcide/exe/Main.hs

@@ -1,7 +1,6 @@
 -- Copyright (c) 2019 The DAML Authors. All rights reserved.
 -- SPDX-License-Identifier: Apache-2.0
 {-# OPTIONS_GHC -Wno-dodgy-imports #-} -- GHC no longer exports def in GHC 8.6 and above
-{-# LANGUAGE CPP             #-}
 {-# LANGUAGE TemplateHaskell #-}
 
 module Main(main) where
@@ -20,6 +19,8 @@ import           Development.IDE.Core.Rules        (mainRule)
 import qualified Development.IDE.Core.Rules        as Rules
 import           Development.IDE.Core.Tracing      (withTelemetryLogger)
 import qualified Development.IDE.Main              as IDEMain
+import qualified Development.IDE.Monitoring.OpenTelemetry as OpenTelemetry
+import qualified Development.IDE.Monitoring.EKG    as EKG
 import qualified Development.IDE.Plugin.HLS.GhcIde as GhcIde
 import           Development.IDE.Types.Logger      (Logger (Logger),
                                                     LoggingColumn (DataColumn, PriorityColumn),
@@ -43,11 +44,6 @@ import           System.Exit                       (exitSuccess)
 import           System.IO                         (hPutStrLn, stderr)
 import           System.Info                       (compilerVersion)
 
-#ifdef MONITORING_EKG
-import qualified Development.IDE.Monitoring.OpenTelemetry as OpenTelemetry
-import qualified Development.IDE.Monitoring.EKG as EKG
-#endif
-
 data Log
   = LogIDEMain IDEMain.Log
   | LogRules Rules.Log
@@ -147,7 +143,5 @@ main = withTelemetryLogger $ \telemetryLogger -> do
                 , optCheckProject = pure $ checkProject config
                 , optRunSubset = not argsConservativeChangeTracking
                 }
-#ifdef MONITORING_EKG
         , IDEMain.argsMonitoring = OpenTelemetry.monitoring <> EKG.monitoring logger argsMonitoringPort
-#endif
         }

ghcide/src/Development/IDE/Main.hs

@@ -1,4 +1,3 @@
-{-# LANGUAGE CPP            #-}
 {-# LANGUAGE PackageImports #-}
 {-# OPTIONS_GHC -Wno-orphans #-}
 module Development.IDE.Main
@@ -63,6 +62,9 @@ import           Development.IDE.LSP.LanguageServer    (runLanguageServer)
 import qualified Development.IDE.LSP.LanguageServer    as LanguageServer
 import           Development.IDE.Main.HeapStats        (withHeapStats)
 import qualified Development.IDE.Main.HeapStats        as HeapStats
+import           Development.IDE.Types.Monitoring      (Monitoring)
+import qualified Development.IDE.Monitoring.EKG as EKG
+import qualified Development.IDE.Monitoring.OpenTelemetry as OpenTelemetry
 import           Development.IDE.Plugin                (Plugin (pluginHandlers, pluginModifyDynflags, pluginRules))
 import           Development.IDE.Plugin.HLS            (asGhcIdePlugin)
 import qualified Development.IDE.Plugin.HLS            as PluginHLS
@@ -130,12 +132,6 @@ import           System.Random                         (newStdGen)
 import           System.Time.Extra                     (Seconds, offsetTime,
                                                         showDuration)
 import           Text.Printf                           (printf)
-import Development.IDE.Types.Monitoring (Monitoring)
-import qualified Development.IDE.Monitoring.OpenTelemetry as OpenTelemetry
-
-#ifdef MONITORING_EKG
-import qualified Development.IDE.Monitoring.EKG as EKG
-#endif
 
 data Log
   = LogHeapStats !HeapStats.Log
@@ -275,10 +271,7 @@ defaultArguments recorder logger = Arguments
                 -- the language server tests without the redirection.
                 putStr " " >> hFlush stdout
                 return newStdout
-        , argsMonitoring = OpenTelemetry.monitoring
-#ifdef MONITORING_EKG
-                        <> EKG.monitoring logger 8999
-#endif
+        , argsMonitoring = OpenTelemetry.monitoring <> EKG.monitoring logger 8999
         }
 
 

ghcide/src/Development/IDE/Monitoring/EKG.hs

@@ -1,12 +1,16 @@
+{-# LANGUAGE CPP #-}
 module Development.IDE.Monitoring.EKG(monitoring) where
+
+import           Development.IDE.Types.Monitoring (Monitoring (..))
+import           Development.IDE.Types.Logger     (Logger)
+#ifdef EKG
 import           Control.Concurrent               (killThread)
 import           Control.Concurrent.Async         (async, waitCatch)
 import           Control.Monad                    (forM_)
 import           Data.Text                        (pack)
-import           Development.IDE.Types.Logger     (Logger, logInfo)
-import           Development.IDE.Types.Monitoring (Monitoring (..))
-import qualified System.Metrics                   as Monitoring
+import           Development.IDE.Types.Logger     (logInfo)
 import qualified System.Remote.Monitoring.Wai     as Monitoring
+import qualified System.Metrics                   as Monitoring
 
 -- | Monitoring using EKG
 monitoring :: Logger -> Int -> IO Monitoring
@@ -35,3 +39,10 @@ monitoring logger port = do
                 logInfo logger "Stopping monitoring server"
                 killThread $ Monitoring.serverThreadId s
     return $ Monitoring {..}
+
+#else
+
+monitoring :: Logger -> Int -> IO Monitoring
+monitoring _ _ = mempty
+
+#endif