StreamAnime.go

package kitsu

import (
	"strings"
	"time"
)

// StreamAnime returns a stream of all anime objects (async).
func StreamAnime() chan *Anime {
	channel := make(chan *Anime)
	url := "anime?page[limit]=20&page[offset]=0"
	ticker := time.NewTicker(500 * time.Millisecond)
	rateLimit := ticker.C

	go func() {
		defer close(channel)
		defer ticker.Stop()

		for {
			page, err := GetAnimePage(url)

			if err != nil {
				panic(err)
			}

			// Feed anime data from current page to the stream
			for _, anime := range page.Data {
				channel <- anime
			}

			nextURL := page.Links.Next

			// Did we reach the end?
			if nextURL == "" {
				break
			}

			// Cut off API base URL
			nextURL = strings.TrimPrefix(nextURL, APIBaseURL)

			// Continue with the next page
			url = nextURL

			// Wait for rate limiter to allow the next request
			<-rateLimit
		}
	}()

	return channel
}