Watch Star Fork

进击的皇虫 / DocHubGo

加入码云
与超过 200 万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
参考百度文库,使用Beego(Golang)开发的开源文库系统 http://wenku.it
一键复制 编辑 原始数据 按行查看 历史
main.go 3.85 KB TruthHun 提交于 2018-05-15 20:09 . 解决由于缺失lazybug.me下的包导致无法编译的问题
package main
import (
"fmt"
"time"
"github.com/TruthHun/DocHub/controllers/HomeControllers"
"github.com/TruthHun/DocHub/helper"
"github.com/TruthHun/DocHub/models"
_ "github.com/TruthHun/DocHub/routers"
"github.com/astaxie/beego"
"github.com/astaxie/beego/orm"
)
//初始化函数
func init() {
//数据库初始化
models.Init()
//初始化日志
helper.InitLogs()
//初始化分词器
go InitSego()
beego.AddFuncMap("TimestampFormat", helper.TimestampFormat)
beego.AddFuncMap("Interface2Int", helper.Interface2Int)
beego.AddFuncMap("Interface2String", helper.Interface2String)
beego.AddFuncMap("Default", helper.Default)
beego.AddFuncMap("FormatByte", helper.FormatByte)
beego.AddFuncMap("CalcInt", helper.CalcInt)
beego.AddFuncMap("StarVal", helper.StarVal)
beego.AddFuncMap("Equal", helper.Equal)
beego.AddFuncMap("SimpleList", SimlpeList) //简易的文档列表
beego.AddFuncMap("HandlePageNum", HandlePageNum) //处理文档页码为0的显示问题
beego.AddFuncMap("DoesCollect", DoesCollect) //判断用户是否已收藏了该文档
beego.AddFuncMap("DoesSign", DoesSign) //用户今日是否已签到
beego.AddFuncMap("Friends", Friends) //友情链接
beego.AddFuncMap("DefPic", models.ModelOss.DefaultPicture) //获取默认图片
beego.AddFuncMap("CategoryName", models.ModelCategory.GetTitleById) //根据分类id获取分类名称
beego.AddFuncMap("IsIllegal", models.ModelDoc.IsIllegal) //根据md5判断文档是否是非法文档
beego.AddFuncMap("IsRemark", models.ModelDocRemark.IsRemark) //根据文档是否存在备注
beego.AddFuncMap("Xmd5", helper.Xmd5) //xmd5,MD5扩展加密
beego.AddFuncMap("BuildURL", helper.BuildURL) //创建URL
beego.AddFuncMap("HeightLight", helper.HeightLight) //高亮
beego.AddFuncMap("ReportReason", helper.ReportReason) //举报原因
beego.AddFuncMap("GetDescByMd5", models.ModelDocText.GetDescByMd5)
beego.AddFuncMap("GetDescByDsId", models.ModelDocText.GetDescByDsId)
beego.AddFuncMap("GetDescByDid", models.ModelDocText.GetDescByDid)
}
func main() {
fmt.Println("")
fmt.Println("Powered By DocHub")
fmt.Println("Author:进击的皇虫(TruthHun@QQ.COM)")
fmt.Println("")
//定义错误和异常处理控制器
beego.ErrorController(&HomeControllers.BaseController{})
beego.Run()
}
//初始化分词词典
func InitSego() {
fmt.Println("加载分词词典...")
helper.Segmenter.LoadDictionary("./dictionary/dictionary.txt")
fmt.Println("分词词典加载完成")
}
//文档简易列表
func SimlpeList(chinelid interface{}) []orm.Params {
data, _, _ := models.ModelDoc.SimpleList(fmt.Sprintf("di.ChanelId=%v", helper.Interface2Int(chinelid)), 5)
return data
}
//处理页数
func HandlePageNum(PageNum interface{}) string {
pn := fmt.Sprintf("%v", PageNum)
if pn == "0" {
return " -- "
}
return pn
}
//是否已收藏文档
func DoesCollect(did, uid int) bool {
if uid == 0 {
return false
}
var params []orm.Params
sql := fmt.Sprintf("select c.Id from %v cf left join %v c on c.cid=cf.id where c.Did=? and cf.Uid=? limit 1", models.GetTable("collect_folder"), models.GetTable("collect"))
rows, err := models.O.Raw(sql, did, uid).Values(&params)
if err != nil {
helper.Logger.Error(err.Error())
}
if rows > 0 && err == nil {
return true
}
return false
}
//检测用户今天是否已签到
func DoesSign(uid int) bool {
if _, rows, err := models.GetList("sign", 1, 1, orm.NewCondition().And("Date", time.Now().Format("20060102")).And("Uid", uid)); err == nil && rows > 0 {
return true
}
return false
}
//获取友链
func Friends() []orm.Params {
rows, _, _ := models.GetList("friend", 1, 100, orm.NewCondition().And("Status", 1), "Sort")
return rows
}

评论 ( 0 )

你可以在登录后,发表评论

5th_float_left_gitee 5th_float_left_close