完善功能

utility/oss/oss.go

@@ -1,26 +1,16 @@
 package oss
 
-import (
-	"context"
-	"server/internal/model"
-)
-
-// OssClient 是所有云存储平台要实现的统一接口
-type OssClient interface {
-	UploadFile(ctx context.Context, in *model.OssUploadFileInput) (out *model.OssOutput, err error)
-	GetFileURL(ctx context.Context, in *model.OssGetFileInput) (out *model.OssOutput, err error)
+type OSSClient interface {
+	Upload(file interface{}) (string, error)
+	UploadLocalFile(path, name string) (string, error)
 }
 
-// registry 存储各个平台的实现
-var clients = make(map[string]OssClient)
+var clients = make(map[string]OSSClient)
 
-// Register 用于注册平台实现
-func Register(name string, client OssClient) {
+func RegisterClient(name string, client OSSClient) {
 	clients[name] = client
 }
 
-// GetClient 获取指定平台的实现
-func GetClient(name string) (OssClient, bool) {
-	client, ok := clients[name]
-	return client, ok
+func GetOSSClient(name string) OSSClient {
+	return clients[name]
 }