mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
438 lines
11 KiB
Go
438 lines
11 KiB
Go
// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved.
|
|
// See License.txt for license information.
|
|
|
|
package store
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/mattermost/platform/model"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type SqlPostStore struct {
|
|
*SqlStore
|
|
}
|
|
|
|
func NewSqlPostStore(sqlStore *SqlStore) PostStore {
|
|
s := &SqlPostStore{sqlStore}
|
|
|
|
for _, db := range sqlStore.GetAllConns() {
|
|
table := db.AddTableWithName(model.Post{}, "Posts").SetKeys(false, "Id")
|
|
table.ColMap("Id").SetMaxSize(26)
|
|
table.ColMap("UserId").SetMaxSize(26)
|
|
table.ColMap("ChannelId").SetMaxSize(26)
|
|
table.ColMap("RootId").SetMaxSize(26)
|
|
table.ColMap("ParentId").SetMaxSize(26)
|
|
table.ColMap("Message").SetMaxSize(4000)
|
|
table.ColMap("Type").SetMaxSize(26)
|
|
table.ColMap("Hashtags").SetMaxSize(1000)
|
|
table.ColMap("Props").SetMaxSize(4000)
|
|
table.ColMap("Filenames").SetMaxSize(4000)
|
|
}
|
|
|
|
return s
|
|
}
|
|
|
|
func (s SqlPostStore) UpgradeSchemaIfNeeded() {
|
|
|
|
// These execs are for upgrading currently created databases to full utf8mb4 compliance
|
|
// Will be removed as seen fit for upgrading
|
|
s.GetMaster().Exec("ALTER TABLE Posts charset=utf8mb4")
|
|
s.GetMaster().Exec("ALTER TABLE Posts MODIFY COLUMN Message varchar(4000) CHARACTER SET utf8mb4")
|
|
}
|
|
|
|
func (s SqlPostStore) CreateIndexesIfNotExists() {
|
|
s.CreateIndexIfNotExists("idx_update_at", "Posts", "UpdateAt")
|
|
s.CreateIndexIfNotExists("idx_create_at", "Posts", "CreateAt")
|
|
s.CreateIndexIfNotExists("idx_channel_id", "Posts", "ChannelId")
|
|
s.CreateIndexIfNotExists("idx_root_id", "Posts", "RootId")
|
|
|
|
s.CreateFullTextIndexIfNotExists("idx_message_txt", "Posts", "Message")
|
|
s.CreateFullTextIndexIfNotExists("idx_hashtags_txt", "Posts", "Hashtags")
|
|
}
|
|
|
|
func (s SqlPostStore) Save(post *model.Post) StoreChannel {
|
|
storeChannel := make(StoreChannel)
|
|
|
|
go func() {
|
|
result := StoreResult{}
|
|
|
|
if len(post.Id) > 0 {
|
|
result.Err = model.NewAppError("SqlPostStore.Save",
|
|
"You cannot update an existing Post", "id="+post.Id)
|
|
storeChannel <- result
|
|
close(storeChannel)
|
|
return
|
|
}
|
|
|
|
post.PreSave()
|
|
if result.Err = post.IsValid(); result.Err != nil {
|
|
storeChannel <- result
|
|
close(storeChannel)
|
|
return
|
|
}
|
|
|
|
if err := s.GetMaster().Insert(post); err != nil {
|
|
result.Err = model.NewAppError("SqlPostStore.Save", "We couldn't save the Post", "id="+post.Id+", "+err.Error())
|
|
} else {
|
|
time := model.GetMillis()
|
|
|
|
if post.Type != model.POST_JOIN_LEAVE {
|
|
s.GetMaster().Exec("UPDATE Channels SET LastPostAt = ?, TotalMsgCount = TotalMsgCount + 1 WHERE Id = ?", time, post.ChannelId)
|
|
} else {
|
|
// don't update TotalMsgCount for unimportant messages so that the channel isn't marked as unread
|
|
s.GetMaster().Exec("UPDATE Channels SET LastPostAt = ? WHERE Id = ?", time, post.ChannelId)
|
|
}
|
|
|
|
if len(post.RootId) > 0 {
|
|
s.GetMaster().Exec("UPDATE Posts SET UpdateAt = ? WHERE Id = ?", time, post.RootId)
|
|
}
|
|
|
|
result.Data = post
|
|
}
|
|
|
|
storeChannel <- result
|
|
close(storeChannel)
|
|
}()
|
|
|
|
return storeChannel
|
|
}
|
|
|
|
func (s SqlPostStore) Update(oldPost *model.Post, newMessage string, newHashtags string) StoreChannel {
|
|
storeChannel := make(StoreChannel)
|
|
|
|
go func() {
|
|
result := StoreResult{}
|
|
|
|
editPost := *oldPost
|
|
editPost.Message = newMessage
|
|
editPost.UpdateAt = model.GetMillis()
|
|
editPost.Hashtags = newHashtags
|
|
|
|
oldPost.DeleteAt = editPost.UpdateAt
|
|
oldPost.UpdateAt = editPost.UpdateAt
|
|
oldPost.OriginalId = oldPost.Id
|
|
oldPost.Id = model.NewId()
|
|
|
|
if result.Err = editPost.IsValid(); result.Err != nil {
|
|
storeChannel <- result
|
|
close(storeChannel)
|
|
return
|
|
}
|
|
|
|
if _, err := s.GetMaster().Update(&editPost); err != nil {
|
|
result.Err = model.NewAppError("SqlPostStore.Update", "We couldn't update the Post", "id="+editPost.Id+", "+err.Error())
|
|
} else {
|
|
time := model.GetMillis()
|
|
s.GetMaster().Exec("UPDATE Channels SET LastPostAt = ? WHERE Id = ?", time, editPost.ChannelId)
|
|
|
|
if len(editPost.RootId) > 0 {
|
|
s.GetMaster().Exec("UPDATE Posts SET UpdateAt = ? WHERE Id = ?", time, editPost.RootId)
|
|
}
|
|
|
|
// mark the old post as deleted
|
|
s.GetMaster().Insert(oldPost)
|
|
|
|
result.Data = &editPost
|
|
}
|
|
|
|
storeChannel <- result
|
|
close(storeChannel)
|
|
}()
|
|
|
|
return storeChannel
|
|
}
|
|
|
|
func (s SqlPostStore) Get(id string) StoreChannel {
|
|
storeChannel := make(StoreChannel)
|
|
|
|
go func() {
|
|
result := StoreResult{}
|
|
pl := &model.PostList{}
|
|
|
|
var post model.Post
|
|
err := s.GetReplica().SelectOne(&post, "SELECT * FROM Posts WHERE Id = ? AND DeleteAt = 0", id)
|
|
if err != nil {
|
|
result.Err = model.NewAppError("SqlPostStore.GetPost", "We couldn't get the post", "id="+id+err.Error())
|
|
}
|
|
|
|
if post.ImgCount > 0 {
|
|
post.Filenames = []string{}
|
|
for i := 0; int64(i) < post.ImgCount; i++ {
|
|
fileUrl := "/api/v1/files/get_image/" + post.ChannelId + "/" + post.Id + "/" + strconv.Itoa(i+1) + ".png"
|
|
post.Filenames = append(post.Filenames, fileUrl)
|
|
}
|
|
}
|
|
|
|
pl.AddPost(&post)
|
|
pl.AddOrder(id)
|
|
|
|
rootId := post.RootId
|
|
|
|
if rootId == "" {
|
|
rootId = post.Id
|
|
}
|
|
|
|
var posts []*model.Post
|
|
_, err = s.GetReplica().Select(&posts, "SELECT * FROM Posts WHERE (Id = ? OR RootId = ?) AND DeleteAt = 0", rootId, rootId)
|
|
if err != nil {
|
|
result.Err = model.NewAppError("SqlPostStore.GetPost", "We couldn't get the post", "root_id="+rootId+err.Error())
|
|
} else {
|
|
for _, p := range posts {
|
|
pl.AddPost(p)
|
|
}
|
|
}
|
|
|
|
result.Data = pl
|
|
|
|
storeChannel <- result
|
|
close(storeChannel)
|
|
}()
|
|
|
|
return storeChannel
|
|
}
|
|
|
|
type etagPosts struct {
|
|
Id string
|
|
UpdateAt int64
|
|
}
|
|
|
|
func (s SqlPostStore) GetEtag(channelId string) StoreChannel {
|
|
storeChannel := make(StoreChannel)
|
|
|
|
go func() {
|
|
result := StoreResult{}
|
|
|
|
var et etagPosts
|
|
err := s.GetReplica().SelectOne(&et, "SELECT Id, UpdateAt FROM Posts WHERE ChannelId = ? ORDER BY UpdateAt DESC LIMIT 1", channelId)
|
|
if err != nil {
|
|
result.Data = fmt.Sprintf("%v.0.%v", model.ETAG_ROOT_VERSION, model.GetMillis())
|
|
} else {
|
|
result.Data = fmt.Sprintf("%v.%v.%v", model.ETAG_ROOT_VERSION, et.Id, et.UpdateAt)
|
|
}
|
|
|
|
storeChannel <- result
|
|
close(storeChannel)
|
|
}()
|
|
|
|
return storeChannel
|
|
}
|
|
|
|
func (s SqlPostStore) Delete(postId string, time int64) StoreChannel {
|
|
storeChannel := make(StoreChannel)
|
|
|
|
go func() {
|
|
result := StoreResult{}
|
|
|
|
_, err := s.GetMaster().Exec("Update Posts SET DeleteAt = ?, UpdateAt = ? WHERE Id = ? OR ParentId = ? OR RootId = ?", time, time, postId, postId, postId)
|
|
if err != nil {
|
|
result.Err = model.NewAppError("SqlPostStore.Delete", "We couldn't delete the post", "id="+postId+", err="+err.Error())
|
|
}
|
|
|
|
storeChannel <- result
|
|
close(storeChannel)
|
|
}()
|
|
|
|
return storeChannel
|
|
}
|
|
|
|
func (s SqlPostStore) GetPosts(channelId string, offset int, limit int) StoreChannel {
|
|
storeChannel := make(StoreChannel)
|
|
|
|
go func() {
|
|
result := StoreResult{}
|
|
|
|
if limit > 1000 {
|
|
result.Err = model.NewAppError("SqlPostStore.GetLinearPosts", "Limit exceeded for paging", "channelId="+channelId)
|
|
storeChannel <- result
|
|
close(storeChannel)
|
|
return
|
|
}
|
|
|
|
rpc := s.getRootPosts(channelId, offset, limit)
|
|
cpc := s.getParentsPosts(channelId, offset, limit)
|
|
|
|
if rpr := <-rpc; rpr.Err != nil {
|
|
result.Err = rpr.Err
|
|
} else if cpr := <-cpc; cpr.Err != nil {
|
|
result.Err = cpr.Err
|
|
} else {
|
|
posts := rpr.Data.([]*model.Post)
|
|
parents := cpr.Data.([]*model.Post)
|
|
|
|
list := &model.PostList{Order: make([]string, 0, len(posts))}
|
|
|
|
for _, p := range posts {
|
|
if p.ImgCount > 0 {
|
|
p.Filenames = []string{}
|
|
for i := 0; int64(i) < p.ImgCount; i++ {
|
|
fileUrl := "/api/v1/files/get_image/" + p.ChannelId + "/" + p.Id + "/" + strconv.Itoa(i+1) + ".png"
|
|
p.Filenames = append(p.Filenames, fileUrl)
|
|
}
|
|
}
|
|
list.AddPost(p)
|
|
list.AddOrder(p.Id)
|
|
}
|
|
|
|
for _, p := range parents {
|
|
if p.ImgCount > 0 {
|
|
p.Filenames = []string{}
|
|
for i := 0; int64(i) < p.ImgCount; i++ {
|
|
fileUrl := "/api/v1/files/get_image/" + p.ChannelId + "/" + p.Id + "/" + strconv.Itoa(i+1) + ".png"
|
|
p.Filenames = append(p.Filenames, fileUrl)
|
|
}
|
|
}
|
|
list.AddPost(p)
|
|
}
|
|
|
|
list.MakeNonNil()
|
|
|
|
result.Data = list
|
|
}
|
|
|
|
storeChannel <- result
|
|
close(storeChannel)
|
|
}()
|
|
|
|
return storeChannel
|
|
}
|
|
|
|
func (s SqlPostStore) getRootPosts(channelId string, offset int, limit int) StoreChannel {
|
|
storeChannel := make(StoreChannel)
|
|
|
|
go func() {
|
|
result := StoreResult{}
|
|
|
|
var posts []*model.Post
|
|
_, err := s.GetReplica().Select(&posts, "SELECT * FROM Posts WHERE ChannelId = ? AND DeleteAt = 0 ORDER BY CreateAt DESC LIMIT ?,?", channelId, offset, limit)
|
|
if err != nil {
|
|
result.Err = model.NewAppError("SqlPostStore.GetLinearPosts", "We couldn't get the posts for the channel", "channelId="+channelId+err.Error())
|
|
} else {
|
|
result.Data = posts
|
|
}
|
|
|
|
storeChannel <- result
|
|
close(storeChannel)
|
|
}()
|
|
|
|
return storeChannel
|
|
}
|
|
|
|
func (s SqlPostStore) getParentsPosts(channelId string, offset int, limit int) StoreChannel {
|
|
storeChannel := make(StoreChannel)
|
|
|
|
go func() {
|
|
result := StoreResult{}
|
|
|
|
var posts []*model.Post
|
|
_, err := s.GetReplica().Select(&posts,
|
|
`SELECT
|
|
q2.*
|
|
FROM
|
|
Posts q2
|
|
INNER JOIN
|
|
(SELECT DISTINCT
|
|
q3.RootId
|
|
FROM
|
|
(SELECT
|
|
RootId
|
|
FROM
|
|
Posts
|
|
WHERE
|
|
ChannelId = ?
|
|
AND DeleteAt = 0
|
|
ORDER BY CreateAt DESC
|
|
LIMIT ?, ?) q3) q1 ON q1.RootId = q2.RootId
|
|
WHERE
|
|
ChannelId = ?
|
|
AND DeleteAt = 0
|
|
ORDER BY CreateAt`,
|
|
channelId, offset, limit, channelId)
|
|
if err != nil {
|
|
result.Err = model.NewAppError("SqlPostStore.GetLinearPosts", "We couldn't get the parent post for the channel", "channelId="+channelId+err.Error())
|
|
} else {
|
|
result.Data = posts
|
|
}
|
|
|
|
storeChannel <- result
|
|
close(storeChannel)
|
|
}()
|
|
|
|
return storeChannel
|
|
}
|
|
|
|
func (s SqlPostStore) Search(teamId string, userId string, terms string, isHashtagSearch bool) StoreChannel {
|
|
storeChannel := make(StoreChannel)
|
|
|
|
go func() {
|
|
result := StoreResult{}
|
|
termMap := map[string]bool{}
|
|
|
|
searchType := "Message"
|
|
if isHashtagSearch {
|
|
searchType = "Hashtags"
|
|
for _, term := range strings.Split(terms, " ") {
|
|
termMap[term] = true
|
|
}
|
|
}
|
|
|
|
// @ has a speical meaning in INNODB FULLTEXT indexes and
|
|
// is reserved for calc'ing distances so you
|
|
// cannot escape it so we replace it.
|
|
terms = strings.Replace(terms, "@", " ", -1)
|
|
|
|
searchQuery := fmt.Sprintf(`SELECT
|
|
*
|
|
FROM
|
|
Posts
|
|
WHERE
|
|
DeleteAt = 0
|
|
AND ChannelId IN (SELECT
|
|
Id
|
|
FROM
|
|
Channels,
|
|
ChannelMembers
|
|
WHERE
|
|
Id = ChannelId AND TeamId = ?
|
|
AND UserId = ?
|
|
AND DeleteAt = 0)
|
|
AND MATCH (%s) AGAINST (? IN BOOLEAN MODE)
|
|
ORDER BY CreateAt DESC
|
|
LIMIT 100`, searchType)
|
|
|
|
var posts []*model.Post
|
|
_, err := s.GetReplica().Select(&posts, searchQuery, teamId, userId, terms)
|
|
if err != nil {
|
|
result.Err = model.NewAppError("SqlPostStore.Search", "We encounted an error while searching for posts", "teamId="+teamId+", err="+err.Error())
|
|
} else {
|
|
|
|
list := &model.PostList{Order: make([]string, 0, len(posts))}
|
|
|
|
for _, p := range posts {
|
|
if searchType == "Hashtags" {
|
|
exactMatch := false
|
|
for _, tag := range strings.Split(p.Hashtags, " ") {
|
|
if termMap[tag] {
|
|
exactMatch = true
|
|
}
|
|
}
|
|
if !exactMatch {
|
|
continue
|
|
}
|
|
}
|
|
list.AddPost(p)
|
|
list.AddOrder(p.Id)
|
|
}
|
|
|
|
list.MakeNonNil()
|
|
|
|
result.Data = list
|
|
}
|
|
storeChannel <- result
|
|
close(storeChannel)
|
|
}()
|
|
|
|
return storeChannel
|
|
}
|