func check cache by endpoint

Signed-off-by: Avelino <avelinorun@gmail.com>

cache/cache.go

@@ -0,0 +1,25 @@
+package cache
+
+import (
+	"log"
+
+	"github.com/prest/prest/config"
+)
+
+// CacheEndpointRules ...
+func CacheEndpointRules(uri string) (cacheEnable bool, time int) {
+	cacheEnable = false
+	if config.PrestConf.Cache && len(config.PrestConf.CacheEndpoints) == 0 {
+		cacheEnable = true
+	}
+	time = config.PrestConf.CacheTime
+	log.Println("cache len:", cacheEnable, config.PrestConf.CacheEndpoints, len(config.PrestConf.CacheEndpoints))
+	for _, endpoint := range config.PrestConf.CacheEndpoints {
+		if endpoint.Endpoint == uri {
+			cacheEnable = true
+			time = endpoint.CacheTime
+			return
+		}
+	}
+	return
+}

cache/cache_test.go

@@ -0,0 +1,30 @@
+package cache
+
+import (
+	"os"
+	"testing"
+
+	"github.com/prest/prest/config"
+)
+
+func TestCacheEndpointRulesEnable(t *testing.T) {
+	os.Setenv("PREST_CONF", "./testdata/prest.toml")
+	os.Setenv("PREST_CACHE", "true")
+	config.Load()
+	cacheEnable, cacheTime := CacheEndpointRules("/prest/public/test")
+	if !cacheEnable {
+		t.Errorf("expected cache endpoint rule true, but got %t", cacheEnable)
+	}
+	if cacheTime != 5 {
+		t.Errorf("expected cache endpoint time 5, but got %d", cacheTime)
+	}
+}
+
+func TestCacheEndpointRulesDisable(t *testing.T) {
+	os.Setenv("PREST_CACHE", "true")
+	config.Load()
+	cacheEnable, _ := CacheEndpointRules("/prest/public/test-disable")
+	if cacheEnable {
+		t.Errorf("expected cache endpoint rule false, but got %t", cacheEnable)
+	}
+}