2016-01-01 20:57:54 +05:30
|
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
{-# LANGUAGE RecordWildCards #-}
|
2015-12-29 17:11:15 +05:30
|
|
|
module Ringo.Generator.Internal where
|
|
|
|
|
|
|
|
import qualified Data.Map as Map
|
|
|
|
import qualified Data.Text as Text
|
|
|
|
|
|
|
|
import Data.List (find)
|
|
|
|
import Data.Monoid ((<>))
|
|
|
|
import Data.Text (Text)
|
|
|
|
|
|
|
|
import Ringo.Extractor.Internal
|
|
|
|
import Ringo.Types
|
|
|
|
|
|
|
|
joinColumnNames :: [ColumnName] -> Text
|
|
|
|
joinColumnNames = Text.intercalate ",\n"
|
|
|
|
|
|
|
|
fullColumnName :: TableName -> ColumnName -> ColumnName
|
|
|
|
fullColumnName tName cName = tName <> "." <> cName
|
|
|
|
|
|
|
|
dimColumnMapping :: Text -> Fact -> TableName -> [(ColumnName, ColumnName)]
|
|
|
|
dimColumnMapping dimPrefix fact dimTableName =
|
|
|
|
[ (dimColumnName dName cName, cName)
|
2015-12-30 12:21:41 +05:30
|
|
|
| DimVal dName cName <- factColumns fact
|
|
|
|
, dimPrefix <> dName == dimTableName ]
|
2015-12-29 17:11:15 +05:30
|
|
|
|
|
|
|
coalesceColumn :: TypeDefaults -> TableName -> Column -> Text
|
|
|
|
coalesceColumn defaults tName Column{..} =
|
|
|
|
if columnNullable == Null
|
|
|
|
then "coalesce(" <> fqColName <> ", " <> defVal columnType <> ")"
|
|
|
|
else fqColName
|
|
|
|
where
|
|
|
|
fqColName = fullColumnName tName columnName
|
|
|
|
|
|
|
|
defVal colType =
|
2015-12-29 18:22:01 +05:30
|
|
|
maybe (error $ "Default value not known for column type: " ++ Text.unpack colType) snd
|
2015-12-29 17:11:15 +05:30
|
|
|
. find (\(k, _) -> k `Text.isPrefixOf` colType)
|
|
|
|
. Map.toList
|
|
|
|
$ defaults
|
2015-12-30 19:57:38 +05:30
|
|
|
|
|
|
|
suffixTableName :: TablePopulationMode -> Text -> TableName -> TableName
|
|
|
|
suffixTableName popMode suffix tableName = case popMode of
|
|
|
|
FullPopulation -> tableName <> suffix
|
|
|
|
IncrementalPopulation -> tableName
|