Synnefo » snf-ganeti

import qualified Data.Attoparsec.Text as A

import qualified Data.Attoparsec.Combinator as AC

import Data.Attoparsec.Text (Parser)

import Data.Maybe

import Data.Text (Text, unpack)

optional parser = (Just <$> parser) <|> pure Nothing

-- | The parser for a whole DRBD status file.

  DRBDStatus <$> versionInfoParser

             <* A.endOfInput

-- | The parser for the version information lines.

              <* A.endOfLine

-- | The parser for a (multi-line) string representing a device.

  deviceNum <- skipSpaces *> A.decimal <* A.char ':'

  cs <- skipSpacesAndString "cs:" connStateParser

  if cs == Unconfigured

      reS <- optional resyncParser

      act <- optional actLogParser

      _ <- additionalEOL

      return $ DeviceInfo deviceNum cs ro ds replicProtocol io pIndicators

    where conditionalSyncStatusParser SyncSource = Just <$> syncStatusParser

          conditionalSyncStatusParser SyncTarget = Just <$> syncStatusParser

  , Time(..)

  , TimeUnit(..)

  , AdditionalInfo(..)

  ) where

import Text.JSON

import Text.Printf

  UnconfiguredDevice Int -- ^ An DRBD minor marked as unconfigured

  | -- | A configured DRBD minor

    DeviceInfo

      , diskStates :: LocalRemote DiskState -- ^ Status of the disks

      } deriving (Eq, Show)

-- | The DeviceInfo instance of JSON.

    ]

  showJSON (DeviceInfo minorNumberF connectionStateF (LocalRemote

    localRole remoteRole) (LocalRemote localState remoteState)

    optFieldsToObj

    [ Just ("minor", showJSON minorNumberF)

    , Just ("connectionState", showJSON connectionStateF)

    , Just ("ioFlags", showJSON ioFlagsF)

    , Just ("perfIndicators", showJSON perfIndicatorsF)

    , optionalJSField "syncStatus" syncStatusF

    ]

  readJSON = error "JSON read instance not implemented for type DeviceInfo"

  , dirty       :: Int

  , changed     :: Int

  } deriving (Eq, Show)

    ((E.try $ readFile proc_drbd) :: IO (Either IOError String)) >>=

      exitIfBad "reading from file" . either (BT.Bad . show) BT.Ok

  output <-

      A.Fail unparsedText contexts errorMessage -> exitErr $

        show (Prelude.take defaultCharNum $ unpack unparsedText) ++ "\n"

          ++ show contexts ++ "\n" ++ errorMessage

testFile :: String -> DRBDStatus -> Assertion

testFile fileName expectedContent = do

    fileContent <- readTestData fileName

        Left msg -> assertFailure $ "Parsing failed: " ++ msg

        Right obtained -> assertEqual fileName expectedContent obtained

          Nothing Nothing Nothing)

        Nothing

        (Just $ AdditionalInfo 0 61 65657 135 0 0 135)

      UnconfiguredDevice 1,

      UnconfiguredDevice 2,

      UnconfiguredDevice 5,

          (Just 'b') (Just 0))

        Nothing

        Nothing

        Nothing,

      DeviceInfo 1 Connected (LocalRemote Secondary Primary)

        (LocalRemote UpToDate UpToDate) 'C' "r---"

          (Just 0))

        Nothing

        Nothing

        Nothing,

      UnconfiguredDevice 2,

      DeviceInfo 3 SyncSource (LocalRemote Primary Secondary)

        (Just $ SyncStatus 2.4 14996 15360 MegaByte (Time 0 4 8) 61736 Nothing

          KiloByte Second)

        Nothing

        Nothing,

      DeviceInfo 4 WFConnection (LocalRemote Primary Unknown)

        (LocalRemote UpToDate DUnknown) 'C' "r----"

        Nothing

        Nothing

        Nothing

    ]

-- | Test a DRBD 8.3 file with an ongoing synchronization.

          (Just 'b') (Just 0))

        Nothing

        Nothing

        Nothing,

      DeviceInfo 1 Connected (LocalRemote Secondary Primary)

        (LocalRemote UpToDate UpToDate) 'C' "r---"

          (Just 0))

        Nothing

        Nothing

        Nothing,

      UnconfiguredDevice 2,

      DeviceInfo 3 SyncTarget (LocalRemote Primary Secondary)

        (Just $ SyncStatus 34.9 346112 524288 MegaByte (Time 0 0 5) 59392

          Nothing KiloByte Second)

        Nothing

        Nothing,

      DeviceInfo 4 WFConnection (LocalRemote Primary Unknown)

        (LocalRemote UpToDate DUnknown) 'C' "r----"

        Nothing

        Nothing

        Nothing

    ]

