add FindAllDevicesByUserIDListAction

This commit is contained in:
Hoang 2021-12-09 11:35:39 +07:00
parent f5ace0d16d
commit 348a8ddb13
2 changed files with 49 additions and 0 deletions

View File

@ -2,10 +2,12 @@ package devicemngmt
import ( import (
"context" "context"
"errors"
"github.com/Selly-Modules/logger" "github.com/Selly-Modules/logger"
"github.com/Selly-Modules/mongodb" "github.com/Selly-Modules/mongodb"
"go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
) )
// FindAllDevicesByUserID ... // FindAllDevicesByUserID ...
@ -33,3 +35,48 @@ func (s Service) FindAllDevicesByUserID(userID string) []Device {
return result return result
} }
// FindAllDevicesByUserIDList ...
func (s Service) FindAllDevicesByUserIDList(userIDList []string) (result []Device, err error) {
var (
ctx = context.Background()
col = s.getDeviceCollection()
)
result = make([]Device, 0)
// Validate
if len(userIDList) <= 0 {
err = errors.New("user list must be greater than 0")
}
if len(userIDList) > limit200 {
err = errors.New("user list must be less than 200")
}
// Get ids
ids := make([]primitive.ObjectID, 0)
for _, value := range userIDList {
if id, valid := mongodb.NewIDFromString(value); valid {
ids = append(ids, id)
}
}
// Condition
cond := bson.M{
"userId": bson.M{
"$in": ids,
},
}
// Find
cursor, err := col.Find(ctx, cond)
if err != nil {
logger.Error("devicemngmt - findAllDevicesByUserIDList ", logger.LogData{
"err": err.Error(),
})
return
}
defer cursor.Close(ctx)
cursor.All(ctx, &result)
return result, nil
}

View File

@ -9,4 +9,6 @@ const (
langVi = "vi" langVi = "vi"
langEn = "en" langEn = "en"
limit200 = 200
) )