60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
package devicemngmt
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/Selly-Modules/logger"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
)
|
|
|
|
// DeleteByDeviceID ...
|
|
func (s Service) DeleteByDeviceID(deviceID string) error {
|
|
var (
|
|
ctx = context.Background()
|
|
col = s.getDeviceCollection()
|
|
cond = bson.M{
|
|
"deviceId": deviceID,
|
|
}
|
|
)
|
|
|
|
total, _ := col.CountDocuments(ctx, cond)
|
|
if total == 0 {
|
|
return errors.New("deviceId not found")
|
|
}
|
|
|
|
// Delete
|
|
if _, err := col.DeleteOne(ctx, cond); err != nil {
|
|
logger.Error("devicemngmt - deleteByDeviceID", logger.LogData{
|
|
"deviceId": deviceID,
|
|
"err": err.Error(),
|
|
})
|
|
return fmt.Errorf("error when delete device: %s", err.Error())
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// DeleteByUserID ...
|
|
func (s Service) DeleteByUserID(userID string) error {
|
|
var (
|
|
ctx = context.Background()
|
|
col = s.getDeviceCollection()
|
|
cond = bson.M{
|
|
"userId": userID,
|
|
}
|
|
)
|
|
|
|
// Delete
|
|
if _, err := col.DeleteMany(ctx, cond); err != nil {
|
|
logger.Error("devicemngmt - deleteByUserID", logger.LogData{
|
|
"userId": userID,
|
|
"err": err.Error(),
|
|
})
|
|
return fmt.Errorf("error when delete device by userId: %s", err.Error())
|
|
}
|
|
|
|
return nil
|
|
}
|