Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-05-13 11:07:34 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

文件判断修改

Browse Source
This commit is contained in:
lifeiyi
2020-03-11 16:35:22 +08:00
parent 4926100b5b
commit be5b1b8999
1 changed files with 32 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
sysservice/originhttp/httpserverervice.go
View File

@@ -35,7 +35,7 @@ type HttpRequest struct {
Header http.Header Header http.Header
Body string Body string
ParamStr string //http://127.0.0.1:7001/aaa/bbb?aa=1中的aa=1部分 ParamStr string //http://127.0.0.1:7001/aaa/bbb?aa=1中的aa=1部分
mapParam map[string]string mapParam map[string]string
URL string URL string
//Req http.Request //Req http.Request
@@ -49,7 +49,7 @@ type HttpRespone struct {
type ServeHTTPRouterMux struct { type ServeHTTPRouterMux struct {
httpfiltrateList []HttpFiltrate httpfiltrateList []HttpFiltrate
allowOrigin bool allowOrigin bool
} }
type ControllerMapsType map[string]reflect.Value type ControllerMapsType map[string]reflect.Value
@@ -205,7 +205,7 @@ func (slf *ServeHTTPRouterMux) ServeHTTP(w http.ResponseWriter, r *http.Request)
w.Header().Set("Access-Control-Allow-Origin", "*") w.Header().Set("Access-Control-Allow-Origin", "*")
w.Header().Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE") w.Header().Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE")
w.Header().Set("Access-Control-Allow-Headers", w.Header().Set("Access-Control-Allow-Headers",
"Action, Module") //有使用自定义头 需要这个,Action, Module是例子 "Action, Module") //有使用自定义头 需要这个,Action, Module是例子
} }
} }
@@ -382,8 +382,8 @@ func staticServer(routerUrl string, routerData RouterStaticResoutceData, w http.
} }
//上传资源 //上传资源
case "POST": case "POST":
// 在这儿处理例外路由接口 // 在这儿处理例外路由接口
/* /*
var errRet error var errRet error
for _, filter := range slf.httpfiltrateList { for _, filter := range slf.httpfiltrateList {
ret := filter(r.URL.Path, w, r) ret := filter(r.URL.Path, w, r)
@@ -398,33 +398,33 @@ func staticServer(routerUrl string, routerData RouterStaticResoutceData, w http.
w.Write([]byte(errRet.Error())) w.Write([]byte(errRet.Error()))
return return
}*/ }*/
r.ParseMultipartForm(32 << 20) // max memory is set to 32MB r.ParseMultipartForm(32 << 20) // max memory is set to 32MB
resourceFile, resourceFileHeader, err := r.FormFile("file") resourceFile, resourceFileHeader, err := r.FormFile("file")
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
writeResp(http.StatusNotFound, err.Error()) writeResp(http.StatusNotFound, err.Error())
return return
} }
defer resourceFile.Close() defer resourceFile.Close()
//重新拼接文件名 //重新拼接文件名
imgFormat := strings.Split(resourceFileHeader.Filename, ".") imgFormat := strings.Split(resourceFileHeader.Filename, ".")
if len(imgFormat) != 2 { if len(imgFormat) < 2 {
writeResp(http.StatusNotFound, "not a file") writeResp(http.StatusNotFound, "not a file")
return return
} }
filePrefixName := uuid.Rand().HexEx() filePrefixName := uuid.Rand().HexEx()
fileName := filePrefixName + "." + imgFormat[1] fileName := filePrefixName + "." + imgFormat[len(imgFormat)-1]
//创建文件 //创建文件
localpath := fmt.Sprintf("%s%s", destLocalPath, fileName) localpath := fmt.Sprintf("%s%s", destLocalPath, fileName)
localfd, err := os.OpenFile(localpath, os.O_WRONLY|os.O_CREATE, 0666) localfd, err := os.OpenFile(localpath, os.O_WRONLY|os.O_CREATE, 0666)
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
writeResp(http.StatusNotFound, "upload fail") writeResp(http.StatusNotFound, "upload fail")
return return
} }
defer localfd.Close() defer localfd.Close()
io.Copy(localfd, resourceFile) io.Copy(localfd, resourceFile)
writeResp(http.StatusOK, upath+"/"+fileName) writeResp(http.StatusOK, upath+"/"+fileName)
} }
} }