完善评论功能

models/comment_vote.go

@@ -0,0 +1,47 @@
+package models
+
+import (
+	"time"
+	"github.com/lifei6671/godoc/conf"
+	"github.com/astaxie/beego/orm"
+)
+
+type CommentVote struct {
+	VoteId int			`orm:"column(vote_id);pk;auto;unique" json:"vote_id"`
+	CommentId int			`orm:"column(comment_id);type(int);index" json:"comment_id"`
+	CommentMemberId int 		`orm:"column(comment_member_id);type(int);index;default(0)" json:"comment_member_id"`
+	VoteMemberId int		`orm:"column(vote_member_id);type(int);index" json:"vote_member_id"`
+	VoteState int 			`orm:"column(vote_state);type(int)" json:"vote_state"`
+	CreateTime time.Time		`orm:"column(create_time);type(datetime);auto_now_add" json:"create_time"`
+}
+
+// TableName 获取对应数据库表名.
+func (m *CommentVote) TableName() string {
+	return "comments"
+}
+// TableEngine 获取数据使用的引擎.
+func (m *CommentVote) TableEngine() string {
+	return "INNODB"
+}
+
+func (m *CommentVote) TableNameWithPrefix() string  {
+	return conf.GetDatabasePrefix() + m.TableName()
+}
+func (u *CommentVote) TableUnique() [][]string {
+	return [][]string{
+		[]string{"comment_id", "vote_member_id"},
+	}
+}
+
+func (m *CommentVote) InsertOrUpdate() (*CommentVote,error)  {
+	o := orm.NewOrm()
+
+	if m.VoteId > 0 {
+		_,err := o.Update(m)
+		return m,err
+	}else{
+		_,err := o.Insert(m)
+
+		return m,err
+	}
+}