Commit3b6151c

committed

Cache tokens in memory

1 parentb7cea93 commit3b6151cCopy full SHA for 3b6151c

File tree

+65

-4

lines changed

+65

-4

lines changed

Lines changed: 10 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -64,3 +64,13 @@ func TestCreateToken(t *testing.T) {`
`64`	`64`	`assert.Nil(t,err)`
`65`	`65`	`assert.Len(t,token,64)`
`66`	`66`	`}`
	`67`	`+`
	`68`	`+funcBenchmarkPing(b*testing.B) {`
	`69`	`+r:=createRouter()`
	`70`	`+req:=testRequest(r,"GET","/account/ping",nil)`
	`71`	`+req.Header.Set("Authorization","token")`
	`72`	`+`
	`73`	`+fori:=0;i<b.N;i++ {`
	`74`	`+req.send()`
	`75`	`+}`
	`76`	`+}`

Lines changed: 5 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -10,9 +10,10 @@ import (`
`10`	`10`	`)`
`11`	`11`
`12`	`12`	`var (`
`13`		`-storeAPIStore`
`14`		`-crawl*crawler`
`15`		`-config*Config`
	`13`	`+storeAPIStore`
	`14`	`+authCache*memAuthCache`
	`15`	`+crawl*crawler`
	`16`	`+config*Config`
`16`	`17`	`)`
`17`	`18`
`18`	`19`	`typeConfigstruct {`
`@@ -34,6 +35,7 @@ func Configure(cfg *Config) {`
`34`	`35`
`35`	`36`	`funcServe() {`
`36`	`37`	`openStore(path.Join(config.Dir,"store"))`
	`38`	`+authCache=newMemAuthCache()`
`37`	`39`
`38`	`40`	`crawl=newCrawler()`
`39`	`41`	`crawl.start(config.MaxDownloadConnections)`

Lines changed: 1 addition & 0 deletions

Lines changed: 34 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,8 @@`
`1`	`1`	`package api`
`2`	`2`
`3`	`3`	`import (`
	`4`	`+"sync"`
	`5`	`+`
`4`	`6`	`"github.com/Castcloud/castcloud-go-server/Godeps/_workspace/src/github.com/labstack/echo"`
`5`	`7`	`)`
`6`	`8`
`@@ -12,14 +14,45 @@ func auth() echo.HandlerFunc {`
`12`	`14`	`returnecho.NewHTTPError(401)`
`13`	`15`	`}`
`14`	`16`
`15`		`-user:=store.GetUserByToken(token)`
	`17`	`+user:=authCache.get(token)`
	`18`	`+ifuser!=nil {`
	`19`	`+c.Set("user",user)`
	`20`	`+returnnil`
	`21`	`+}`
	`22`	`+`
	`23`	`+user=store.GetUserByToken(token)`
`16`	`24`	`ifuser==nil {`
`17`	`25`	`returnecho.NewHTTPError(401)`
`18`	`26`	`}`
`19`	`27`
	`28`	`+authCache.set(token,user)`
`20`	`29`	`c.Set("user",user)`
`21`	`30`	`}`
`22`	`31`
`23`	`32`	`returnnil`
`24`	`33`	`}`
`25`	`34`	`}`
	`35`	`+`
	`36`	`+typememAuthCachestruct {`
	`37`	`+usersmap[string]*User`
	`38`	`+lock sync.Mutex`
	`39`	`+}`
	`40`	`+`
	`41`	`+funcnewMemAuthCache()*memAuthCache {`
	`42`	`+return&memAuthCache{`
	`43`	`+users:make(map[string]*User),`
	`44`	`+}`
	`45`	`+}`
	`46`	`+`
	`47`	`+func (cmemAuthCache)get(tokenstring)User {`
	`48`	`+c.lock.Lock()`
	`49`	`+user:=c.users[token]`
	`50`	`+c.lock.Unlock()`
	`51`	`+returnuser`
	`52`	`+}`
	`53`	`+`
	`54`	`+func (cmemAuthCache)set(tokenstring,userUser) {`
	`55`	`+c.lock.Lock()`
	`56`	`+c.users[token]=user`
	`57`	`+c.lock.Unlock()`
	`58`	`+}`

Lines changed: 15 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -31,3 +31,18 @@ func TestAuth(t *testing.T) {`
`31`	`31`	`c.Request().URL.Path="/account/login"`
`32`	`32`	`assert.Nil(t,mw(c))`
`33`	`33`	`}`
	`34`	`+`
	`35`	`+funcBenchmarkAuth(b*testing.B) {`
	`36`	`+fori:=0;i<b.N;i++ {`
	`37`	`+store.GetUserByToken("token")`
	`38`	`+}`
	`39`	`+}`
	`40`	`+`
	`41`	`+funcBenchmarkAuthCache(b*testing.B) {`
	`42`	`+cache:=newMemAuthCache()`
	`43`	`+cache.set("token",store.GetUserByToken("token"))`
	`44`	`+`
	`45`	`+fori:=0;i<b.N;i++ {`
	`46`	`+cache.get("token")`
	`47`	`+}`
	`48`	`+}`

Comments

(0)