module Aws.Kinesis.Reshard.Metrics
( getBytesPerSecond
) where
import AWS
import AWS.CloudWatch
import AWS.CloudWatch.Types
import Aws.Kinesis.Reshard.Monad
import Control.Applicative
import Control.Applicative.Unicode
import Control.Concurrent.Async.Lifted
import Control.Lens
import Control.Monad.Trans
import Control.Monad.Unicode
import qualified Data.Text.Encoding as T
import Data.Time
import Prelude.Unicode
getCredential
∷ MonadReshard m
⇒ m Credential
getCredential =
pure newCredential
⊛ view (oAccessKey ∘ to T.encodeUtf8)
⊛ view (oSecretAccessKey ∘ to T.encodeUtf8)
dimensionFilters
∷ MonadReshard m
⇒ m [DimensionFilter]
dimensionFilters =
(:[]) ∘ ("StreamName",)
<$> view oStreamName
getBytesPerSecond
∷ MonadReshard m
⇒ m Double
getBytesPerSecond = do
cred ← getCredential
filters ← dimensionFilters
duration ← view oSampleDuration
runCloudWatch cred $ do
setRegion =≪ lift (view oRegion)
endTime ← liftIO getCurrentTime
let startTime = addUTCTime (fromInteger $ duration) endTime
let getMetrics metric = do
(datapoints, _) ← getMetricStatistics
filters
startTime
endTime
metric
"AWS/Kinesis"
(fromIntegral duration)
[StatisticSum]
Nothing
return $ foldr (\Datapoint{..} n → maybe n (+ n) datapointSum) 0 datapoints
totalBytes ← runConcurrently $
pure (+)
⊛ Concurrently (getMetrics "PutRecord.Bytes")
⊛ Concurrently (getMetrics "PutRecords.Bytes")
return $ totalBytes / fromIntegral duration