Added codeToFlagMap

takapi327 · Feb 3, 2025 · a562a20 · a562a20
1 parent 3cb2db4
commit a562a20
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/module/ldbc-connector/shared/src/main/scala/ldbc/connector/data/ColumnDefinitionFlags.scala b/module/ldbc-connector/shared/src/main/scala/ldbc/connector/data/ColumnDefinitionFlags.scala
@@ -47,13 +47,18 @@ enum ColumnDefinitionFlags(val code: Long):
 
 object ColumnDefinitionFlags:
 
+  private val codeToFlagMap: Map[Long, ColumnDefinitionFlags] =
+    ColumnDefinitionFlags.values.map(flag => flag.code -> flag).toMap
+
   /** Get bitset objects from numeric bitset. */
   def apply(bitset: Long): Seq[ColumnDefinitionFlags] =
-    toEnumSeq(bitset)
+    codeToFlagMap.collect {
+      case (code, flag) if (bitset & code) == code => flag
+    }.toSeq
 
   /** Get bitset objects from numeric bitsets. */
   def apply(bitset: Seq[Short]): Seq[ColumnDefinitionFlags] =
-    bitset.flatMap(b => toEnumSeq(toCode(b)))
+    bitset.flatMap(b => codeToFlagMap.get(toCode(b)))
 
   /** Convert bitNum to BitFlag numbers */
   def toCode(bitNum: Short): Long =