新增配置管理功能及多存储支持

- 添加管理员端 API,用于获取和更新完整配置。
- 添加公共端 API,用于获取非敏感配置信息。
- 增加本地存储(LocalStorage)、S3(S3Storage)、和 WebDAV(WebDAVStorage)存储类型的实现。
- 支持热更新存储配置和保存配置文件至磁盘。
- 更新 Swagger 文档以包含新接口定义。
This commit is contained in:
2026-01-14 14:15:20 +08:00
parent 1ffa16cf48
commit fe656fb298
13 changed files with 1043 additions and 6 deletions

47
internal/storage/local.go Normal file
View File

@@ -0,0 +1,47 @@
package storage
import (
"context"
"io"
"os"
"path/filepath"
)
type LocalStorage struct {
RootPath string
}
func NewLocalStorage(rootPath string) *LocalStorage {
// 确保根目录存在
if _, err := os.Stat(rootPath); os.IsNotExist(err) {
os.MkdirAll(rootPath, 0755)
}
return &LocalStorage{RootPath: rootPath}
}
func (s *LocalStorage) Save(ctx context.Context, path string, reader io.Reader) error {
fullPath := filepath.Join(s.RootPath, path)
dir := filepath.Dir(fullPath)
if _, err := os.Stat(dir); os.IsNotExist(err) {
os.MkdirAll(dir, 0755)
}
file, err := os.Create(fullPath)
if err != nil {
return err
}
defer file.Close()
_, err = io.Copy(file, reader)
return err
}
func (s *LocalStorage) Open(ctx context.Context, path string) (io.ReadCloser, error) {
fullPath := filepath.Join(s.RootPath, path)
return os.Open(fullPath)
}
func (s *LocalStorage) Delete(ctx context.Context, path string) error {
fullPath := filepath.Join(s.RootPath, path)
return os.Remove(fullPath)
}

72
internal/storage/s3.go Normal file
View File

@@ -0,0 +1,72 @@
package storage
import (
"context"
"io"
"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/config"
"github.com/aws/aws-sdk-go-v2/credentials"
"github.com/aws/aws-sdk-go-v2/service/s3"
)
type S3Storage struct {
client *s3.Client
bucket string
}
func NewS3Storage(ctx context.Context, endpoint, region, accessKey, secretKey, bucket string, useSSL bool) (*S3Storage, error) {
customResolver := aws.EndpointResolverWithOptionsFunc(func(service, region string, options ...interface{}) (aws.Endpoint, error) {
if endpoint != "" {
return aws.Endpoint{
URL: endpoint,
SigningRegion: region,
HostnameImmutable: true,
}, nil
}
return aws.Endpoint{}, &aws.EndpointNotFoundError{}
})
cfg, err := config.LoadDefaultConfig(ctx,
config.WithRegion(region),
config.WithCredentialsProvider(credentials.NewStaticCredentialsProvider(accessKey, secretKey, "")),
config.WithEndpointResolverWithOptions(customResolver),
)
if err != nil {
return nil, err
}
client := s3.NewFromConfig(cfg)
return &S3Storage{
client: client,
bucket: bucket,
}, nil
}
func (s *S3Storage) Save(ctx context.Context, path string, reader io.Reader) error {
_, err := s.client.PutObject(ctx, &s3.PutObjectInput{
Bucket: aws.String(s.bucket),
Key: aws.String(path),
Body: reader,
})
return err
}
func (s *S3Storage) Open(ctx context.Context, path string) (io.ReadCloser, error) {
output, err := s.client.GetObject(ctx, &s3.GetObjectInput{
Bucket: aws.String(s.bucket),
Key: aws.String(path),
})
if err != nil {
return nil, err
}
return output.Body, nil
}
func (s *S3Storage) Delete(ctx context.Context, path string) error {
_, err := s.client.DeleteObject(ctx, &s3.DeleteObjectInput{
Bucket: aws.String(s.bucket),
Key: aws.String(path),
})
return err
}

View File

@@ -0,0 +1,14 @@
package storage
import (
"context"
"io"
)
type Storage interface {
Save(ctx context.Context, path string, reader io.Reader) error
Open(ctx context.Context, path string) (io.ReadCloser, error)
Delete(ctx context.Context, path string) error
}
var GlobalStorage Storage

View File

@@ -0,0 +1,54 @@
package storage
import (
"context"
"io"
"path/filepath"
"strings"
"github.com/studio-b12/gowebdav"
)
type WebDAVStorage struct {
client *gowebdav.Client
root string
}
func NewWebDAVStorage(url, username, password, root string) *WebDAVStorage {
client := gowebdav.NewClient(url, username, password)
return &WebDAVStorage{
client: client,
root: root,
}
}
func (s *WebDAVStorage) getFullPath(path string) string {
return filepath.ToSlash(filepath.Join(s.root, path))
}
func (s *WebDAVStorage) Save(ctx context.Context, path string, reader io.Reader) error {
fullPath := s.getFullPath(path)
// Ensure directory exists
dir := filepath.Dir(fullPath)
if dir != "." && dir != "/" {
parts := strings.Split(strings.Trim(dir, "/"), "/")
current := ""
for _, part := range parts {
current += "/" + part
_ = s.client.Mkdir(current, 0755)
}
}
return s.client.WriteStream(fullPath, reader, 0644)
}
func (s *WebDAVStorage) Open(ctx context.Context, path string) (io.ReadCloser, error) {
fullPath := s.getFullPath(path)
return s.client.ReadStream(fullPath)
}
func (s *WebDAVStorage) Delete(ctx context.Context, path string) error {
fullPath := s.getFullPath(path)
return s.client.Remove(fullPath)
}