Návod na filmové API v Go

Vytvořte nástroj pro příkazový řádek v Go, který vypíše nejlépe hodnocené filmy zvoleného žánru a roku vydání. Dekóduje JSON do typovaných struktur, po 429 opakuje pokus po prodlevě z Retry-After a ukládá odpovědi na disk. Používá jen standardní knihovnu, vyžaduje Go 1.22 nebo novější a zabere asi 20 minut.

.go Go Aktualizováno Zobrazit jako Markdown

Vytvoříte topfilms, nástroj pro příkazový řádek v Go. Spusťte go run . -genre 878 -year 2010 -n 10 a vypíše deset nejlépe hodnocených sci-fi filmů z roku 2010 se skóre a počtem hlasů. Používá typované struktury, jen standardní knihovnu a Go 1.22 nebo novější. Počítejte s 20 minutami.

Krok 1: Získejte API klíč

Začněte na stránce ceník, kde zvolíte předplacené období a jeden z tarifů s 20, 40 nebo 120 požadavky za minutu. Pak vytvořte klíč v panelu účtu a exportujte ho:

Program ho čte pomocí os.Getenv("AMODB_KEY"), takže nikdy nezůstane ve zdrojovém kódu.

Shell
export AMODB_KEY="YOUR_API_KEY"

Krok 2: Vytvořte modul

Vytvořte složku a inicializujte modul. Není co stahovat.

go version by měl vypsat 1.22 nebo vyšší. Veškerý kód níže vložte do souboru main.go.

Shell
mkdir topfilms && cd topfilms
go mod init topfilms
go version

Krok 3: Odešlete první požadavek

Nástroj volá GET /3/discover/movie, který filtruje a řadí katalog. Žánr 878 je sci-fi a úplný seznam ID je na stránce žánry. Tento kompletní program jen vypíše stavový řádek.

Pro volání API nikdy nepoužívejte http.DefaultClient, protože nemá timeout.

Go
package main

import (
	"fmt"
	"net/http"
	"os"
	"time"
)

func main() {
	url := "https://api.amodb.dev/3/discover/movie?with_genres=878"
	req, _ := http.NewRequest(http.MethodGet, url, nil)
	req.Header.Set("Authorization", "Bearer "+os.Getenv("AMODB_KEY"))
	res, err := (&http.Client{Timeout: 10 * time.Second}).Do(req)
	if err != nil {
		panic(err)
	}
	defer res.Body.Close()
	fmt.Println(res.Status)
}

Krok 4: Dekódujte do typovaných struktur

Deklarujte jen pole, která čtete. encoding/json zbytek ignoruje, takže nová pole v odpovědi nástroj nerozbijí.

Dotaz žádá o sort_by=vote_average.desc a vote_count.gte=500. Bez minima hlasů by seznam vedly tituly s jediným hodnocením 10 hvězdiček. Každý filtr je zdokumentován na stránce discover.

Go
type Movie struct {
	ID          int     `json:"id"`
	Title       string  `json:"title"`
	ReleaseDate string  `json:"release_date"`
	VoteAverage float64 `json:"vote_average"`
	VoteCount   int     `json:"vote_count"`
}

type DiscoverPage struct {
	TotalResults int     `json:"total_results"`
	Results      []Movie `json:"results"`
}

Krok 5: Opakujte pokus při 429 a hlaste chyby

Přečtěte tělo jednou a pak rozhodněte podle stavového kódu.

StavCo dělá klient
200Zapíše soubor cache a dekóduje tělo.
401Vrátí chybu, která uvádí AMODB_KEY. Bez opakování, protože další volání by také selhalo.
429Spí Retry-After sekund (1, když chybí, nejvýše 30) a zkusí to znovu, nejvýše 3 pokusy.
Ostatní 4xx a 5xxVrátí status_code a status_message z těla chyby TMDB.
Chyba přenosuVrátí network error obalené pomocí %w.

Pomocná funkce pro prodlevu je malá:

Všechny kódy najdete na stránce chyby a klouzavé okno na stránce limity požadavků.

Go
func retryDelay(header string) time.Duration {
	seconds, err := strconv.Atoi(header)
	if err != nil || seconds < 1 {
		seconds = 1
	}
	return time.Duration(min(seconds, 30)) * time.Second
}

Krok 6: Ukládejte odpovědi na disk

Edge drží seznamy discover 30 minut. Klient uloží každé úspěšné tělo do uživatelské složky cache pod názvem podle SHA-256 celého URL a znovu ho poskytne, dokud je soubor mladší než TTL. Opakovaný příkaz pak neposílá žádný požadavek.

Krok 7: Kompletní program

Nahraďte main.go tímto a spusťte go run . -genre 878 -year 2010 -n 10.

Vyzkoušejte jiné filtry pomocí přepínačů. go run . -genre 18 -year 1999 -n 5 vypíše pět nejlépe hodnocených dramat z roku 1999. Spusťte stejný příkaz dvakrát a druhé spuštění odpoví ze souboru cache, aniž by sáhlo na síť, dokud neuplyne 30 minut. Pokud pro vaši kombinaci nemá žádný film 500 hlasů, nástroj vypíše No movies matched. a skončí normálně.

Smyčka opakování se ve všech případech vrací zevnitř switch, kromě 429, která má ještě zbývající pokusy. Proto funkce končí prostou chybou: kompilátor potřebuje return za smyčkou a zpráva by se objevila, jen kdyby se změnila logika smyčky.

Go
package main

