UploadFile:

Author: devfeel

context.go

@@ -152,10 +152,7 @@ func (ctx *HttpContext) FormFile(key string) (*UploadFile, error) {
 	if err != nil {
 		return nil, err
 	} else {
-		return &UploadFile{
-			File:   file,
-			Header: header,
-		}, nil
+		return NewUploadFile(file, header), nil
 	}
 }
 

example/uploadfile/main.go

@@ -49,7 +49,8 @@ func FileUpload(ctx *dotweb.HttpContext) {
 			ctx.WriteString("SaveFile error => " + err.Error())
 			return
 		} else {
-			ctx.WriteString("SaveFile success || " + upload.FileName() + " || " + fmt.Sprint(upload.Size()))
+			ctx.WriteString("SaveFile success || " + upload.FileName() + " || " + upload.GetFileExt() + " || " + fmt.Sprint(upload.Size()))
+
 			return
 		}
 	}

framework/file/file.go

@@ -15,3 +15,17 @@ func GetCurrentDirectory() string {
 	}
 	return strings.Replace(dir, "\\", "/", -1)
 }
+
+//get filename extensions
+func GetFileExt(fileName string) string {
+	if fileName == "" {
+		return ""
+	} else {
+		index := strings.LastIndex(fileName, ".")
+		if index < 0 {
+			return ""
+		} else {
+			return string(fileName[index:])
+		}
+	}
+}

uploadfile.go

@@ -2,6 +2,7 @@ package dotweb
 
 import (
 	"errors"
+	files "github.com/devfeel/dotweb/framework/file"
 	"io"
 	"mime/multipart"
 	"os"
@@ -10,17 +11,28 @@ import (
 type UploadFile struct {
 	File     multipart.File
 	Header   *multipart.FileHeader
+	fileExt  string //file extensions
+	fileName string
 	fileSize int64
 }
 
+func NewUploadFile(file multipart.File, header *multipart.FileHeader) *UploadFile {
+	return &UploadFile{
+		File:     file,
+		Header:   header,
+		fileName: header.Filename,
+		fileExt:  files.GetFileExt(header.Filename),
+	}
+}
+
 // 获取文件大小的接口
 type Sizer interface {
 	Size() int64
 }
 
 //get upload file client-local name
 func (f *UploadFile) FileName() string {
-	return f.Header.Filename
+	return f.fileName
 }
 
 //get upload file size
@@ -48,3 +60,8 @@ func (f *UploadFile) SaveFile(fileName string) (size int64, err error) {
 	size, err = io.Copy(fileWriter, f.File)
 	return size, err
 }
+
+//get upload file extensions
+func (f *UploadFile) GetFileExt() string {
+	return f.fileExt
+}