zio#657 - Add uuid() function to MySqlModule

vidyasankarv · May 26, 2022 · a15889e · a15889e
1 parent be1f30a
commit a15889e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/mysql/src/main/scala/zio/sql/mysql/MysqlSqlModule.scala b/mysql/src/main/scala/zio/sql/mysql/MysqlSqlModule.scala
@@ -2,6 +2,7 @@ package zio.sql.mysql
 
 import java.sql.ResultSet
 import java.time.{ LocalDate, LocalTime, OffsetTime, Year, ZonedDateTime }
+import java.util.UUID
 import zio.sql.Sql
 
 trait MysqlSqlModule extends Sql { self =>
@@ -21,7 +22,6 @@ trait MysqlSqlModule extends Sql { self =>
               r => Right(r)
             )
       }
-
     }
   }
 
@@ -38,6 +38,6 @@ trait MysqlSqlModule extends Sql { self =>
     val Now         = FunctionDef[Any, ZonedDateTime](FunctionName("now"))
     val Pi          = Expr.FunctionCall0[Double](FunctionDef[Any, Double](FunctionName("pi")))
     val RPad        = FunctionDef[(String, Int, String), String](FunctionName("rpad"))
+    val Uuid        = Expr.FunctionCall0[UUID](FunctionDef[Any, UUID](FunctionName("uuid")))
   }
-
 }
diff --git a/mysql/src/test/scala/zio/sql/mysql/FunctionDefSpec.scala b/mysql/src/test/scala/zio/sql/mysql/FunctionDefSpec.scala
@@ -148,6 +148,9 @@ object FunctionDefSpec extends MysqlRunnableSpec with ShopSchema {
 
       assertZIO(testResult.runHead.some)(equalTo(expected))
     },
+    test("uuid") {
+      assertZIO(execute(select(Uuid)).runHead.some)(!isNull)
+    },
     test("rpad") {
       val cases = Seq(("hi", 5, "?", "hi???"), ("hi", 1, "?", "h"))
       check(Gen.fromIterable(cases)) { case (str, len, pad, exp) =>