Implement cache and logger middlewares for gemini

1 files changed, 14 insertions, 9 deletions

diff --git a/src/http/middleware.go b/src/http/middleware.go
index b82fc29..a21511f 100644
--- a/src/http/middleware.go
+++ b/src/http/middleware.go
@@ -4,11 +4,12 @@ import (
 	"bytes"
 	"net"
 	"net/http"
+	"net/url"
 	"path/filepath"
 	"sync"
 	"time"
 
-	lru "github.com/hashicorp/golang-lru"
+	"github.com/mediocregopher/blog.mediocregopher.com/srv/cache"
 	"github.com/mediocregopher/blog.mediocregopher.com/srv/http/apiutil"
 	"github.com/mediocregopher/mediocre-go-lib/v2/mctx"
 	"github.com/mediocregopher/mediocre-go-lib/v2/mlog"
@@ -86,6 +87,9 @@ func (rw *wrappedResponseWriter) WriteHeader(statusCode int) {
 }
 
 func logReqMiddleware(h http.Handler) http.Handler {
+
+	type logCtxKey string
+
 	return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
 
 		wrw := newWrappedResponseWriter(rw)
@@ -94,8 +98,6 @@ func logReqMiddleware(h http.Handler) http.Handler {
 		h.ServeHTTP(wrw, r)
 		took := time.Since(started)
 
-		type logCtxKey string
-
 		ctx := r.Context()
 		ctx = mctx.Annotate(ctx,
 			logCtxKey("took"), took.String(),
@@ -139,7 +141,7 @@ func (rw *cacheResponseWriter) Write(b []byte) (int, error) {
 	return rw.wrappedResponseWriter.Write(b)
 }
 
-func cacheMiddleware(cache *lru.Cache) middleware {
+func cacheMiddleware(cache cache.Cache, publicURL *url.URL) middleware {
 
 	type entry struct {
 		body      []byte
@@ -153,11 +155,14 @@ func cacheMiddleware(cache *lru.Cache) middleware {
 	return func(h http.Handler) http.Handler {
 		return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
 
-			id := r.URL.RequestURI()
+			// r.URL doesn't have Scheme or Host populated, better to add the
+			// public url to the key to make sure there's no possiblity of
+			// collision with other protocols using the cache.
+			id := publicURL.String() + "|" + r.URL.String()
 
-			if val, ok := cache.Get(id); ok {
+			if value := cache.Get(id); value != nil {
 
-				entry := val.(entry)
+				entry := value.(entry)
 
 				reader := pool.Get().(*bytes.Reader)
 				defer pool.Put(reader)
@@ -174,7 +179,7 @@ func cacheMiddleware(cache *lru.Cache) middleware {
 			h.ServeHTTP(cacheRW, r)
 
 			if cacheRW.statusCode == 200 {
-				cache.Add(id, entry{
+				cache.Set(id, entry{
 					body:      cacheRW.buf.Bytes(),
 					createdAt: time.Now(),
 				})
@@ -183,7 +188,7 @@ func cacheMiddleware(cache *lru.Cache) middleware {
 	}
 }
 
-func purgeCacheOnOKMiddleware(cache *lru.Cache) middleware {
+func purgeCacheOnOKMiddleware(cache cache.Cache) middleware {
 	return func(h http.Handler) http.Handler {
 		return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {