Skip to content

Commit 9d7a92a

Browse files
pfcoperezjlopezmalla
authored andcommitted
Reduce method complexity in order to improve its readability & performance (apache#63)
1 parent 67f274f commit 9d7a92a

File tree

1 file changed

+8
-22
lines changed

1 file changed

+8
-22
lines changed

core/src/main/scala/org/apache/spark/security/ConfigSecurity.scala

Lines changed: 8 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -102,30 +102,16 @@ object ConfigSecurity extends Logging{
102102
}
103103
}
104104

105-
106105
private def prepareEnvironment(vaultHost: String,
107106
vaultToken: String,
108107
secretOptions: Map[String,
109-
Map[String, String]]): Map[String, String] = {
110-
val seqOp: (Map[String, String], (String, Map[String, String])) => Map[String, String] =
111-
(agg: Map[String, String], value: (String, Map[String, String])) => {
112-
val (key, options) = value
113-
val secretOptions = key match {
114-
case "kerberos" => KerberosConfig.prepareEnviroment(vaultHost,
115-
vaultToken,
116-
options)
117-
case "datastore" => SSLConfig.prepareEnvironment(vaultHost,
118-
vaultToken,
119-
SSLConfig.sslTypeDataStore,
120-
options)
121-
case _ => Map[String, String]()
122-
}
123-
secretOptions ++ agg
124-
}
125-
val combOp: (Map[String, String], Map[String, String]) => Map[String, String] =
126-
(agg1: Map[String, String], agg2: Map[String, String]) => agg1 ++ agg2
127-
128-
secretOptions.aggregate(Map[String, String]())(seqOp, combOp)
129-
}
108+
Map[String, String]]): Map[String, String] =
109+
secretOptions flatMap {
110+
case ("kerberos", options) =>
111+
KerberosConfig.prepareEnviroment(vaultHost, vaultToken, options)
112+
case ("datastore", options) =>
113+
SSLConfig.prepareEnvironment(vaultHost, vaultToken, SSLConfig.sslTypeDataStore, options)
114+
case _ => Map.empty[String, String]
115+
}
130116

131117
}

0 commit comments

Comments
 (0)