83 lines
2.0 KiB
Go
83 lines
2.0 KiB
Go
|
package mongodb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"github.com/thoas/go-funk"
|
||
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
||
|
|
"sort"
|
||
|
|
)
|
||
|
|
|
||
|
|
// GetAppIDFromHex ...
|
||
|
|
func GetAppIDFromHex(s string) primitive.ObjectID {
|
||
|
|
id, _ := primitive.ObjectIDFromHex(s)
|
||
|
|
return id
|
||
|
|
}
|
||
|
|
|
||
|
|
// ConvertObjectIDsToStrings ...
|
||
|
|
func ConvertObjectIDsToStrings(ids []primitive.ObjectID) []string {
|
||
|
|
return funk.Map(ids, func(item primitive.ObjectID) string {
|
||
|
|
return item.Hex()
|
||
|
|
}).([]string)
|
||
|
|
}
|
||
|
|
|
||
|
|
// UniqObjectIds ...
|
||
|
|
func UniqObjectIds(ids []primitive.ObjectID) []primitive.ObjectID {
|
||
|
|
idStr := ConvertObjectIDsToStrings(ids)
|
||
|
|
idStr = funk.UniqString(idStr)
|
||
|
|
return ConvertStringsToObjectIDs(idStr)
|
||
|
|
}
|
||
|
|
|
||
|
|
// ConvertStringsToObjectIDs ...
|
||
|
|
func ConvertStringsToObjectIDs(strValues []string) []primitive.ObjectID {
|
||
|
|
return funk.Map(strValues, func(item string) primitive.ObjectID {
|
||
|
|
return GetAppIDFromHex(item)
|
||
|
|
}).([]primitive.ObjectID)
|
||
|
|
}
|
||
|
|
|
||
|
|
// ConvertStringToObjectID ...
|
||
|
|
func ConvertStringToObjectID(id string) primitive.ObjectID {
|
||
|
|
ID, _ := primitive.ObjectIDFromHex(id)
|
||
|
|
return ID
|
||
|
|
}
|
||
|
|
|
||
|
|
// SortObjectId ...
|
||
|
|
func SortObjectId(ids []primitive.ObjectID) []primitive.ObjectID {
|
||
|
|
sort.Slice(ids, func(i, j int) bool {
|
||
|
|
return ids[i].Timestamp().Before(ids[j].Timestamp())
|
||
|
|
})
|
||
|
|
return ids
|
||
|
|
}
|
||
|
|
|
||
|
|
// ConvertInterfacesToObjectIDs ...
|
||
|
|
func ConvertInterfacesToObjectIDs(values []interface{}) []primitive.ObjectID {
|
||
|
|
s := make([]primitive.ObjectID, len(values))
|
||
|
|
for i, v := range values {
|
||
|
|
s[i] = v.(primitive.ObjectID)
|
||
|
|
}
|
||
|
|
return s
|
||
|
|
}
|
||
|
|
|
||
|
|
// ConvertInterfacesToStrings ...
|
||
|
|
func ConvertInterfacesToStrings(values []interface{}) []string {
|
||
|
|
s := make([]string, len(values))
|
||
|
|
for i, v := range values {
|
||
|
|
s[i] = v.(string)
|
||
|
|
}
|
||
|
|
return s
|
||
|
|
}
|
||
|
|
|
||
|
|
// ConvertStringsToInterfaces ...
|
||
|
|
func ConvertStringsToInterfaces(values []string) []interface{} {
|
||
|
|
s := make([]interface{}, len(values))
|
||
|
|
for _, v := range values {
|
||
|
|
s = append(s, v)
|
||
|
|
}
|
||
|
|
return s
|
||
|
|
}
|
||
|
|
|
||
|
|
// ConvertToJSONString ...
|
||
|
|
func ConvertToJSONString(data interface{}) string {
|
||
|
|
dataByte, _ := json.Marshal(data)
|
||
|
|
return string(dataByte)
|
||
|
|
}
|