adjust self
This commit is contained in:
@@ -175,24 +175,24 @@ func GenericRawUpdateOne(set DBObject, id string, a Accessor) (DBObject, int, er
|
||||
return a.LoadOne(id)
|
||||
}
|
||||
|
||||
func GetMySelf(wfa Accessor) (string, error) {
|
||||
func GetMySelf(wfa Accessor) (ShallowDBObject, error) {
|
||||
id, err := GenerateNodeID()
|
||||
if err != nil {
|
||||
return "", err
|
||||
return nil, err
|
||||
}
|
||||
datas, _, _ := wfa.Search(nil, id, false)
|
||||
if len(datas) > 0 {
|
||||
return datas[0].GetID(), nil
|
||||
return datas[0], nil
|
||||
}
|
||||
return "", errors.New("peer not found")
|
||||
return nil, errors.New("peer not found")
|
||||
}
|
||||
|
||||
func IsMySelf(peerID string, wfa Accessor) (bool, string) {
|
||||
id, err := GetMySelf(wfa)
|
||||
if err != nil {
|
||||
pp, err := GetMySelf(wfa)
|
||||
if err != nil || pp == nil {
|
||||
return false, ""
|
||||
}
|
||||
return peerID == id, id
|
||||
return peerID == pp.GetID(), pp.GetID()
|
||||
}
|
||||
|
||||
func GenerateNodeID() (string, error) {
|
||||
|
||||
Reference in New Issue
Block a user