UploadController.go

package HomeControllers

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"
	"time"

	"github.com/astaxie/beego/orm"

	"DocHub/helper"
	"DocHub/helper/conv"
	"DocHub/models"
)

type UploadController struct {
	BaseController
}

//分词
func (this *UploadController) SegWord() {
	var wds string
	if this.IsLogin > 0 {
		wds = helper.SegWord(this.GetString("word"))
	}
	this.ResponseJson(true, "分词成功", wds)
}

//文档上传页面
func (this *UploadController) Get() {
	cond := orm.NewCondition().And("status", 1)
	data, _, _ := models.GetList(models.GetTableCategory(), 1, 2000, cond, "Sort", "Title")
	this.Xsrf()
	this.Data["Seo"] = models.NewSeo().GetByPage("PC-Upload", "文档上传-文档分享", "文档上传,文档分享", "文档上传-文档分享", this.Sys.Site)
	this.Data["Cates"], _ = conv.InterfaceToJson(data)
	this.Data["json"] = data
	this.Data["IsUpload"] = true
	this.Data["PageId"] = "wenku-upload"
	this.Data["MaxSize"] = models.NewSys().GetByField("MaxFile").MaxFile
	this.TplName = "index.html"
}

//文档执行操作
//处理流程：
//1、检测用户是否已登录，未登录不允许上传
//2、检测是否存在了该文档的md5，如果已存在，则根据md5查询存储在文档存档表中的数据；如果文档已经在文档存储表中存在，则该文档不需要再获取封面、大小、页码等数据
//3、检测文档格式是否符合要求。
//4、计算文档md5，然后根据md5再比对一次文档是否在存档表中存在
//5、文档未存在，则将文档数据录入文档存储表(document_store)
//6、执行文档转pdf，并获取文档页数、封面、摘要等
//7、获取文档大小
func (this *UploadController) Post() {
	var (
		ext  string //文档扩展名
		dir  = fmt.Sprintf("./uploads/%v/%v", time.Now().Format("2006/01/02"), this.IsLogin)
		form models.FormUpload
		err  error
	)

	if this.IsLogin == 0 {
		this.ResponseJson(false, "您当前未登录，请先登录")
	}

	this.ParseForm(&form)

	//文件在文档库中未存在，则接收文件并做处理
	f, fh, err := this.GetFile("File")
	if err == nil {
		defer f.Close()
		os.MkdirAll(dir, os.ModePerm)
		ext = strings.ToLower(filepath.Ext(fh.Filename))
		if _, ok := helper.AllowedUploadDocsExt[ext]; !ok {
			this.ResponseJson(false, "您上传的文档格式不正确，请上传正确格式的文档")
		}
		file := fmt.Sprintf("%v-%v-%v", this.IsLogin, time.Now().UnixNano(), ext)
		form.TmpFile = filepath.Join(dir, file)
		form.Ext = ext
		err = this.SaveToFile("File", form.TmpFile)
		if err != nil {
			helper.Logger.Error(err.Error())
			this.ResponseJson(false, "文件保存失败")
		}
	}

	// 文档处理
	err = models.DocumentProcess(this.IsLogin, form)
	if err != nil {
		this.ResponseJson(false, err.Error())
	}
	this.ResponseJson(true, "恭喜您，文档上传成功")
}