Руководство по Go API фильмов

Соберите утилиту командной строки на Go, которая выводит лучшие по рейтингу фильмы заданного жанра и года выхода. Она разбирает JSON в типизированные структуры, повторяет запрос при 429 после паузы Retry-After и кэширует ответы на диске. Нужна только стандартная библиотека, Go 1.22 или новее и около 20 минут.

.go Go Обновлено Открыть в Markdown

Вы построите topfilms, утилиту командной строки на Go. Выполните go run . -genre 878 -year 2010 -n 10, и она выведет десять лучших по рейтингу научно-фантастических фильмов 2010 года с оценкой и числом голосов. Она использует типизированные структуры, только стандартную библиотеку и Go 1.22 или новее. Рассчитывайте на 20 минут.

Шаг 1: получите ключ API

Начните со страницы цены, где вы выбираете оплаченный период и один из тарифов на 20, 40 или 120 запросов в минуту. Затем создайте ключ в панели аккаунта и экспортируйте его:

Программа читает его через os.Getenv("AMODB_KEY"), поэтому в исходном коде его нет.

Shell
export AMODB_KEY="YOUR_API_KEY"

Шаг 2: создайте модуль

Создайте папку и инициализируйте модуль. Скачивать ничего не нужно.

go version должен вывести 1.22 или выше. Весь код ниже помещайте в файл main.go.

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

Шаг 3: отправьте первый запрос

Утилита вызывает GET /3/discover/movie, который фильтрует и сортирует каталог. Жанр 878 это Science Fiction, а полный список ID есть на странице жанры. Эта полная программа только печатает строку статуса.

Никогда не используйте http.DefaultClient для вызовов API, потому что у него нет таймаута.

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)
}

Шаг 4: разбор в типизированные структуры

Объявляйте только те поля, которые читаете. encoding/json игнорирует остальные, поэтому новые поля в ответе не сломают утилиту.

Запрос просит sort_by=vote_average.desc и vote_count.gte=500. Без нижнего порога голосов в начале списка оказались бы названия с единственным голосом 10 из 10. Все фильтры описаны на странице 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"`
}

Шаг 5: повтор при 429 и отчет об ошибках

Прочитайте тело один раз, затем решайте по коду статуса.

СтатусЧто делает клиент
200Записывает файл кэша и разбирает тело.
401Возвращает ошибку с упоминанием AMODB_KEY. Без повтора, потому что следующий вызов тоже завершится ошибкой.
429Ждет Retry-After секунд (1, если заголовка нет, не больше 30) и пробует снова, всего до 3 попыток.
Другие 4xx и 5xxВозвращает status_code и status_message из тела ошибки TMDB.
Ошибка транспортаВозвращает network error, обернутую через %w.

Вспомогательная функция задержки небольшая:

Все коды описаны на странице ошибки, а скользящее окно на странице лимиты запросов.

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
}

Шаг 6: кэшируйте ответы на диске

Граница сети хранит списки discover 30 минут. Клиент сохраняет каждое успешное тело в пользовательской папке кэша под именем SHA-256 от полного URL и отдает его снова, пока файл моложе TTL. Повторная команда тогда не делает запросов.

Шаг 7: полная программа

Замените main.go этим кодом и выполните go run . -genre 878 -year 2010 -n 10.

Пробуйте другие фильтры через флаги. go run . -genre 18 -year 1999 -n 5 выводит пять лучших по рейтингу драм 1999 года. Выполните ту же команду дважды, и второй запуск ответит из файла кэша, не обращаясь к сети, пока не пройдет 30 минут. Если для вашей комбинации нет фильма с 500 голосами, утилита печатает No movies matched. и завершается как обычно.

Цикл повторов возвращает результат из switch во всех случаях, кроме 429, когда еще остались попытки. Поэтому функция заканчивается обычной ошибкой: после цикла компилятору нужен return, а сообщение появилось бы, только если изменить логику цикла.

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)
	}
}

Дальнейшие шаги

  • Добавьте флаг -page и передайте его как параметр page, чтобы пройти дальше по списку.
  • Получите каждый ID фильма через фильмы и добавьте append_to_response=ratings, чтобы прочитать оценку IMDb из ratings.imdb.value.
  • Запускайте несколько жанров одновременно через горутины, оставаясь в пределах запросов в минуту вашего тарифа.
  • Перенесите ту же идею на Java или прочитайте быстрый старт.