1 Commits

Author SHA1 Message Date
336dea8147 ensure multiple index via db interface 2022-02-10 16:10:38 +01:00
2 changed files with 15 additions and 3 deletions

View File

@@ -9,10 +9,22 @@ import (
) )
// EnsureIndex ensured mongodb index reflecting model struct index tag // EnsureIndex ensured mongodb index reflecting model struct index tag
func (db *MgoDatabase) EnsureIndex(m ModelInterface) error { func (db *MgoDatabase) EnsureIndex(m ModelInterface, index ...Index) error {
colName := GetCollectionName(m) colName := GetCollectionName(m)
col := db.C(colName) col := db.C(colName)
if len(index) > 0 {
// only ensure given index
for _, i := range index {
err := col.EnsureIndex(i)
if err != nil {
return err
}
}
return nil
}
// ensure index by struct fields
mType := reflect.TypeOf(m) mType := reflect.TypeOf(m)
textFields := []string{} textFields := []string{}

View File

@@ -22,9 +22,9 @@ type Session interface {
type Database interface { type Database interface {
Session() Session Session() Session
C(name string) Collection // C(name string) Collection
Name() string Name() string
EnsureIndex(m ModelInterface) error EnsureIndex(m ModelInterface, index ...Index) error
ValidateObject(m ModelInterface, changes bson.M) error ValidateObject(m ModelInterface, changes bson.M) error
ReadDocument(m ModelInterface, selector bson.M) error ReadDocument(m ModelInterface, selector bson.M) error
CreateDocument(m ModelInterface) error CreateDocument(m ModelInterface) error