chore: get ws

This commit is contained in:
zzw
2026-10-28 16:10:10 +08:00
parent 747ab25325
commit bb94b1a327
4 changed files with 6 additions and 9 deletions
+1 -1
View File
@@ -27,7 +27,7 @@ func (c *Client) SkinAnalyze(opts SkinAnalyzeOptions) (*SkinAnalyzeResponse, err
case opts.ImageURL != "": case opts.ImageURL != "":
values.Set("image_url", opts.ImageURL) values.Set("image_url", opts.ImageURL)
default: default:
return nil, errors.New("facepp: image_url, image_file or image_base64 is required") return nil, errors.New("facepp: image is required")
} }
applyOptionalParams(values, opts) applyOptionalParams(values, opts)
+4 -4
View File
@@ -13,7 +13,7 @@ import (
alioss "github.com/aliyun/aliyun-oss-go-sdk/oss" alioss "github.com/aliyun/aliyun-oss-go-sdk/oss"
) )
const MaxUploadBytes = 128 << 20 // 128MB const MaxUploadBytes = 128 << 20
var unsafeFileChars = regexp.MustCompile(`[^\w.\-\p{Han}]+`) var unsafeFileChars = regexp.MustCompile(`[^\w.\-\p{Han}]+`)
@@ -23,12 +23,12 @@ type Config struct {
BucketName string BucketName string
Endpoint string Endpoint string
Host string Host string
Dir string // 固定前缀,如 test/chore/files/ Dir string
} }
type Result struct { type Result struct {
URL string `json:"url"` URL string `json:"url"`
Path string `json:"path"` // object key,如 test/chore/files/20260403/a.zip Path string `json:"path"`
Host string `json:"host"` Host string `json:"host"`
} }
@@ -36,7 +36,7 @@ type UploadInput struct {
Filename string Filename string
Content []byte Content []byte
ContentType string ContentType string
Dir string // 可选,覆盖配置 oss.dir Dir string
} }
func LoadConfig() Config { func LoadConfig() Config {
@@ -64,9 +64,6 @@ func expandImageToTemp(srcPath string, ratio float64) (string, error) {
func saveFaceppTrainImage(srcPath string) (string, error) { func saveFaceppTrainImage(srcPath string) (string, error) {
saveDir := utils.GetConfigString("facepp.train_image_path") saveDir := utils.GetConfigString("facepp.train_image_path")
if saveDir == utils.StringEmpty {
saveDir = "/data/facepp/train/images"
}
if err := os.MkdirAll(saveDir, os.ModePerm); err != nil { if err := os.MkdirAll(saveDir, os.ModePerm); err != nil {
return utils.StringEmpty, err return utils.StringEmpty, err
} }
@@ -6,7 +6,7 @@ import (
"lone-services/pkg/redis" "lone-services/pkg/redis"
"lone-services/pkg/utils" "lone-services/pkg/utils"
pkgws "lone-services/pkg/ws" pkgws "lone-services/pkg/ws"
"lone-services/rpc/percutaneous/pb" percutaneous "lone-services/rpc/percutaneous/pb"
"lone-services/services/percutaneous/internal/dao" "lone-services/services/percutaneous/internal/dao"
"lone-services/services/percutaneous/internal/svc" "lone-services/services/percutaneous/internal/svc"