Use concurrency safe context by default.

Allow use of unsafe context.

Author: andreiavrammsd

context.go

@@ -13,6 +13,7 @@ import (
 	"os"
 	"path/filepath"
 	"strings"
+	"sync"
 )
 
 type (
@@ -198,6 +199,7 @@ type (
 		handler  HandlerFunc
 		store    Map
 		echo     *Echo
+		lock     sync.RWMutex
 	}
 )
 
@@ -360,10 +362,18 @@ func (c *context) Cookies() []*http.Cookie {
 }
 
 func (c *context) Get(key string) interface{} {
+	if !c.echo.UnsafeContext {
+		c.lock.RLock()
+		defer c.lock.RUnlock()
+	}
 	return c.store[key]
 }
 
 func (c *context) Set(key string, val interface{}) {
+	if !c.echo.UnsafeContext {
+		c.lock.Lock()
+		defer c.lock.Unlock()
+	}
 	if c.store == nil {
 		c.store = make(Map)
 	}
@@ -597,4 +607,3 @@ func (c *context) Reset(r *http.Request, w http.ResponseWriter) {
 	// NOTE: Don't reset because it has to have length c.echo.maxParam at all times
 	// c.pvalues = nil
 }
-

context_test.go

@@ -523,12 +523,59 @@ func TestContextRedirect(t *testing.T) {
 }
 
 func TestContextStore(t *testing.T) {
-	var c Context
-	c = new(context)
+	e := &Echo{}
+
+	c := &context{
+		echo: e,
+	}
+
 	c.Set("name", "Jon Snow")
 	testify.Equal(t, "Jon Snow", c.Get("name"))
 }
 
+func BenchmarkContext_Store(b *testing.B) {
+	e := &Echo{}
+
+	c := &context{
+		echo: e,
+	}
+
+	for n := 0; n < b.N; n++ {
+		c.Set("name", "Jon Snow")
+		if c.Get("name") != "Jon Snow" {
+			b.Fail()
+		}
+	}
+}
+
+func TestContextUnsafeStore(t *testing.T) {
+	e := &Echo{}
+	e.UnsafeContext = true
+
+	c := &context{
+		echo: e,
+	}
+
+	c.Set("name", "Jon Safe")
+	testify.Equal(t, "Jon Safe", c.Get("name"))
+}
+
+func BenchmarkContext_UnsafeStore(b *testing.B) {
+	e := &Echo{}
+	e.UnsafeContext = true
+
+	c := &context{
+		echo: e,
+	}
+
+	for n := 0; n < b.N; n++ {
+		c.Set("name", "Jon Snow")
+		if c.Get("name") != "Jon Snow" {
+			b.Fail()
+		}
+	}
+}
+
 func TestContextHandler(t *testing.T) {
 	e := New()
 	r := e.Router()
@@ -543,4 +590,3 @@ func TestContextHandler(t *testing.T) {
 	c.Handler()(c)
 	testify.Equal(t, "handler", b.String())
 }
-

echo.go

@@ -84,6 +84,7 @@ type (
 		Validator        Validator
 		Renderer         Renderer
 		Logger           Logger
+		UnsafeContext    bool
 	}
 
 	// Route contains a handler and information for matching against requests.