fix domain cache verification and add cache timeout config

weni-ai · Jan 20, 2025 · b65ea6c · b65ea6c
1 parent abadd8b
commit b65ea6c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/config/config.go b/config/config.go
@@ -20,7 +20,8 @@ type Configuration struct {
 
 	RestrictDomains bool `default:"false" env:"WWC_RESTRICT_DOMAINS"`
 
-	FlowsURL string `default:"flows.weni.ai" env:"WWC_FLOWS_URL"`
+	FlowsURL        string `default:"flows.weni.ai" env:"WWC_FLOWS_URL"`
+	MemCacheTimeout int64  `default:"5" env:"WWC_MEM_CACHE_TIMEOUT"`
 }
 
 type S3 struct {

diff --git a/pkg/websocket/client.go b/pkg/websocket/client.go
@@ -181,15 +181,16 @@ func CheckAllowedDomain(app *App, channelUUID string, originDomain string) bool
 	var allowedDomains []string = nil
 	var err error
 	cachedDomains, notexpired := cacheChannelDomains.Get(channelUUID)
-	if !notexpired {
+	if notexpired {
 		allowedDomains = cachedDomains
 	} else {
 		allowedDomains, err = app.FlowsClient.GetChannelAllowedDomains(channelUUID)
 		if err != nil {
 			log.Error("Error on get allowed domains", err)
 			return false
 		}
-		cacheChannelDomains.Set(channelUUID, allowedDomains, time.Minute*5)
+		cacheTimeout := config.Get().MemCacheTimeout
+		cacheChannelDomains.Set(channelUUID, allowedDomains, time.Minute*time.Duration(cacheTimeout))
 	}
 	if len(allowedDomains) > 0 {
 		for _, domain := range allowedDomains {