58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package cache
|
|
|
|
import (
|
|
"io"
|
|
"log/slog"
|
|
"net/http"
|
|
"path/filepath"
|
|
)
|
|
|
|
func (c *Cache) FetchDB(repo string) error {
|
|
dbFile := repo + ".db.tar.gz"
|
|
dbPath := filepath.Join(repo, "os/x86_64", dbFile)
|
|
url := c.nextMirror() + dbPath
|
|
|
|
// set the user agent
|
|
req, err := http.NewRequest("GET", url, nil)
|
|
if err != nil {
|
|
slog.Error("failed create request", "err", err)
|
|
return err
|
|
}
|
|
req.Header.Set("User-Agent", userAgent)
|
|
|
|
resp, err := c.client.Do(req)
|
|
if err != nil {
|
|
slog.Warn("fetch failed", "url", url, "err", err)
|
|
return err
|
|
}
|
|
if resp.StatusCode != 200 {
|
|
slog.Info("fetch returned", "url", url, "status", resp.StatusCode)
|
|
return &UpstreamError{StatusCode: resp.StatusCode}
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
tmpPath := dbPath + ".tmp"
|
|
tmpFile, err := c.cr.Create(tmpPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := io.Copy(tmpFile, resp.Body); err != nil {
|
|
tmpFile.Close()
|
|
c.cr.Remove(tmpPath)
|
|
return err
|
|
}
|
|
|
|
if err := tmpFile.Close(); err != nil {
|
|
c.cr.Remove(tmpPath)
|
|
return err
|
|
}
|
|
if err := c.cr.Rename(tmpPath, dbPath); err != nil {
|
|
c.cr.Remove(tmpPath)
|
|
slog.Error("failed to move tmpfile to permanent file", "file", tmpFile, "err", err)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|