chrly/api/mojang/queue/queue.go

package queue

import (
	"strings"
	"sync"
	"time"

	"github.com/elyby/chrly/api/mojang"
)

var onFirstCall sync.Once
var queue = jobsQueue{}

func ScheduleTexturesForUsername(username string) (resultChan chan *mojang.SignedTexturesResponse) {
	onFirstCall.Do(func() {
		queue.New()
		startQueue()
	})

	// TODO: prevent of adding the same username more than once
	queue.Enqueue(&jobItem{username, resultChan})

	return
}

func startQueue() {
	go func() {
		for {
			start := time.Now()
			queueRound()
			time.Sleep(time.Second - time.Since(start))
		}
	}()
}

func queueRound() {
	if queue.IsEmpty() {
		return
	}

	jobs := queue.Dequeue(100)
	var usernames []string
	for _, job := range jobs {
		usernames = append(usernames, job.Username)
	}

	profiles, err := mojang.UsernamesToUuids(usernames)
	switch err.(type) {
	case *mojang.TooManyRequestsError:
		for _, job := range jobs {
			job.RespondTo <- nil
		}

		return
	case error:
		panic(err)
	}

	var wg sync.WaitGroup
	for _, job := range jobs {
		wg.Add(1)
		go func() {
			var result *mojang.SignedTexturesResponse
			shouldCache := true
			var uuid string
			for _, profile := range profiles {
				if strings.EqualFold(job.Username, profile.Name) {
					uuid = profile.Id
					break
				}
			}

			if uuid != "" {
				result, err = mojang.UuidToTextures(uuid, true)
				if err != nil {
					if _, ok := err.(*mojang.TooManyRequestsError); !ok {
						panic(err)
					}

					shouldCache = false
				}
			}

			wg.Done()

			job.RespondTo <- result

			if shouldCache {
				// TODO: store result to cache
			}
		}()
	}

	wg.Wait()
}