From 2e7c8c2f521c9e50bb3aea4df16771c22fe70e58 Mon Sep 17 00:00:00 2001 From: Niall Sheridan Date: Sat, 10 Sep 2016 20:14:20 +0100 Subject: Allow filtering results --- server/store/mongo.go | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'server/store/mongo.go') diff --git a/server/store/mongo.go b/server/store/mongo.go index 79df69d..8a3ccda 100644 --- a/server/store/mongo.go +++ b/server/store/mongo.go @@ -67,12 +67,17 @@ func (m *mongoDB) SetRecord(record *CertRecord) error { return m.collection.Insert(record) } -func (m *mongoDB) List() ([]*CertRecord, error) { +func (m *mongoDB) List(includeExpired bool) ([]*CertRecord, error) { if err := m.session.Ping(); err != nil { return nil, err } var result []*CertRecord - err := m.collection.Find(bson.M{"expires": bson.M{"$gte": time.Now().UTC()}}).All(&result) + var err error + if includeExpired { + err = m.collection.Find(nil).All(&result) + } else { + err = m.collection.Find(bson.M{"expires": bson.M{"$gte": time.Now().UTC()}}).All(&result) + } return result, err } -- cgit v1.2.3