mirror of
https://github.com/woodchen-ink/webp_server_go.git
synced 2025-07-18 13:42:02 +08:00
115 lines
2.6 KiB
Go
115 lines
2.6 KiB
Go
package handler
|
||
|
||
import (
|
||
"fmt"
|
||
"io"
|
||
"net/http"
|
||
"os"
|
||
"path"
|
||
"path/filepath"
|
||
"strconv"
|
||
"time"
|
||
"webp_server_go/config"
|
||
"webp_server_go/helper"
|
||
|
||
"github.com/gofiber/fiber/v2"
|
||
log "github.com/sirupsen/logrus"
|
||
)
|
||
|
||
// Given /path/to/node.png
|
||
// Delete /path/to/node.png*
|
||
func cleanProxyCache(cacheImagePath string) {
|
||
// Delete /node.png*
|
||
files, err := filepath.Glob(cacheImagePath + "*")
|
||
if err != nil {
|
||
log.Infoln(err)
|
||
}
|
||
for _, f := range files {
|
||
if err := os.Remove(f); err != nil {
|
||
log.Info(err)
|
||
}
|
||
}
|
||
}
|
||
|
||
func downloadFile(filepath string, url string) error {
|
||
resp, err := http.Get(url)
|
||
if err != nil {
|
||
log.Errorln("下载文件时连接到远程错误！")
|
||
return err
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
if resp.StatusCode != fiber.StatusOK {
|
||
log.Errorf("获取远程图像时远程返回 %s", resp.Status)
|
||
return fmt.Errorf("意外的状态: %s", resp.Status)
|
||
}
|
||
|
||
// 创建目标文件
|
||
_ = os.MkdirAll(path.Dir(filepath), 0755)
|
||
out, err := os.Create(filepath)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer out.Close()
|
||
|
||
// 使用小缓冲区流式写入文件
|
||
buf := make([]byte, 32*1024)
|
||
_, err = io.CopyBuffer(out, resp.Body, buf)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
func fetchRemoteImg(url string, subdir string) config.MetaFile {
|
||
var metadata config.MetaFile
|
||
|
||
metadata = helper.ReadMetadata(url, "", subdir)
|
||
localRawImagePath := path.Join(config.Config.RemoteRawPath, subdir, metadata.Id)
|
||
|
||
if !helper.ImageExists(localRawImagePath) {
|
||
log.Info("在远程原始文件中找不到远程文件，正在获取...")
|
||
downloadFile(localRawImagePath, url)
|
||
// 重新读取更新后的元数据
|
||
metadata = helper.WriteMetadata(url, "", subdir)
|
||
}
|
||
|
||
return metadata
|
||
}
|
||
|
||
func pingURL(url string) (string, int64, time.Time) {
|
||
var etag string
|
||
var size int64
|
||
var lastModified time.Time
|
||
|
||
resp, err := http.Head(url)
|
||
if err != nil {
|
||
log.Errorf("pingUrl 时连接到远程错误: %v", err)
|
||
return "", 0, time.Time{}
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
if resp.StatusCode == fiber.StatusOK {
|
||
etag = resp.Header.Get("ETag")
|
||
sizeStr := resp.Header.Get("Content-Length")
|
||
size, _ = strconv.ParseInt(sizeStr, 10, 64)
|
||
lastModifiedStr := resp.Header.Get("Last-Modified")
|
||
lastModified, _ = time.Parse(time.RFC1123, lastModifiedStr)
|
||
|
||
if etag == "" {
|
||
log.Warn("远程未在标头中返回 ETag，使用 Last-Modified（如果可用）")
|
||
etag = lastModifiedStr
|
||
}
|
||
|
||
if etag == "" && lastModified.IsZero() {
|
||
log.Warn("ETag 和 Last-Modified 都不可用，使用 Content-Length 作为后备")
|
||
etag = sizeStr
|
||
}
|
||
} else {
|
||
log.Warnf("意外的状态代码: %d 当 ping URL 时: %s", resp.StatusCode, url)
|
||
}
|
||
|
||
return etag, size, lastModified
|
||
}
|