forked from golang/hotime
67 lines
1.9 KiB
Go
67 lines
1.9 KiB
Go
// Copyright (C) MongoDB, Inc. 2017-present.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
// not use this file except in compliance with the License. You may obtain
|
|
// a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
package description
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
)
|
|
|
|
// TopologyVersion represents a software version.
|
|
type TopologyVersion struct {
|
|
ProcessID primitive.ObjectID
|
|
Counter int64
|
|
}
|
|
|
|
// NewTopologyVersion creates a TopologyVersion based on doc
|
|
func NewTopologyVersion(doc bson.Raw) (*TopologyVersion, error) {
|
|
elements, err := doc.Elements()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var tv TopologyVersion
|
|
var ok bool
|
|
for _, element := range elements {
|
|
switch element.Key() {
|
|
case "processId":
|
|
tv.ProcessID, ok = element.Value().ObjectIDOK()
|
|
if !ok {
|
|
return nil, fmt.Errorf("expected 'processId' to be a objectID but it's a BSON %s", element.Value().Type)
|
|
}
|
|
case "counter":
|
|
tv.Counter, ok = element.Value().Int64OK()
|
|
if !ok {
|
|
return nil, fmt.Errorf("expected 'counter' to be an int64 but it's a BSON %s", element.Value().Type)
|
|
}
|
|
}
|
|
}
|
|
return &tv, nil
|
|
}
|
|
|
|
// CompareToIncoming compares the receiver, which represents the currently known TopologyVersion for a server, to an
|
|
// incoming TopologyVersion extracted from a server command response.
|
|
//
|
|
// This returns -1 if the receiver version is less than the response, 0 if the versions are equal, and 1 if the
|
|
// receiver version is greater than the response. This comparison is not commutative.
|
|
func (tv *TopologyVersion) CompareToIncoming(responseTV *TopologyVersion) int {
|
|
if tv == nil || responseTV == nil {
|
|
return -1
|
|
}
|
|
if tv.ProcessID != responseTV.ProcessID {
|
|
return -1
|
|
}
|
|
if tv.Counter == responseTV.Counter {
|
|
return 0
|
|
}
|
|
if tv.Counter < responseTV.Counter {
|
|
return -1
|
|
}
|
|
return 1
|
|
}
|