added count to repo

2018-04-08 13:35:33 +02:00 · 2018-04-08 13:35:33 +02:00 · cae54c39b0
parent 426a4cbd83
commit cae54c39b0
1 changed files with 128 additions and 0 deletions
--- a/api/repo/data.go
+++ b/api/repo/data.go
@ -110,6 +110,134 @@ func (r *DataRepository) Close() {
 	r.ms.Close()
 }

+func (r *DataRepository) Count(p *DataParams) (int, error) {
+	ms := r.ms.Copy()
+	defer ms.Close()
+	c := ms.DB(r.database).C(r.collection)
+	q := bson.M{}
+	// string
+	if p.Name {
+		switch p.NameQuery.Operation {
+		case "eq":
+			q["name"] = p.NameQuery.Query
+		case "regex":
+			q["name"] = bson.RegEx{Pattern: p.NameQuery.Query, Options: p.NameQuery.Options}
+		}
+	}
+
+	// string
+	if p.Address {
+		switch p.AddressQuery.Operation {
+		case "eq":
+			q["address"] = p.AddressQuery.Query
+		case "regex":
+			q["address"] = bson.RegEx{Pattern: p.AddressQuery.Query, Options: p.AddressQuery.Options}
+		}
+	}
+
+	// string
+	if p.Location {
+		switch p.LocationQuery.Operation {
+		case "eq":
+			q["address"] = p.LocationQuery.Query
+		case "regex":
+			q["address"] = bson.RegEx{Pattern: p.LocationQuery.Query, Options: p.LocationQuery.Options}
+		}
+	}
+
+	// int64
+	if p.Age {
+		switch p.AgeQuery.Operation {
+		case "eq":
+			q["age"] = p.AgeQuery.From
+		case "gt-lt":
+			q["age"] = bson.M{"$gt": p.AgeQuery.From, "$lt": p.AgeQuery.To}
+		case "gte-lte":
+			q["age"] = bson.M{"$gte": p.AgeQuery.From, "$lte": p.AgeQuery.To}
+		case "gte-lt":
+			q["age"] = bson.M{"$gte": p.AgeQuery.From, "$lt": p.AgeQuery.To}
+		case "gt-lte":
+			q["age"] = bson.M{"$gt": p.AgeQuery.From, "$lte": p.AgeQuery.To}
+		case "gt":
+			q["age"] = bson.M{"$gt": p.AgeQuery.From}
+		case "gte":
+			q["age"] = bson.M{"$gte": p.AgeQuery.From}
+		case "lt":
+			q["age"] = bson.M{"$lt": p.AgeQuery.To}
+		case "lte":
+			q["age"] = bson.M{"$lte": p.AgeQuery.To}
+		}
+	}
+
+	// float64
+	if p.Price {
+		switch p.PriceQuery.Operation {
+		case "eq":
+			q["price"] = p.PriceQuery.From
+		case "gt":
+			q["price"] = bson.M{"$gt": p.PriceQuery.From}
+		case "gte":
+			q["price"] = bson.M{"$gte": p.PriceQuery.From}
+		case "lt":
+			q["price"] = bson.M{"$lt": p.PriceQuery.To}
+		case "lte":
+			q["price"] = bson.M{"$lte": p.PriceQuery.To}
+		case "gt-lt":
+			q["price"] = bson.M{"$gt": p.PriceQuery.From, "$lt": p.PriceQuery.To}
+		case "gte-lte":
+			q["price"] = bson.M{"$gte": p.PriceQuery.From, "$lte": p.PriceQuery.To}
+		case "gte-lt":
+			q["price"] = bson.M{"$gte": p.PriceQuery.From, "$lt": p.PriceQuery.To}
+		case "gt-lte":
+			q["price"] = bson.M{"$gt": p.PriceQuery.From, "$lte": p.PriceQuery.To}
+		}
+	}
+
+	// bool
+	if p.Enabled {
+		q["enabled"] = p.EnabledQuery.Query
+	}
+
+	// time
+	if p.When {
+		switch p.WhenQuery.Operation {
+		case "eq":
+			q["when"] = p.WhenQuery.From
+		case "gt":
+			q["when"] = bson.M{"$gt": p.WhenQuery.From}
+		case "gte":
+			q["when"] = bson.M{"$gte": p.WhenQuery.From}
+		case "lt":
+			q["when"] = bson.M{"$lt": p.WhenQuery.To}
+		case "lte":
+			q["when"] = bson.M{"$lte": p.WhenQuery.To}
+		case "gt-lt":
+			q["when"] = bson.M{"$gt": p.WhenQuery.From, "$lt": p.WhenQuery.To}
+		case "gte-lte":
+			q["when"] = bson.M{"$gte": p.WhenQuery.From, "$lte": p.WhenQuery.To}
+		case "gte-lt":
+			q["when"] = bson.M{"$gte": p.WhenQuery.From, "$lt": p.WhenQuery.To}
+		case "gt-lte":
+			q["when"] = bson.M{"$gt": p.WhenQuery.From, "$lte": p.WhenQuery.To}
+		}
+	}
+
+	rq := c.Find(q)
+
+	if p.Sort != "" {
+		rq = rq.Sort(p.Sort)
+	}
+
+	if p.Start > 0 && p.Limit > 0 {
+		rq = rq.Skip(p.Start).Limit(p.Limit)
+	} else if p.Start > 0 && p.Limit < 1 {
+		rq = rq.Skip(p.Start)
+	} else if p.Start < 1 && p.Limit > 0 {
+		rq = rq.Limit(p.Limit)
+	}
+	return rq.Count()
+}
+
 func (r *DataRepository) List(p *DataParams) ([]*model.Data, error) {
 	ms := r.ms.Copy()
 	defer ms.Close()