Pass only specified options to spark dependencies (#708)

* Pass only specified options to spark dependencies Signed-off-by: Pavol Loffay <ploffay@redhat.com> * generate Signed-off-by: Pavol Loffay <ploffay@redhat.com> * fmt Signed-off-by: Pavol Loffay <ploffay@redhat.com>
jaegertracing · Oct 18, 2019 · 01e39a3 · 01e39a3
1 parent 3507174
commit 01e39a3
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 6 deletions.
diff --git a/pkg/apis/jaegertracing/v1/jaeger_types.go b/pkg/apis/jaegertracing/v1/jaeger_types.go
@@ -375,10 +375,10 @@ type JaegerDependenciesSpec struct {
 	CassandraClientAuthEnabled bool `json:"cassandraClientAuthEnabled,omitempty"`
 
 	// +optional
-	ElasticsearchClientNodeOnly bool `json:"elasticsearchClientNodeOnly,omitempty"`
+	ElasticsearchClientNodeOnly *bool `json:"elasticsearchClientNodeOnly,omitempty"`
 
 	// +optional
-	ElasticsearchNodesWanOnly bool `json:"elasticsearchNodesWanOnly,omitempty"`
+	ElasticsearchNodesWanOnly *bool `json:"elasticsearchNodesWanOnly,omitempty"`
 
 	// +optional
 	TTLSecondsAfterFinished *int32 `json:"ttlSecondsAfterFinished,omitempty"`

diff --git a/pkg/apis/jaegertracing/v1/zz_generated.deepcopy.go b/pkg/apis/jaegertracing/v1/zz_generated.deepcopy.go
diff --git a/pkg/cronjob/spark_dependencies.go b/pkg/cronjob/spark_dependencies.go
@@ -134,14 +134,19 @@ func getStorageEnvs(s v1.JaegerStorageSpec) []corev1.EnvVar {
 			{Name: "CASSANDRA_CLIENT_AUTH_ENABLED", Value: strconv.FormatBool(s.Dependencies.CassandraClientAuthEnabled)},
 		}
 	case "elasticsearch":
-		return []corev1.EnvVar{
+		vars := []corev1.EnvVar{
 			{Name: "ES_NODES", Value: sFlagsMap["es.server-urls"]},
 			{Name: "ES_INDEX_PREFIX", Value: sFlagsMap["es.index-prefix"]},
 			{Name: "ES_USERNAME", Value: sFlagsMap["es.username"]},
 			{Name: "ES_PASSWORD", Value: sFlagsMap["es.password"]},
-			{Name: "ES_CLIENT_NODE_ONLY", Value: strconv.FormatBool(s.Dependencies.ElasticsearchClientNodeOnly)},
-			{Name: "ES_NODES_WAN_ONLY", Value: strconv.FormatBool(s.Dependencies.ElasticsearchNodesWanOnly)},
 		}
+		if s.Dependencies.ElasticsearchNodesWanOnly != nil {
+			vars = append(vars, corev1.EnvVar{Name: "ES_NODES_WAN_ONLY", Value: strconv.FormatBool(*s.Dependencies.ElasticsearchNodesWanOnly)})
+		}
+		if s.Dependencies.ElasticsearchClientNodeOnly != nil {
+			vars = append(vars, corev1.EnvVar{Name: "ES_CLIENT_NODE_ONLY", Value: strconv.FormatBool(*s.Dependencies.ElasticsearchClientNodeOnly)})
+		}
+		return vars
 	default:
 		return nil
 	}

diff --git a/pkg/cronjob/spark_dependencies_test.go b/pkg/cronjob/spark_dependencies_test.go
@@ -28,6 +28,8 @@ func TestRemoveEmptyVars(t *testing.T) {
 }
 
 func TestStorageEnvs(t *testing.T) {
+	trueVar := true
+	falseVar := false
 	tests := []struct {
 		storage  v1.JaegerStorageSpec
 		expected []corev1.EnvVar
@@ -65,8 +67,18 @@ func TestStorageEnvs(t *testing.T) {
 				{Name: "ES_INDEX_PREFIX", Value: "haha"},
 				{Name: "ES_USERNAME", Value: "jdoe"},
 				{Name: "ES_PASSWORD", Value: "none"},
-				{Name: "ES_CLIENT_NODE_ONLY", Value: "false"},
+			}},
+		{storage: v1.JaegerStorageSpec{Type: "elasticsearch",
+			Options: v1.NewOptions(map[string]interface{}{"es.server-urls": "lol:hol", "es.index-prefix": "haha",
+				"es.username": "jdoe", "es.password": "none"}),
+			Dependencies: v1.JaegerDependenciesSpec{ElasticsearchClientNodeOnly: &trueVar, ElasticsearchNodesWanOnly: &falseVar}},
+			expected: []corev1.EnvVar{
+				{Name: "ES_NODES", Value: "lol:hol"},
+				{Name: "ES_INDEX_PREFIX", Value: "haha"},
+				{Name: "ES_USERNAME", Value: "jdoe"},
+				{Name: "ES_PASSWORD", Value: "none"},
 				{Name: "ES_NODES_WAN_ONLY", Value: "false"},
+				{Name: "ES_CLIENT_NODE_ONLY", Value: "true"},
 			}},
 	}
 	for _, test := range tests {