119 lines
4.6 KiB
Haskell
119 lines
4.6 KiB
Haskell
module Ringo.Types where
|
|
|
|
import qualified Data.Text as Text
|
|
|
|
import Data.Map (Map)
|
|
import Data.Text (Text)
|
|
|
|
type ColumnName = Text
|
|
type ColumnType = Text
|
|
type TableName = Text
|
|
|
|
data Nullable = Null | NotNull deriving (Eq, Enum, Show)
|
|
|
|
data Column = Column
|
|
{ columnName :: !ColumnName
|
|
, columnType :: !ColumnType
|
|
, columnNullable :: !Nullable
|
|
} deriving (Eq, Show)
|
|
|
|
data TableConstraint = PrimaryKey !ColumnName
|
|
| UniqueKey ![ColumnName]
|
|
| ForeignKey !TableName ![(ColumnName, ColumnName)]
|
|
deriving (Eq, Show)
|
|
|
|
data Table = Table
|
|
{ tableName :: !TableName
|
|
, tableColumns :: ![Column]
|
|
, tableConstraints :: ![TableConstraint]
|
|
} deriving (Eq, Show)
|
|
|
|
data TimeUnit = Second | Minute | Hour | Day | Week
|
|
deriving (Eq, Enum, Show, Read)
|
|
|
|
timeUnitName :: TimeUnit -> Text
|
|
timeUnitName = Text.toLower . Text.pack . show
|
|
|
|
timeUnitToSeconds :: TimeUnit -> Int
|
|
timeUnitToSeconds Second = 1
|
|
timeUnitToSeconds Minute = 60 * timeUnitToSeconds Second
|
|
timeUnitToSeconds Hour = 60 * timeUnitToSeconds Minute
|
|
timeUnitToSeconds Day = 24 * timeUnitToSeconds Hour
|
|
timeUnitToSeconds Week = 7 * timeUnitToSeconds Day
|
|
|
|
data Fact = Fact
|
|
{ factName :: !TableName
|
|
, factTableName :: !TableName
|
|
, factParentNames :: ![TableName]
|
|
, factColumns :: ![FactColumn]
|
|
} deriving (Eq, Show)
|
|
|
|
data FactColumn = DimTime !ColumnName
|
|
| NoDimId !ColumnName
|
|
| DimId !TableName !ColumnName
|
|
| DimVal !TableName !ColumnName
|
|
| FactCount !ColumnName
|
|
| FactSum !ColumnName !ColumnName
|
|
| FactAverage !ColumnName !ColumnName
|
|
| FactCountDistinct !ColumnName
|
|
deriving (Eq, Show)
|
|
|
|
factColumnName :: FactColumn -> Maybe ColumnName
|
|
factColumnName (DimTime cName) = Just cName
|
|
factColumnName (NoDimId cName) = Just cName
|
|
factColumnName (DimId _ cName) = Just cName
|
|
factColumnName (DimVal _ cName) = Just cName
|
|
factColumnName (FactCount _) = Nothing
|
|
factColumnName (FactSum cName _) = Just cName
|
|
factColumnName (FactAverage cName _) = Just cName
|
|
factColumnName (FactCountDistinct _) = Nothing
|
|
|
|
data Settings = Settings
|
|
{ settingDimPrefix :: !Text
|
|
, settingFactPrefix :: !Text
|
|
, settingTimeUnit :: !TimeUnit
|
|
, settingAvgCountColumSuffix :: !Text
|
|
, settingAvgSumColumnSuffix :: !Text
|
|
, settingCountDistinctColumSuffix :: !Text
|
|
, settingDimTableIdColumnName :: !Text
|
|
, settingDimTableIdColumnType :: !Text
|
|
, settingFactCountColumnType :: !Text
|
|
, settingFactInfix :: !Text
|
|
, settingDependenciesJSONFileName :: !Text
|
|
, settingFactsJSONFileName :: !Text
|
|
, settingDimensionJSONFileName :: !Text
|
|
} deriving (Eq, Show)
|
|
|
|
defSettings :: Settings
|
|
defSettings = Settings
|
|
{ settingDimPrefix = "dim_"
|
|
, settingFactPrefix = "fact_"
|
|
, settingTimeUnit = Minute
|
|
, settingAvgCountColumSuffix = "_count"
|
|
, settingAvgSumColumnSuffix = "_sum"
|
|
, settingCountDistinctColumSuffix = "_hll"
|
|
, settingDimTableIdColumnName = "id"
|
|
, settingDimTableIdColumnType = "serial"
|
|
, settingFactCountColumnType = "integer"
|
|
, settingFactInfix = "_by_"
|
|
, settingDependenciesJSONFileName = "dependencies.json"
|
|
, settingFactsJSONFileName = "facts.json"
|
|
, settingDimensionJSONFileName = "dimensions.json"
|
|
}
|
|
|
|
data ValidationError = MissingTable !TableName
|
|
| MissingFact !TableName
|
|
| MissingColumn !TableName !ColumnName
|
|
| MissingTimeColumn !TableName
|
|
deriving (Eq, Show)
|
|
|
|
data Env = Env
|
|
{ envTables :: ![Table]
|
|
, envFacts :: ![Fact]
|
|
, envSettings :: !Settings
|
|
} deriving (Eq, Show)
|
|
|
|
data TablePopulationMode = FullPopulation | IncrementalPopulation deriving (Eq, Show)
|
|
|
|
type Dependencies = Map TableName [TableName]
|