pkg/settings/cresettings: add CL_CRE_SETTINGS for default overrides (#1499)

Author: jmank88

pkg/config/rate.go

@@ -33,8 +33,8 @@ func ParseRate(s string) (Rate, error) {
 		}
 		rateLimit.Limit = rate.Every(d)
 	} else {
-		s = strings.TrimSuffix(s, "rps") // allowed but not required
-		f, err := strconv.ParseFloat(s, 64)
+		parts[0] = strings.TrimSuffix(parts[0], "rps") // allowed but not required
+		f, err := strconv.ParseFloat(parts[0], 64)
 		if err != nil {
 			return Rate{}, err
 		}

pkg/settings/cresettings/settings.go

@@ -2,7 +2,9 @@
 package cresettings
 
 import (
+	"encoding/json"
 	"log"
+	"os"
 	"time"
 
 	"golang.org/x/time/rate"
@@ -12,6 +14,12 @@ import (
 )
 
 func init() {
+	if v, ok := os.LookupEnv("CL_CRE_SETTINGS"); ok {
+		err := json.Unmarshal([]byte(v), &Default)
+		if err != nil {
+			log.Fatalf("failed to initialize defaults: %v", err)
+		}
+	}
 	err := InitConfig(&Default)
 	if err != nil {
 		log.Fatalf("failed to initialize keys: %v", err)

pkg/settings/cresettings/settings_test.go

@@ -7,9 +7,14 @@ import (
 	"log"
 	"os"
 	"testing"
+	"time"
 
 	"github.com/pelletier/go-toml/v2"
+	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
+	"golang.org/x/time/rate"
+
+	"github.com/smartcontractkit/chainlink-common/pkg/config"
 )
 
 var update = flag.Bool("update", false, "update the golden files of this test")
@@ -52,3 +57,54 @@ func TestDefault(t *testing.T) {
 		}
 	})
 }
+
+func TestSchema_Unmarshal(t *testing.T) {
+	cfg := Default
+	require.NoError(t, json.Unmarshal([]byte(`{
+	"WorkflowLimit": "500",
+	"HTTPTrigger": {
+		"AuthRateLimit": "200rps:50"
+	},
+	"PerOrg": {
+		"ZeroBalancePruningTimeout": "48h"
+	},
+	"PerOwner": {
+		"WorkflowExecutionConcurrencyLimit": "99"
+	},
+	"PerWorkflow": {
+		"WASMMemoryLimit": "250mb",
+		"CRONTrigger": {
+			"RateLimit": "every10s:5"
+		},
+		"HTTPTrigger": {
+			"IncomingPayloadSizeLimit": "14kb"
+		},
+		"LogTrigger": {
+			"EventRateLimit": "every13s:6"
+		},
+		"HTTPAction": {
+			"RateLimit": "every3s:5"
+		},
+		"ChainWrite": {
+			"EVM": {
+				"TransactionGasLimit": "500000"
+			}
+		},
+		"ChainRead": {
+			"CallLimit": "3"
+		}
+	}
+}`), &cfg))
+
+	assert.Equal(t, 500, cfg.WorkflowLimit.DefaultValue)
+	assert.Equal(t, config.Rate{Limit: 200, Burst: 50}, cfg.HTTPTrigger.AuthRateLimit.DefaultValue)
+	assert.Equal(t, 48*time.Hour, cfg.PerOrg.ZeroBalancePruningTimeout.DefaultValue)
+	assert.Equal(t, 99, cfg.PerOwner.WorkflowExecutionConcurrencyLimit.DefaultValue)
+	assert.Equal(t, 250*config.MByte, cfg.PerWorkflow.WASMMemoryLimit.DefaultValue)
+	assert.Equal(t, config.Rate{Limit: rate.Every(10 * time.Second), Burst: 5}, cfg.PerWorkflow.CRONTrigger.RateLimit.DefaultValue)
+	assert.Equal(t, 14*config.KByte, cfg.PerWorkflow.HTTPTrigger.IncomingPayloadSizeLimit.DefaultValue)
+	assert.Equal(t, config.Rate{Limit: rate.Every(13 * time.Second), Burst: 6}, cfg.PerWorkflow.LogTrigger.EventRateLimit.DefaultValue)
+	assert.Equal(t, config.Rate{Limit: rate.Every(3 * time.Second), Burst: 5}, cfg.PerWorkflow.HTTPAction.RateLimit.DefaultValue)
+	assert.Equal(t, 500000, cfg.PerWorkflow.ChainWrite.EVM.TransactionGasLimit.DefaultValue)
+	assert.Equal(t, 3, cfg.PerWorkflow.ChainRead.CallLimit.DefaultValue)
+}

pkg/settings/settings.go

@@ -44,6 +44,17 @@ func (s Setting[T]) MarshalText() ([]byte, error) {
 	return []byte(fmt.Sprintf("%v", s.DefaultValue)), nil
 }
 
+func (s *Setting[T]) UnmarshalText(b []byte) (err error) {
+	if len(b) >= 2 && b[0] == '"' && b[len(b)-1] == '"' {
+		b = b[1 : len(b)-1] // unquote string
+	}
+	s.DefaultValue, err = s.Parse(string(b))
+	if err != nil {
+		err = fmt.Errorf("%s: failed to parse %s: %w", s.Key, string(b), err)
+	}
+	return
+}
+
 func (s *Setting[T]) initSetting(key string, scope Scope, unit *string) error {
 	s.Key = key
 	s.Scope = scope