{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE QuasiQuotes #-}

module PGExtras.Queries.LongRunningQueries (longRunningQueriesSQL, displayLongRunningQueries) where

import PGExtras.Helpers (maybeInt, maybeText, maybeZonedTime)
import Database.PostgreSQL.Simple
import Text.RawString.QQ
import qualified Data.Text as Text
import Control.Monad (forM_)
import Data.List (intercalate)
import Data.Time (ZonedTime)

longRunningQueriesSQL :: Query
longRunningQueriesSQL :: Query
longRunningQueriesSQL = [r|SELECT
  pid,
  pg_stat_activity.query_start,
  query AS query
FROM
  pg_stat_activity
WHERE
  pg_stat_activity.query <> ''::text
  AND state <> 'idle'
  AND now() - pg_stat_activity.query_start > interval '5 minutes'
ORDER BY
  now() - pg_stat_activity.query_start DESC;|]

displayLongRunningQueries :: [(Maybe Int, Maybe ZonedTime, Maybe Text.Text)] -> IO ()
displayLongRunningQueries :: [(Maybe Int, Maybe ZonedTime, Maybe Text)] -> IO ()
displayLongRunningQueries rows :: [(Maybe Int, Maybe ZonedTime, Maybe Text)]
rows = do
  String -> IO ()
putStrLn (String -> IO ()) -> String -> IO ()
forall a b. (a -> b) -> a -> b
$ String
description
  String -> IO ()
putStrLn (String -> IO ()) -> String -> IO ()
forall a b. (a -> b) -> a -> b
$ String -> [String] -> String
forall a. [a] -> [[a]] -> [a]
intercalate " | " [String]
tableHeaders
  [(Maybe Int, Maybe ZonedTime, Maybe Text)]
-> ((Maybe Int, Maybe ZonedTime, Maybe Text) -> IO ()) -> IO ()
forall (t :: * -> *) (m :: * -> *) a b.
(Foldable t, Monad m) =>
t a -> (a -> m b) -> m ()
forM_ [(Maybe Int, Maybe ZonedTime, Maybe Text)]
rows (((Maybe Int, Maybe ZonedTime, Maybe Text) -> IO ()) -> IO ())
-> ((Maybe Int, Maybe ZonedTime, Maybe Text) -> IO ()) -> IO ()
forall a b. (a -> b) -> a -> b
$ \(arg1 :: Maybe Int
arg1, arg2 :: Maybe ZonedTime
arg2, arg3 :: Maybe Text
arg3) ->
    String -> IO ()
putStrLn (String -> IO ()) -> String -> IO ()
forall a b. (a -> b) -> a -> b
$ Maybe Int -> String
maybeInt(Maybe Int
arg1) String -> String -> String
forall a. [a] -> [a] -> [a]
++ " | " String -> String -> String
forall a. [a] -> [a] -> [a]
++ Maybe ZonedTime -> String
maybeZonedTime(Maybe ZonedTime
arg2) String -> String -> String
forall a. [a] -> [a] -> [a]
++ " | " String -> String -> String
forall a. [a] -> [a] -> [a]
++ Maybe Text -> String
maybeText(Maybe Text
arg3)

description :: [Char]
description :: String
description = "All queries longer than five minutes by descending duration"

tableHeaders :: [[Char]]
tableHeaders :: [String]
tableHeaders = ["pid", "query_start", "query"]