background/model/blog.go

178 lines
4.1 KiB
Go
Raw Normal View History

2019-05-16 10:05:20 +00:00
package model
import (
"background/db"
"background/logs"
"fmt"
2019-10-05 17:25:44 +00:00
"strings"
2020-08-23 18:08:26 +00:00
2020-09-26 04:30:23 +00:00
"github.com/pkg/errors"
2020-08-23 18:08:26 +00:00
"qiniupkg.com/x/log.v7"
2019-05-16 10:05:20 +00:00
)
type Doc struct {
2020-08-23 18:08:26 +00:00
ID int64 `sql:"id" json:"id"`
Title string `sql:"title" json:"title"`
Type int64 `sql:"type" json:"type"`
Content string `sql:"content" json:"content"`
Author string `sql:"author" json:"author"`
IsPublic int `sql:"is_public" json:"is_public"`
2019-05-17 09:45:50 +00:00
}
2019-05-16 10:05:20 +00:00
2020-07-22 16:06:07 +00:00
type DocGroup struct {
2020-08-23 18:08:26 +00:00
Int int32 `sql:"int"`
2020-07-22 16:06:07 +00:00
Name string `sql:"name"`
}
2019-05-17 09:45:50 +00:00
type ArticleType struct {
2020-08-23 18:08:26 +00:00
Id int64 `sql:"id" json:"id"`
Name string `sql:"type_name" json:"type_name"`
Author string `sql:"author" json:"author"`
Group int32 `sql:"group" json:"group"`
2020-07-22 16:06:07 +00:00
GroupName string `json:"group_name"`
2019-05-17 09:45:50 +00:00
}
2019-06-30 14:59:00 +00:00
func GetArticlesType() []ArticleType {
2019-05-17 09:45:50 +00:00
ret := []ArticleType{}
2020-01-25 14:26:41 +00:00
sql := fmt.Sprintf("select * from doc_type")
2020-09-26 04:30:23 +00:00
e := db.GetMysqlClient().Query2(sql, &ret)
log.Print(ret)
2020-08-23 18:08:26 +00:00
for k, _ := range ret {
2020-07-22 16:06:07 +00:00
group := []DocGroup{}
2020-08-23 18:08:26 +00:00
sql = fmt.Sprintf("select * from doc_group where doc_group.int = %d", ret[k].Group)
2020-09-26 04:30:23 +00:00
db.GetMysqlClient().Query2(sql, &group)
2020-08-23 18:08:26 +00:00
if len(group) > 0 {
2020-07-22 16:06:07 +00:00
ret[k].GroupName = group[0].Name
}
}
2019-06-30 14:59:00 +00:00
if nil != e {
2019-05-17 09:45:50 +00:00
logs.Error(e.Error())
return nil
2019-05-16 10:05:20 +00:00
}
2019-05-17 09:45:50 +00:00
return ret
}
2020-01-25 14:26:41 +00:00
2019-05-17 09:45:50 +00:00
func CreateDoc(doc Doc) error {
2020-03-26 12:15:04 +00:00
sql := fmt.Sprintf(`INSERT INTO doc ( doc.title, doc.content, doc.author, doc.type,doc.is_public) SELECT
2020-01-20 11:19:08 +00:00
'%s',
'%s',
'%s',
2020-03-26 12:15:04 +00:00
%d ,
%d
2020-01-20 11:19:08 +00:00
FROM
DUAL
WHERE
2020-03-26 12:15:04 +00:00
NOT EXISTS ( SELECT * FROM doc WHERE doc.title = '%s' );`, doc.Title, strings.Replace(doc.Content, "'", "\\'", -1),
2020-08-23 18:08:26 +00:00
doc.Author, doc.Type, doc.IsPublic, doc.Title)
2019-06-30 14:59:00 +00:00
_, e := db.GetMysqlClient().Query(sql)
if nil != e {
2019-05-16 10:05:20 +00:00
logs.Error(e.Error())
return e
}
return nil
2019-06-30 14:59:00 +00:00
}
2019-08-01 14:52:57 +00:00
2020-08-23 18:08:26 +00:00
func UpdateDoc(doc Doc) error {
2019-08-01 14:52:57 +00:00
sql := fmt.Sprintf(`update doc set doc.author = '%s' ,doc.title = '%s',doc.type = '%d',doc.content = '%s' where doc.id = '%d'; `,
2020-08-23 18:08:26 +00:00
doc.Author, doc.Title, doc.Type,
strings.Replace(doc.Content, "'", "\\'", -1), doc.ID)
2019-08-01 14:52:57 +00:00
_, e := db.GetMysqlClient().Query(sql)
if nil != e {
logs.Error(e.Error())
return e
}
return nil
}
2019-09-06 07:53:04 +00:00
2020-08-23 18:08:26 +00:00
func DeleteDoc(id int64) error {
sql := fmt.Sprintf(`delete from doc where id = %d`, id)
2019-09-06 07:53:04 +00:00
_, e := db.GetMysqlClient().Query(sql)
if nil != e {
logs.Error(e.Error())
return e
}
return nil
2019-09-06 16:31:01 +00:00
}
2020-01-24 16:07:37 +00:00
2020-08-23 18:08:26 +00:00
func AddArticleType(t ArticleType) error {
sql := fmt.Sprintf("insert into doc_type(id,type_name,`group`) values ('%d','%s','%d')", t.Id, t.Name, t.Group)
2020-07-26 07:03:42 +00:00
log.Print(sql)
2020-01-24 16:07:37 +00:00
_, e := db.GetMysqlClient().Query(sql)
if nil != e {
logs.Error(e.Error())
return e
}
return nil
2020-01-25 17:39:31 +00:00
}
2020-08-23 18:08:26 +00:00
func UpdateArticleType(t ArticleType) error {
sql := fmt.Sprintf("update doc_type set type_name = '%s' and group = '%d' where id = %d", t.Name, t.Group, t.Id)
2020-01-26 11:11:48 +00:00
_, e := db.GetMysqlClient().Query(sql)
if nil != e {
logs.Error(e.Error())
return e
}
return nil
2020-08-23 18:08:26 +00:00
2020-01-26 11:11:48 +00:00
}
2020-08-23 18:08:26 +00:00
func DeleteArticleType(id int32) error {
sql := fmt.Sprintf("delete from doc_type where id = '%d'", id)
2020-01-25 17:39:31 +00:00
_, e := db.GetMysqlClient().Query(sql)
if nil != e {
logs.Error(e.Error())
return e
}
return nil
}
2020-08-23 18:08:26 +00:00
func GetAllDocs() ([]Doc, error) {
ret := []Doc{}
sql := fmt.Sprintf("select * from doc")
2020-08-23 18:08:26 +00:00
e := db.GetMysqlClient().Query2(sql, &ret)
if nil != e {
logs.Error(e.Error())
2020-08-23 18:08:26 +00:00
return nil, e
}
2020-08-23 18:08:26 +00:00
return ret, nil
2020-07-26 07:03:42 +00:00
}
2020-08-23 18:08:26 +00:00
func GetAllGroup() ([]DocGroup, error) {
2020-07-26 07:03:42 +00:00
ret := []DocGroup{}
sql := fmt.Sprintf("select * from doc_group")
2020-08-23 18:08:26 +00:00
e := db.GetMysqlClient().Query2(sql, &ret)
2020-07-26 07:03:42 +00:00
if nil != e {
logs.Error(e.Error())
2020-08-23 18:08:26 +00:00
return nil, e
2020-07-26 07:03:42 +00:00
}
2020-08-23 18:08:26 +00:00
return ret, nil
}
2020-09-26 04:30:23 +00:00
func GetTypeGroup(id int32) (DocGroup, error) {
ret := []DocGroup{}
sql := fmt.Sprintf("select * from doc_group where doc_group = ?", id)
e := db.GetMysqlClient().Query2(sql, &ret)
if nil != e {
logs.Error(e.Error())
return DocGroup{}, e
}
if len(ret) > 0 {
return ret[0], nil
} else {
return DocGroup{}, errors.New("no existed")
}
}
func GetGroupTypes(id int32) ([]ArticleType, error) {
ret := []ArticleType{}
sql := fmt.Sprintf("select * from doc_type where doc_type.group = %d", id)
log.Print(sql)
e := db.GetMysqlClient().Query2(sql, &ret)
if nil != e {
logs.Error(e.Error())
return nil, e
}
if len(ret) > 0 {
return ret, nil
} else {
return nil, errors.New("no existed")
}
}