fix bug where some graphql pumps are not reading env properly (#780)

TykTechnologies · Jan 29, 2024 · d9d64dc · d9d64dc
1 parent 8ca8646
commit d9d64dc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/pumps/graph_mongo.go b/pumps/graph_mongo.go
@@ -13,6 +13,8 @@ import (
 
 const mongoGraphPrefix = "mongo-graph-pump"
 
+var mongoGraphDefaultEnv = PUMPS_ENV_PREFIX + "_MONGOGRAPH" + PUMPS_ENV_META_PREFIX
+
 type GraphMongoPump struct {
 	CommonPumpConfig
 	MongoPump
@@ -60,6 +62,7 @@ func (g *GraphMongoPump) Init(config interface{}) error {
 	if err := mapstructure.Decode(config, &g.dbConf.BaseMongoConf); err != nil {
 		return err
 	}
+	processPumpEnvVars(g, g.log, g.dbConf, mongoGraphDefaultEnv)
 
 	if g.dbConf.MaxInsertBatchSizeBytes == 0 {
 		g.log.Info("-- No max batch size set, defaulting to 10MB")

diff --git a/pumps/graph_sql_aggregate.go b/pumps/graph_sql_aggregate.go
@@ -25,6 +25,10 @@ func (s *GraphSQLAggregatePump) GetName() string {
 	return "Sql Graph Aggregate Pump"
 }
 
+func (s *GraphSQLAggregatePump) GetEnvPrefix() string {
+	return s.SQLConf.EnvPrefix
+}
+
 func (s *GraphSQLAggregatePump) New() Pump {
 	return &GraphSQLAggregatePump{}
 }