Merge pull request #12 from Ecwid/bool_type

Add support for bool type
Ecwid · Nov 30, 2023 · 011c7ad · 011c7ad
2 parents 321edbd + 0653bcd
commit 011c7ad
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 0 deletions.
diff --git a/src/main/kotlin/com/ecwid/clickhouse/convert/Convert.kt b/src/main/kotlin/com/ecwid/clickhouse/convert/Convert.kt
@@ -6,6 +6,32 @@ import java.util.*
 
 object Convert {
 
+	object Bool {
+		@JvmStatic
+		fun toValue(str: String?) = requireNotNull(str).toBoolean()
+
+		@JvmStatic
+		fun toNullableValue(str: String?) = str?.toBoolean()
+
+		@JvmStatic
+		fun toArray(array: List<String?>) = array.map(::toValue)
+
+		@JvmStatic
+		fun toNullableArray(array: List<String?>) = array.map(::toNullableValue)
+
+		@JvmStatic
+		fun fromValue(value: Boolean) = value.toString()
+
+		@JvmStatic
+		fun fromNullableValue(value: Boolean?) = value?.toString()
+
+		@JvmStatic
+		fun fromArray(array: List<Boolean>) = array.map(::fromValue)
+
+		@JvmStatic
+		fun fromNullableArray(array: List<Boolean?>) = array.map(::fromNullableValue)
+	}
+
 	object Int8 {
 		@JvmStatic
 		fun toValue(str: String?) = requireNotNull(str).toByte()

diff --git a/src/main/kotlin/com/ecwid/clickhouse/typed/TypedRow.kt b/src/main/kotlin/com/ecwid/clickhouse/typed/TypedRow.kt
@@ -12,6 +12,13 @@ data class TypedRow(
 
 	fun getMeta() = rawRow.getMeta()
 
+	// ----------------- Bool ---------------------
+
+	fun getBool(columnIndex: Int): Boolean {
+		val scalar = rawRow.getScalarValue(columnIndex)
+		return Convert.Bool.toValue(scalar)
+	}
+
 	// ----------------- INT_8 --------------------
 	fun getInt8(columnIndex: Int): Byte {
 		val scalar = rawRow.getScalarValue(columnIndex)