import (
	"crypto/sha256"
	"encoding/json"
	"errors"
	"flag"
	"fmt"
	"io"
	"net/http"
	"net/url"
	"os"
	"path/filepath"
	"strconv"
	"time"
)

const baseURL = "https://api.amodb.dev/3"

type Movie struct {
	ID          int     `json:"id"`
	Title       string  `json:"title"`
	ReleaseDate string  `json:"release_date"`
	VoteAverage float64 `json:"vote_average"`
	VoteCount   int     `json:"vote_count"`
}

type DiscoverPage struct {
	TotalResults int     `json:"total_results"`
	Results      []Movie `json:"results"`
}

type apiError struct {
	StatusCode    int    `json:"status_code"`
	StatusMessage string `json:"status_message"`
}

type Client struct {
	key      string
	hc       *http.Client
	cacheDir string
}

func NewClient(key string) (*Client, error) {
	base, err := os.UserCacheDir()
	if err != nil {
		base = os.TempDir()
	}
	dir := filepath.Join(base, "amodb-top-films")
	if err := os.MkdirAll(dir, 0o755); err != nil {
		return nil, err
	}
	return &Client{key: key, hc: &http.Client{Timeout: 10 * time.Second}, cacheDir: dir}, nil
}

func (c *Client) Get(path string, query url.Values, ttl time.Duration, out any) error {
	endpoint := baseURL + path + "?" + query.Encode()
	cacheFile := filepath.Join(c.cacheDir, fmt.Sprintf("%x.json", sha256.Sum256([]byte(endpoint))))
	if info, err := os.Stat(cacheFile); err == nil && time.Since(info.ModTime()) < ttl {
		if body, err := os.ReadFile(cacheFile); err == nil {
			return json.Unmarshal(body, out)
		}
	}

	for attempt := 1; attempt <= 3; attempt++ {
		req, err := http.NewRequest(http.MethodGet, endpoint, nil)
		if err != nil {
			return err
		}
		req.Header.Set("Authorization", "Bearer "+c.key)
		res, err := c.hc.Do(req)
		if err != nil {
			return fmt.Errorf("network error: %w", err)
		}
		body, err := io.ReadAll(res.Body)
		res.Body.Close()
		if err != nil {
			return fmt.Errorf("reading reply: %w", err)
		}

		switch {
		case res.StatusCode == http.StatusOK:
			_ = os.WriteFile(cacheFile, body, 0o644)
			return json.Unmarshal(body, out)
		case res.StatusCode == http.StatusTooManyRequests && attempt < 3:
			time.Sleep(retryDelay(res.Header.Get("Retry-After")))
		default:
			return statusError(res.StatusCode, body)
		}
	}
	return errors.New("retries exhausted")
}

func retryDelay(header string) time.Duration {
	seconds, err := strconv.Atoi(header)
	if err != nil || seconds < 1 {
		seconds = 1
	}
	return time.Duration(min(seconds, 30)) * time.Second
}

func statusError(status int, body []byte) error {
	if status == http.StatusUnauthorized {
		return errors.New("401: the key is invalid or suspended, check AMODB_KEY")
	}
	var e apiError
	if json.Unmarshal(body, &e) == nil && e.StatusMessage != "" {
		return fmt.Errorf("amodb returned %d (status_code %d): %s", status, e.StatusCode, e.StatusMessage)
	}
	return fmt.Errorf("amodb returned %d", status)
}

func yearOf(date string) string {
	if len(date) >= 4 {
		return date[:4]
	}
	return "n/a"
}

func main() {
	genre := flag.Int("genre", 878, "genre id, 878 is Science Fiction")
	year := flag.Int("year", 2010, "primary release year")
	limit := flag.Int("n", 10, "how many movies to print, 1 to 20")
	flag.Parse()

	key := os.Getenv("AMODB_KEY")
	if key == "" {
		fmt.Fprintln(os.Stderr, "Set AMODB_KEY first.")
		os.Exit(1)
	}
	client, err := NewClient(key)
	if err != nil {
		fmt.Fprintln(os.Stderr, err)
		os.Exit(1)
	}

	query := url.Values{}
	query.Set("language", "en-US")
	query.Set("with_genres", strconv.Itoa(*genre))
	query.Set("primary_release_year", strconv.Itoa(*year))
	query.Set("sort_by", "vote_average.desc")
	query.Set("vote_count.gte", "500")

	var page DiscoverPage
	if err := client.Get("/discover/movie", query, 30*time.Minute, &page); err != nil {
		fmt.Fprintln(os.Stderr, "error:", err)
		os.Exit(1)
	}
	if len(page.Results) == 0 {
		fmt.Println("No movies matched.")
		return
	}
	fmt.Printf("%d movies match, best first:\n", page.TotalResults)
	for i, m := range page.Results {
		if i >= *limit {
			break
		}
		fmt.Printf("%2d. %s (%s) %.1f from %d votes\n", i+1, m.Title, yearOf(m.ReleaseDate), m.VoteAverage, m.VoteCount)
	}
}

Další kroky

  • Přidejte přepínač -page a předejte ho jako parametr page, abyste procházeli seznam dál.
  • Vyhledejte každé ID filmu přes filmy a přidejte append_to_response=ratings, abyste četli skóre IMDb z ratings.imdb.value.
  • Spusťte několik žánrů najednou pomocí goroutin a držte se pod požadavky za minutu svého tarifu.
  • Přeneste stejný nápad do Javy nebo si přečtěte rychlý start.