52 lines
1018 B
Go
52 lines
1018 B
Go
|
package mongo
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/dalu/gopiwik/piwik"
|
||
|
|
"gopkg.in/mgo.v2"
|
||
|
|
"gopkg.in/mgo.v2/bson"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewVisitStorage(ms *mgo.Session, db, col string) VisitStorage {
|
||
|
|
return VisitStorage{
|
||
|
|
ms: ms.Clone(),
|
||
|
|
db: db,
|
||
|
|
col: col,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type VisitStorage struct {
|
||
|
|
ms *mgo.Session
|
||
|
|
db, col string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s VisitStorage) Add(visit *piwik.Visit) error {
|
||
|
|
ms := s.ms.Copy()
|
||
|
|
defer ms.Close()
|
||
|
|
c := ms.DB(s.db).C(s.col)
|
||
|
|
visit.ID = bson.NewObjectId().Hex()
|
||
|
|
return c.Insert(visit)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s VisitStorage) Find(query map[string]interface{}) (*piwik.Visit, error) {
|
||
|
|
ms := s.ms.Copy()
|
||
|
|
defer ms.Close()
|
||
|
|
c := ms.DB(s.db).C(s.col)
|
||
|
|
visit := new(piwik.Visit)
|
||
|
|
if e := c.Find(query).One(visit); e != nil {
|
||
|
|
return nil, e
|
||
|
|
}
|
||
|
|
return visit, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s VisitStorage) FindAll(query map[string]interface{}) ([]*piwik.Visit, error) {
|
||
|
|
ms := s.ms.Copy()
|
||
|
|
defer ms.Close()
|
||
|
|
c := ms.DB(s.db).C(s.col)
|
||
|
|
visits := []*piwik.Visit{}
|
||
|
|
if e := c.Find(query).All(visits); e != nil {
|
||
|
|
return nil, e
|
||
|
|
}
|
||
|
|
return visits, nil
|
||
|
|
|
||
|
|
}
|