You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.1 KiB
56 lines
1.1 KiB
package posts
|
|
|
|
import (
|
|
"log"
|
|
"sync"
|
|
|
|
"git.aiterp.net/rpdata/api/internal/store"
|
|
"git.aiterp.net/rpdata/api/models"
|
|
"github.com/globalsign/mgo"
|
|
)
|
|
|
|
var collection *mgo.Collection
|
|
var mutex sync.RWMutex
|
|
|
|
func find(query interface{}) (models.Post, error) {
|
|
post := models.Post{}
|
|
err := collection.Find(query).One(&post)
|
|
if err != nil {
|
|
return models.Post{}, err
|
|
}
|
|
|
|
return post, nil
|
|
}
|
|
|
|
func list(query interface{}, limit int, sort ...string) ([]models.Post, error) {
|
|
size := 64
|
|
if limit > 0 {
|
|
size = limit
|
|
}
|
|
posts := make([]models.Post, 0, size)
|
|
|
|
err := collection.Find(query).Limit(limit).Sort(sort...).All(&posts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return posts, nil
|
|
}
|
|
|
|
func init() {
|
|
store.HandleInit(func(db *mgo.Database) {
|
|
collection = db.C("logbot3.posts")
|
|
|
|
collection.EnsureIndexKey("logId")
|
|
collection.EnsureIndexKey("time")
|
|
collection.EnsureIndexKey("kind")
|
|
collection.EnsureIndexKey("position")
|
|
|
|
err := collection.EnsureIndex(mgo.Index{
|
|
Key: []string{"$text:text"},
|
|
})
|
|
if err != nil {
|
|
log.Fatalln("init logbot3.logs:", err)
|
|
}
|
|
})
|
|
}
|