mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-17 16:56:38 +08:00
43 lines
1.0 KiB
Go
43 lines
1.0 KiB
Go
package main
|
|
|
|
import (
|
|
"chatplus/store/model"
|
|
"chatplus/utils"
|
|
"fmt"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
"log"
|
|
"os"
|
|
"path"
|
|
)
|
|
|
|
func main() {
|
|
MysqlDns := "root:12345678@tcp(localhost:3306)/chatgpt_plus?charset=utf8mb4&collation=utf8mb4_unicode_ci&parseTime=True&loc=Local"
|
|
db, err := gorm.Open(mysql.Open(MysqlDns), &gorm.Config{})
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
_ = os.MkdirAll("static/upload/images", 0755)
|
|
var jobs []model.MidJourneyJob
|
|
db.Find(&jobs)
|
|
for _, job := range jobs {
|
|
basename := path.Base(job.ImgURL)
|
|
imageData, err := utils.DownloadImage(job.ImgURL, "")
|
|
if err != nil {
|
|
fmt.Println("图片下载失败：" + job.ImgURL)
|
|
continue
|
|
}
|
|
newImagePath := fmt.Sprintf("static/upload/images/%s", basename)
|
|
err = os.WriteFile(newImagePath, imageData, 0644)
|
|
if err != nil {
|
|
fmt.Println("Error writing image file:", err)
|
|
continue
|
|
}
|
|
fmt.Println("图片保存成功！", newImagePath)
|
|
// 更新数据库
|
|
job.ImgURL = fmt.Sprintf("http://localhost:5678/%s", newImagePath)
|
|
db.Updates(&job)
|
|
}
|
|
|
|
}
|