Impl ListContributionCollections

Author: gedorinku

infra/store/github/github_store.go

@@ -4,12 +4,14 @@ import (
 	"context"
 	"database/sql"
 	"fmt"
+	"strconv"
 	"strings"
 	"time"
 
 	"github.com/go-redis/redis"
 	"github.com/pkg/errors"
 	"github.com/volatiletech/sqlboiler/boil"
+	"github.com/volatiletech/sqlboiler/queries/qm"
 
 	"github.com/ProgrammingLab/prolab-accounts/infra/record"
 	"github.com/ProgrammingLab/prolab-accounts/infra/store"
@@ -33,15 +35,15 @@ func NewGitHubStore(ctx context.Context, db *sqlutil.DB, cli *redis.Client) stor
 }
 
 const (
-	contributionTotalCount = "contributions-total-count"
+	contributionTotalCountKey = "contributions-total-count"
 )
 
 func (s *githubStoreImpl) UpdateContributionDays(c *model.GitHubContributionCollection) ([]*record.GithubContributionDay, error) {
 	z := redis.Z{
 		Score:  float64(c.TotalCount),
 		Member: int64(c.UserID),
 	}
-	err := s.cli.ZAdd(contributionTotalCount, z).Err()
+	err := s.cli.ZAdd(contributionTotalCountKey, z).Err()
 	if err != nil {
 		return nil, errors.WithStack(err)
 	}
@@ -64,6 +66,77 @@ func (s *githubStoreImpl) UpdateContributionDays(c *model.GitHubContributionColl
 	return days, nil
 }
 
+func (s *githubStoreImpl) ListContributionCollections(usersLimit int) ([]*model.GitHubContributionCollection, error) {
+	cols, err := s.getTopUsers(usersLimit)
+	if err != nil {
+		return nil, err
+	}
+
+	err = s.loadDays(cols)
+	if err != nil {
+		return nil, err
+	}
+
+	return cols, nil
+}
+
+func (s *githubStoreImpl) getTopUsers(usersLimit int) ([]*model.GitHubContributionCollection, error) {
+	values, err := s.cli.ZRevRangeWithScores(contributionTotalCountKey, 0, int64(usersLimit)-1).Result()
+	if err != nil {
+		return nil, errors.WithStack(err)
+	}
+
+	cols := make([]*model.GitHubContributionCollection, 0, len(values))
+	for _, v := range values {
+		id, err := strconv.ParseInt(v.Member.(string), 10, 64)
+		if err != nil {
+			return nil, errors.WithStack(err)
+		}
+		col := &model.GitHubContributionCollection{
+			UserID:     model.UserID(id),
+			TotalCount: int(v.Score),
+		}
+		cols = append(cols, col)
+	}
+
+	return cols, nil
+}
+
+func (s *githubStoreImpl) loadDays(cols []*model.GitHubContributionCollection) error {
+	userIDs := make([]int64, 0, len(cols))
+	for _, c := range cols {
+		userIDs = append(userIDs, int64(c.UserID))
+	}
+
+	mods := []qm.QueryMod{
+		qm.Load("User.Profile.Department"),
+		qm.Load("User.Profile.Role"),
+		qm.From("github_contribution_days as days"),
+		qm.InnerJoin("users on users.id = days.user_id"),
+		qm.InnerJoin("profiles on profiles.id = users.profile_id"),
+		qm.Where("profiles.profile_scope = ?", model.Public),
+		qm.WhereIn("users.id in ?", sqlutil.Int64SliceToAbstractSlice(userIDs)...),
+		qm.OrderBy("days.date"),
+	}
+	var days []*record.GithubContributionDay
+	err := record.NewQuery(mods...).Bind(s.ctx, s.db, &days)
+	if err != nil {
+		return errors.WithStack(err)
+	}
+
+	daysMap := make(map[model.UserID][]*record.GithubContributionDay, len(cols))
+	for _, d := range days {
+		id := model.UserID(d.UserID)
+		daysMap[id] = append(daysMap[id], d)
+	}
+
+	for _, c := range cols {
+		c.Days = daysMap[c.UserID]
+	}
+
+	return nil
+}
+
 func bulkInsert(ctx context.Context, tx *sql.Tx, days []*record.GithubContributionDay) error {
 	if len(days) == 0 {
 		return nil

infra/store/github_store.go

@@ -8,4 +8,5 @@ import (
 // GitHubStore provides github data
 type GitHubStore interface {
 	UpdateContributionDays(c *model.GitHubContributionCollection) ([]*record.GithubContributionDay, error)
+	ListContributionCollections(usersLimit int) ([]*model.GitHubContributionCollection, error)
 }

sqlutil/conv.go

@@ -0,0 +1,19 @@
+package sqlutil
+
+// Int32SliceToInt64Slice converts int32 array to int64 array
+func Int32SliceToInt64Slice(a []int32) []int64 {
+	res := make([]int64, 0, len(a))
+	for _, v := range a {
+		res = append(res, int64(v))
+	}
+	return res
+}
+
+// Int64SliceToAbstractSlice converts int64 array to []interface{}
+func Int64SliceToAbstractSlice(a []int64) []interface{} {
+	res := []interface{}{}
+	for _, v := range a {
+		res = append(res, v)
+	}
+	return res
+}