package alyoss

import (
	"bytes"
	"context"
	filechunk "gitee.com/h79/goutils/common/file/chunk"
	fileconfig "gitee.com/h79/goutils/common/file/config"
	"gitee.com/h79/goutils/common/logger"
	"gitee.com/h79/goutils/common/result"
	"github.com/aliyun/aliyun-oss-go-sdk/oss"
	"sync"
)

// MultiUpload 分片处理
func (of *OssFile) MultiUpload(ctx context.Context, buff []byte, bucket *oss.Bucket, fileResult *fileconfig.Result) error {
	imur, err := bucket.InitiateMultipartUpload(of.bf.Key)
	if err != nil {
		return result.Errorf(result.ErrUpload, "[File]Bucket InitiateMultipartUpload %s", err)
	}
	size := int64(len(buff))
	partNum := filechunk.Num(size, of.config.MinChunkSize)
	chunks := filechunk.Split(size, partNum)

	// 启动多个goroutine进行分片上传
	var wg sync.WaitGroup
	var parts []oss.UploadPart
	var partMu sync.Mutex
	for i := 0; i < len(chunks); i++ {
		wg.Add(1)
		go func(chunk *filechunk.Chunk) {
			defer wg.Done()
			partBuff := bytes.NewReader(buff[chunk.Offset:chunk.Length()])
			part, er := bucket.UploadPart(imur, partBuff, chunk.Size, chunk.Number)
			if er != nil {
				return
			}
			chunk.ETag = part.ETag
			chunk.Done = true
			partMu.Lock()
			parts = append(parts, part)
			partMu.Unlock()
		}(&chunks[i])
	}
	wg.Wait()

	// 完成分片上传
	cmu, err := bucket.CompleteMultipartUpload(imur, parts)
	if err != nil {
		return result.Errorf(result.ErrUpload, "[File]Bucket complete part %s", err)
	}
	logger.I("UploadFile", "Complete part %v", cmu)
	fileResult.TaskId = imur.UploadID
	fileResult.Key = cmu.Key
	fileResult.Url = cmu.Location
	fileResult.Tag = cmu.ETag
	return nil
}