52 lines
1002 B
Go
52 lines
1002 B
Go
package system
|
|
|
|
import (
|
|
"ai4m/model/common/response"
|
|
"ai4m/model/system"
|
|
"github.com/gin-gonic/gin"
|
|
"os"
|
|
)
|
|
|
|
type SFTPApi struct{}
|
|
|
|
func (sftp *SFTPApi) SFTPUpload(c *gin.Context) {
|
|
var sftpObj system.SFTPObj
|
|
err := c.Bind(&sftpObj)
|
|
if err != nil {
|
|
response.FailWithMessage(err.Error(), c)
|
|
return
|
|
}
|
|
pwd, err := os.Getwd()
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = c.SaveUploadedFile(sftpObj.File, pwd+"\\"+sftpObj.File.Filename)
|
|
if err != nil {
|
|
response.FailWithMessage(err.Error(), c)
|
|
return
|
|
}
|
|
err = sftpService.SFTPUpload(sftpObj)
|
|
|
|
if err != nil {
|
|
response.FailWithMessage(err.Error(), c)
|
|
return
|
|
}
|
|
response.OkWithData("上传成功！", c)
|
|
}
|
|
|
|
func (sftp *SFTPApi) SFTPDownload(c *gin.Context) {
|
|
var sftpObj system.SFTPObj
|
|
err := c.Bind(&sftpObj)
|
|
if err != nil {
|
|
response.FailWithMessage(err.Error(), c)
|
|
return
|
|
}
|
|
resFile, err := sftpService.SFTPDownload(sftpObj)
|
|
if err != nil {
|
|
response.FailWithMessage(err.Error(), c)
|
|
return
|
|
}
|
|
|
|
response.OkWithFile(resFile, c)
|
|
}
|