Analysis.hs

-- Copyright (c) 2013-2014 PivotCloud, Inc.
--
-- Aws.Kinesis.Reshard.Analysis
--
-- Please feel free to contact us at licensing@pivotmail.com with any
-- contributions, additions, or other feedback; we would love to hear from
-- you.
--
-- Licensed under the Apache License, Version 2.0 (the "License"); you may
-- not use this file except in compliance with the License. You may obtain a
-- copy of the License at http://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
-- WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
-- License for the specific language governing permissions and limitations
-- under the License.
--

{-# LANGUAGE ConstraintKinds #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE MultiWayIf #-}
{-# LANGUAGE UnicodeSyntax #-}

module Aws.Kinesis.Reshard.Analysis
( analyzeStream
) where

import Aws.Kinesis.Reshard.Shards
import Aws.Kinesis.Reshard.Metrics
import Aws.Kinesis.Reshard.Monad
import Control.Concurrent.Async.Lifted
import Control.Lens
import Prelude.Unicode

analyzeStream
  ∷ MonadReshard m
  ⇒ m (Maybe ReshardingAction)
analyzeStream = do
  (totalBps, shardCount) ← getBytesPerSecond `concurrently` countOpenShards
  threshold ← view oShardCapacityThreshold
  maximumShardCount ← view oMaximumShardCount
  let percentCapacity = shardlyBps / maximumBps
      maximumBps = 1000000
      shardlyBps = totalBps / fromIntegral shardCount
  return $
    if | shardCount > maximumShardCount → Just MergeShardsAction
       | percentCapacity < threshold * 0.5 ∧ shardCount > 1 → Just MergeShardsAction
       | percentCapacity ≥ threshold ∧ shardCount < maximumShardCount → Just SplitShardsAction
       | otherwise → Nothing