多态方案实现文件上传管理

在后台系统的文件上传功能中，常常需要根据文件类型、文件大小、存储位置等不同的需求实现不同的上传方式。通过使用多态（Polymorphism）设计模式，可以实现灵活且可扩展的文件上传管理方案。本文将介绍如何在Go语言中利用多态方案实现文件上传的管理。

1. 多态的概念

多态是面向对象编程中的一项重要特性，它允许不同的对象以相同的接口进行操作。通过多态，程序可以在运行时决定调用哪个具体的实现，增加了代码的灵活性和可扩展性。

在文件上传管理中，我们可以根据文件类型（如图片、视频、文档等）定义不同的上传策略，每种策略的实现是一个具体的上传方式。这些上传方式都实现一个统一的接口，通过多态机制动态调用相应的上传方式。

2. 多态方案实现文件上传管理

我们将实现一个文件上传的多态方案，其中包含多个文件上传策略（如本地存储上传、云存储上传等），每种上传策略都实现一个统一的接口。根据文件的不同需求，动态选择合适的上传策略进行处理。

2.1 定义上传接口

首先，我们定义一个Uploader接口，所有的上传方式都需要实现这个接口。接口中包含一个Upload方法，用于执行具体的上传操作。

package main

import "fmt"

// 定义文件上传接口
type Uploader interface {
	Upload(fileName string, fileData []byte) error
}

2.2 本地存储上传策略

接下来，我们实现一个本地存储上传策略，它将文件保存到服务器的本地目录中。

import (
	"os"
	"path/filepath"
)

// 本地存储上传实现
type LocalUploader struct{}

// 实现Uploader接口的Upload方法
func (l *LocalUploader) Upload(fileName string, fileData []byte) error {
	// 创建文件路径
	filePath := filepath.Join("uploads", fileName)
	// 创建文件
	file, err := os.Create(filePath)
	if err != nil {
		return fmt.Errorf("无法创建文件: %v", err)
	}
	defer file.Close()

	// 写入文件内容
	_, err = file.Write(fileData)
	if err != nil {
		return fmt.Errorf("写入文件失败: %v", err)
	}

	fmt.Println("文件上传到本地存储成功:", filePath)
	return nil
}

2.3 云存储上传策略

然后，我们实现一个云存储上传策略，它将文件上传到云存储平台（如阿里云OSS、AWS S3等）。此处使用一个模拟的云存储上传方式。

// 云存储上传实现
type CloudUploader struct{}

// 实现Uploader接口的Upload方法
func (c *CloudUploader) Upload(fileName string, fileData []byte) error {
	// 模拟云存储上传操作
	fmt.Println("文件上传到云存储平台成功:", fileName)
	return nil
}

2.4 文件上传管理器

为了管理不同的上传策略，我们创建一个UploadManager结构体，它根据不同的条件（如文件类型、文件大小等）选择合适的上传策略。

type UploadManager struct {
	uploader Uploader // 用于保存具体的上传策略
}

// 设置具体的上传策略
func (m *UploadManager) SetUploader(u Uploader) {
	m.uploader = u
}

// 执行文件上传
func (m *UploadManager) UploadFile(fileName string, fileData []byte) error {
	return m.uploader.Upload(fileName, fileData)
}

2.5 文件上传的使用示例

接下来，演示如何根据不同的需求选择不同的上传策略，并使用UploadManager进行上传操作。

func main() {
	// 模拟上传的文件数据
	fileName := "example.txt"
	fileData := []byte("这是文件内容")

	// 初始化文件上传管理器
	manager := &UploadManager{}

	// 根据条件选择上传策略（这里假设为本地存储上传）
	manager.SetUploader(&LocalUploader{})
	if err := manager.UploadFile(fileName, fileData); err != nil {
		fmt.Println("文件上传失败:", err)
	}

	// 模拟选择云存储上传策略
	manager.SetUploader(&CloudUploader{})
	if err := manager.UploadFile(fileName, fileData); err != nil {
		fmt.Println("文件上传失败:", err)
	}
}