Add custom beam/postgres operators

Author: vinit-j13

lib/mobility-core/src/Kernel/Storage/ClickhouseV2/Internal/Types.hs

@@ -57,6 +57,9 @@ data Column (a :: IsAggregated) t v where
   Distinct :: (ClickhouseTable t, ClickhouseValue v) => Column a t v -> Column a t v -- should not be used in where clause
   Max :: (ClickhouseTable t, ClickhouseValue v) => Column a t v -> Column a t v
   Add :: (ClickhouseTable t, ClickhouseNum v) => Column a t v -> Column a t v -> Column a t v
+  Sub :: (ClickhouseTable t, ClickhouseNum v) => Column a t v -> Column a t v -> Column a t v
+  Mul :: (ClickhouseTable t, ClickhouseNum v) => Column a t v -> Column a t v -> Column a t v
+  Div :: (ClickhouseTable t, ClickhouseNum v) => Column a t v -> Column a t v -> Column a t Double
   CoerceNum :: (ClickhouseTable t, ClickhouseNum v1, ClickhouseNum v2) => Column a t v1 -> Column a t v2
   ToDate :: (ClickhouseTable t, ClickhouseValue DateTime, ClickhouseValue Time.Day) => Column a t DateTime -> Column a t Time.Day -- FIXME create some generic constructor for different clickhouse functions
   ToHour :: (ClickhouseTable t, ClickhouseValue DateTime, ClickhouseValue Int) => Column a t DateTime -> Column a t Int
@@ -266,6 +269,9 @@ showColumn (Avg column) = "AVG" <> addBrackets' (showColumn column)
 showColumn (Distinct column) = "DISTINCT" <> addBrackets' (showColumn column)
 showColumn (Max column) = "MAX" <> addBrackets' (showColumn column)
 showColumn (Add column1 column2) = addBrackets' (showColumn column1 <> "+" <> showColumn column2)
+showColumn (Sub column1 column2) = addBrackets' (showColumn column1 <> "-" <> showColumn column2)
+showColumn (Mul column1 column2) = addBrackets' (showColumn column1 <> "*" <> showColumn column2)
+showColumn (Div column1 column2) = addBrackets' (showColumn column1 <> "/" <> showColumn column2)
 showColumn (CoerceNum column) = showColumn column
 showColumn (ToDate column) = "toDate" <> addBrackets' (showColumn column)
 showColumn (ToHour column) = "toHour" <> addBrackets' (showColumn column)

lib/mobility-core/src/Kernel/Storage/ClickhouseV2/Operators.hs

@@ -194,6 +194,21 @@ max = Max
 
 infixl 6 +.
 
+(-.) :: (ClickhouseTable t, ClickhouseNum v) => Column a t v -> Column a t v -> Column a t v
+(-.) = Sub
+
+infixl 6 -.
+
+(*.) :: (ClickhouseTable t, ClickhouseNum v) => Column a t v -> Column a t v -> Column a t v
+(*.) = Mul
+
+infixl 7 *.
+
+(/.) :: (ClickhouseTable t, ClickhouseNum v) => Column a t v -> Column a t v -> Column a t Double
+(/.) = Div
+
+infixl 7 /.
+
 -- for example we can use for columns sum with different types
 unsafeCoerceNum :: forall v1 v2 a t. (ClickhouseTable t, ClickhouseNum v1, ClickhouseNum v2) => Column a t v1 -> Column a t v2
 unsafeCoerceNum = CoerceNum