set revision to 0 if empty

simerplaha · Sep 8, 2020 · 3bf7049 · 3bf7049
1 parent c199129
commit 3bf7049
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/macros/src/main/scala/swaydb/macros/VersionReader.scala b/macros/src/main/scala/swaydb/macros/VersionReader.scala
@@ -83,15 +83,18 @@ object VersionReader {
       }
 
     //get major.minor.revision from version.
-    val regex = """(\d+)\.(\d+)\.(\d+)""".r
+    val regex = """(\d+)\.(\d+)\.?(\d+)?""".r
 
     val (major, minor, revision) =
       regex.findFirstMatchIn(versionString) match {
         case Some(value) =>
           val major = value.group(1)
           val minor = value.group(2)
           val revision = value.group(3)
-          (major.toInt, minor.toInt, revision.toInt)
+          if (revision == null)
+            (major.toInt, minor.toInt, 0)
+          else
+            (major.toInt, minor.toInt, revision.toInt)
 
         case None =>
           c.abort(