func TestReposCache(t *testing.T) { g := goblin.Goblin(t) g.Describe("Repo List Cache", func() { var c *gin.Context g.BeforeEach(func() { c = new(gin.Context) cache.ToContext(c, cache.Default()) }) g.It("should skip when no user session", func() { Perms(c) _, ok := c.Get("perm") g.Assert(ok).IsFalse() }) g.It("should get repos from cache", func() { c.Set("user", fakeUser) cache.SetRepos(c, fakeUser, fakeRepos) Repos(c) repos, ok := c.Get("repos") g.Assert(ok).IsTrue() g.Assert(repos).Equal(fakeRepos) }) }) }
// Repos is a middleware function that attempts to cache the // user's list of remote repositories (ie in GitHub) to minimize // remote calls that might be expensive, slow or rate-limited. func Repos(c *gin.Context) { var user, _ = c.Get("user") if user == nil { c.Next() return } // if the item already exists in the cache // we can continue the middleware chain and // exit afterwards. v := cache.GetRepos(c, user.(*model.User)) if v != nil { c.Set("repos", v) c.Next() return } // otherwise, if the item isn't cached we execute // the middleware chain and then cache the permissions // after the request is processed. c.Next() repos, ok := c.Get("repos") if ok { cache.SetRepos(c, user.(*model.User), repos.([]*model.RepoLite), ) } }