-- | Test a DRBD 8.3 file not from git sources, with an ongoing synchronization

        (Just $ SyncStatus 44.4 588416 1048576 KiloByte (Time 0 0 8) 65736

          (Just 61440) KiloByte Second)

        Nothing

        Nothing,

      UnconfiguredDevice 1,

      UnconfiguredDevice 2,

          (Just 'b') (Just 0))

        Nothing

        Nothing

        Nothing,

      DeviceInfo 1 Connected (LocalRemote Secondary Primary)

        (LocalRemote UpToDate UpToDate) 'C' "r---"

          (Just 0))

        Nothing

        Nothing

        Nothing,

      UnconfiguredDevice 2,

      DeviceInfo 4 WFConnection (LocalRemote Primary Unknown)

          (Just 'b') (Just 0))

        Nothing

        Nothing

        Nothing,

      DeviceInfo 5 Connected (LocalRemote Primary Secondary)

        (LocalRemote UpToDate Diskless) 'C' "r----"

          (Just 'b') (Just 0))

        Nothing

        Nothing

        Nothing,

      DeviceInfo 6 Connected (LocalRemote Secondary Primary)

        (LocalRemote Diskless UpToDate) 'C' "r---"

          (Just 0))

        Nothing

        Nothing

        Nothing,

      DeviceInfo 7 WFConnection (LocalRemote Secondary Unknown)

        (LocalRemote UpToDate DUnknown) 'C' "r---"

          (Just 0))

        Nothing

        Nothing

        Nothing,

      DeviceInfo 8 StandAlone (LocalRemote Secondary Unknown)

        (LocalRemote UpToDate DUnknown) ' ' "r---"

        Nothing

        Nothing

        Nothing

    ]

-- | Test a DRBD 8.0 file with a missing device.

          Nothing Nothing)

        Nothing

        (Just $ AdditionalInfo 0 61 0 0 0 0 0)

      DeviceInfo 1 Connected (LocalRemote Secondary Primary)

        (LocalRemote UpToDate UpToDate) 'C' "r---"

        (PerfIndicators 738320 0 738320 554400 67 0 0 0 0 0 Nothing

          Nothing Nothing)

        Nothing

        (Just $ AdditionalInfo 0 61 0 0 0 0 0)

      UnconfiguredDevice 2,

      DeviceInfo 4 WFConnection (LocalRemote Primary Unknown)

        (LocalRemote UpToDate DUnknown) 'C' "r---"

          Nothing Nothing)

        Nothing

        (Just $ AdditionalInfo 0 61 0 0 0 0 0)

      DeviceInfo 5 Connected (LocalRemote Primary Secondary)

        (LocalRemote UpToDate Diskless) 'C' "r---"

        (PerfIndicators 4375581 0 4446283 674 1069 69 0 0 0 0 Nothing

          Nothing Nothing)

        Nothing

        (Just $ AdditionalInfo 0 61 0 0 0 0 0)

      DeviceInfo 6 Connected (LocalRemote Secondary Primary)

        (LocalRemote Diskless UpToDate) 'C'  "r---"

        (PerfIndicators 0 4375581 5186925 327 75 214 0 0 0 0 Nothing

          Nothing Nothing)

        Nothing

        Nothing

        Nothing,

      DeviceInfo 7 WFConnection (LocalRemote Secondary Unknown)

        (LocalRemote UpToDate DUnknown) 'C' "r---"

        (PerfIndicators 0 0 0 0 0 0 0 0 0 0 Nothing Nothing Nothing)

        Nothing

        (Just $ AdditionalInfo 0 61 0 0 0 0 0)

      DeviceInfo 8 StandAlone (LocalRemote Secondary Unknown)

        (LocalRemote UpToDate DUnknown) ' ' "r---"

        (PerfIndicators 0 0 0 0 0 0 0 0 0 0 Nothing Nothing Nothing)

        Nothing

        (Just $ AdditionalInfo 0 61 0 0 0 0 0)

        (Just $ AdditionalInfo 0 257 0 0 0 0 0)

    ]

-- | Function for splitting a list in chunks of a given size.

  ep <- genMaybe natural

  wo <- genMaybe wOrderFlag

  oos <- genMaybe natural

  let obtained =

          showJSON $

            DeviceInfo minor state (LocalRemote locRole remRole)

              Nothing

              Nothing

              Nothing

      perfInd =

        PerfIndicators ns nr dw dr al bm lc pe ua ap ep wo oos

      expected =

          , ("replicationProtocol", showJSON alg)

          , ("ioFlags", showJSON "r----")

          , ("perfIndicators", showJSON perfInd)

          ]

  obtained ==? expected