Thomas Peetz
75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
package tysc
|
|
|
|
import (
|
|
"gitlab.thpeetz.de/kontor/kontor-go/pkg/dao"
|
|
|
|
mgo "gopkg.in/mgo.v2"
|
|
"gopkg.in/mgo.v2/bson"
|
|
)
|
|
|
|
// SportDAO extends the type BaseDAO.
|
|
type SportDAO struct {
|
|
Db dao.BaseDAO
|
|
}
|
|
|
|
const (
|
|
// SPORTCOLLECTION defines the collection name for storing sports.
|
|
SPORTCOLLECTION = "sport"
|
|
// SPORTMODEL defines the name of the sport data model.
|
|
SPORTMODEL = "kontor.tysc.sport"
|
|
)
|
|
|
|
// FindAll retrieves the list of sports from the database.
|
|
func (m *SportDAO) FindAll() ([]Sport, error) {
|
|
m.Db.Connect()
|
|
var sports []Sport
|
|
err := m.Db.MongoDb.C(SPORTCOLLECTION).Find(bson.M{"model": SPORTMODEL}).All(&sports)
|
|
return sports, err
|
|
}
|
|
|
|
// FindByID returns a sport with given id or returns the error.
|
|
func (m *SportDAO) FindByID(id string) (Sport, error) {
|
|
m.Db.Connect()
|
|
var sport Sport
|
|
err := m.Db.MongoDb.C(SPORTCOLLECTION).FindId(bson.ObjectIdHex(id)).One(&sport)
|
|
return sport, err
|
|
}
|
|
|
|
// FindByName returns a sport with given name or returns the error.
|
|
func (m *SportDAO) FindByName(name string) (Sport, error) {
|
|
m.Db.Connect()
|
|
var sport Sport
|
|
err := m.Db.MongoDb.C(SPORTCOLLECTION).Find(bson.M{"name": name, "model": SPORTMODEL}).One(&sport)
|
|
return sport, err
|
|
}
|
|
|
|
// Insert a sport into database.
|
|
func (m *SportDAO) Insert(sport Sport) error {
|
|
m.Db.Connect()
|
|
sport.Model = SPORTMODEL
|
|
err := m.Db.MongoDb.C(SPORTCOLLECTION).Insert(&sport)
|
|
return err
|
|
}
|
|
|
|
// Upsert a sport into database.
|
|
func (m *SportDAO) Upsert(sport Sport) (*mgo.ChangeInfo, error) {
|
|
m.Db.Connect()
|
|
sport.Model = SPORTMODEL
|
|
info, err := m.Db.MongoDb.C(SPORTCOLLECTION).Upsert(bson.M{"name": sport.Name}, &sport)
|
|
return info, err
|
|
}
|
|
|
|
// Update an existing sport.
|
|
func (m *SportDAO) Update(sport Sport) error {
|
|
m.Db.Connect()
|
|
err := m.Db.MongoDb.C(SPORTCOLLECTION).UpdateId(sport.ID, &sport)
|
|
return err
|
|
}
|
|
|
|
// Delete an existing sport.
|
|
func (m *SportDAO) Delete(sport Sport) error {
|
|
m.Db.Connect()
|
|
err := m.Db.MongoDb.C(SPORTCOLLECTION).Remove(&sport)
|
|
return err
|
|
}
|