From 3cb2db4295daf3e4ed09d13b8c1df9fb9b9c22f2 Mon Sep 17 00:00:00 2001 From: takapi327 Date: Mon, 3 Feb 2025 21:47:01 +0900 Subject: [PATCH] Added codeToTypeMap --- .../main/scala/ldbc/connector/data/ColumnDataType.scala | 7 ++++--- 1 file changed, 4 insertions(+), 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 index 4996f6185..34b3fe07d 100644 --- 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"))