Added codeToTypeMap

takapi327 · Feb 3, 2025 · 3cb2db4 · 3cb2db4
1 parent a1c8284
commit 3cb2db4
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/module/ldbc-connector/shared/src/main/scala/ldbc/connector/data/ColumnDataType.scala b/module/ldbc-connector/shared/src/main/scala/ldbc/connector/data/ColumnDataType.scala
@@ -53,7 +53,8 @@ enum ColumnDataType(val code: Long, val name: String):
 
 object ColumnDataType:
 
+  private val codeToTypeMap: Map[Long, ColumnDataType] =
+    ColumnDataType.values.map(t => t.code -> t).toMap
+
   def apply(code: Long): ColumnDataType =
-    ColumnDataType.values
-      .find(_.code == code)
-      .getOrElse(throw new IllegalArgumentException(s"Unknown column data type code: $code"))
+    codeToTypeMap.getOrElse(code, throw new IllegalArgumentException(s"Unknown column data type code: $code"))