Add two apply methods which will be used to extract StructField(s) fr…

…om a StructType.
apache · Jul 11, 2014 · eca7d04 · eca7d04
1 parent 949d6bb
commit eca7d04
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/types/dataTypes.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/types/dataTypes.scala
@@ -292,6 +292,15 @@ object StructType {
 case class StructType(fields: Seq[StructField]) extends DataType {
   require(StructType.validateFields(fields), "Found fields with the same name.")
 
+  def apply(name: String): StructField = {
+    fields.find(f => f.name == name).orNull
+  }
+
+  def apply(names: String*): StructType = {
+    val nameSet = names.toSet
+    StructType(fields.filter(f => nameSet.contains(f.name)))
+  }
+
   def toAttributes = fields.map(f => AttributeReference(f.name, f.dataType, f.nullable)())
 
   def schemaString: String = {