Merge pull request #4 from Selly-Modules/feature/delete-getUserIDByTokenMethod

add delete and getUserIDByToken method
This commit is contained in:
Nam Huynh 2021-11-07 07:32:18 +07:00 committed by GitHub
commit 1e95e10fc4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 63 additions and 0 deletions

31
action_delete.go Normal file
View File

@ -0,0 +1,31 @@
package devicemngmt
import (
"context"
"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,
}
)
// Delete
if _, err := col.DeleteOne(ctx, cond); err != nil {
logger.Error("devicemngt - deleteByDeviceID", logger.LogData{
"deviceID": deviceID,
"err": err.Error(),
})
return fmt.Errorf("error when delete device: %s", err.Error())
}
return nil
}

32
action_get_user_id.go Normal file
View File

@ -0,0 +1,32 @@
package devicemngmt
import (
"context"
"github.com/Selly-Modules/logger"
"go.mongodb.org/mongo-driver/bson"
)
// GetUserIDByAuthToken ...
func (s Service) GetUserIDByAuthToken(authToken string) (userID string) {
var (
ctx = context.Background()
col = s.getDeviceCollection()
device = Device{}
cond = bson.M{
"authToken": authToken,
}
)
// Find
if err := col.FindOne(ctx, cond).Decode(&device); err != nil {
logger.Error("devicemngt - getUserIDByAuthToken", logger.LogData{
"authToken": authToken,
"err": err.Error(),
})
return
}
userID = device.UserID
return
}