mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-12-04 04:38:04 +08:00
48 lines
978 B
Go
48 lines
978 B
Go
package upload
|
|
|
|
import (
|
|
httputil "github.com/goodrain/rainbond/util/http"
|
|
"net/http"
|
|
)
|
|
|
|
type Storage struct {
|
|
output string
|
|
verbosity int
|
|
}
|
|
|
|
func (s *Storage) StorageDir() string {
|
|
return s.output
|
|
}
|
|
|
|
func NewStorage(rootDir string) *Storage {
|
|
return &Storage{output: rootDir}
|
|
}
|
|
|
|
// UploadHandler is the endpoint for uploading and storing files.
|
|
func (s *Storage) UploadHandler(w http.ResponseWriter, r *http.Request) {
|
|
|
|
// Performs the processing of writing data into chunk files.
|
|
files, err := process(r, s.StorageDir())
|
|
|
|
if err == incomplete {
|
|
httputil.ReturnSuccess(r, w, nil)
|
|
return
|
|
}
|
|
if err != nil {
|
|
httputil.ReturnError(r, w, 500, err.Error())
|
|
return
|
|
}
|
|
|
|
data := make([]map[string]interface{}, 0)
|
|
|
|
for _, file := range files {
|
|
attachment, err := create(s.StorageDir(), file, true)
|
|
if err != nil {
|
|
httputil.ReturnError(r, w, 500, err.Error())
|
|
return
|
|
}
|
|
data = append(data, attachment.ToJson())
|
|
}
|
|
httputil.ReturnSuccess(r, w, data)
|
|
}
|