feat: Add helpers to BQField

nrkno · Jun 15, 2023 · 7196a85 · 7196a85
1 parent 89f6b1c
commit 7196a85
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/core/src/main/scala/no/nrk/bigquery/BQField.scala b/core/src/main/scala/no/nrk/bigquery/BQField.scala
@@ -39,6 +39,19 @@ case class BQField(
       mode = if (mode == Field.Mode.REQUIRED) Field.Mode.NULLABLE else mode,
       subFields = subFields.map(_.recursivelyNullable)
     )
+
+  def withName(newName: String) = copy(name = newName)
+  def required = copy(mode = Field.Mode.REQUIRED)
+
+  def withType(newType: StandardSQLTypeName) = copy(tpe = newType)
+
+  def addSubFields(newSubFields: BQField*) =
+    withSubFields(subFields ::: newSubFields.toList)
+  def withSubFields(newSubFields: List[BQField]) =
+    if (tpe == StandardSQLTypeName.STRUCT) copy(subFields = newSubFields) else this
+
+  def addPolicyTags(newTags: String*) = copy(policyTags = policyTags ::: newTags.toList)
+  def withPolicyTags(newTags: List[String]) = copy(policyTags = newTags)
 }
 
 object BQField {