oid.h

// oid.h
 
/*    Copyright 2009 10gen Inc.
 *
 *    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
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */
 
#pragma once
 
#include <string>
 
#include "mongo/base/data_view.h"
#include "mongo/bson/util/builder.h"
#include "mongo/client/export_macros.h"
#include "mongo/util/time_support.h"
 
namespace mongo {
class SecureRandom;
 
class MONGO_CLIENT_API OID {
public:
    struct Hasher {
        size_t operator()(const OID& oid) const;
    };
 
    OID() : _data() {}
 
    enum { kOIDSize = 12, kTimestampSize = 4, kInstanceUniqueSize = 5, kIncrementSize = 3 };
 
    explicit OID(const std::string& s) {
        init(s);
    }
 
    explicit OID(const unsigned char(&arr)[kOIDSize]) {
        std::memcpy(_data, arr, sizeof(arr));
    }
 
    void clear() {
        std::memset(_data, 0, kOIDSize);
    }
 
    int compare(const OID& other) const {
        return memcmp(_data, other._data, kOIDSize);
    }
 
    std::string toString() const;
    std::string toIncString() const;
 
    static OID MONGO_CLIENT_FUNC gen() {
        OID o((no_initialize_tag()));
        o.init();
        return o;
    }
 
    // Caller must ensure that the buffer is valid for kOIDSize bytes.
    // this is templated because some places use unsigned char vs signed char
    template <typename T>
    static OID MONGO_CLIENT_FUNC from(T* buf) {
        OID o((no_initialize_tag()));
        std::memcpy(o._data, buf, OID::kOIDSize);
        return o;
    }
 
    static OID MONGO_CLIENT_FUNC max() {
        OID o((no_initialize_tag()));
        std::memset(o._data, 0xFF, kOIDSize);
        return o;
    }
 
    void init();
 
    void init(const std::string& s);
 
    void init(Date_t date, bool max = false);
 
    time_t asTimeT() const;
    Date_t asDateT() const {
        return asTimeT() * 1000LL;
    }
 
    // True iff the OID is not empty
    bool isSet() const {
        return compare(OID()) != 0;
    }
 
    void hash_combine(size_t& seed) const;
 
    static void MONGO_CLIENT_FUNC justForked();
 
    static unsigned MONGO_CLIENT_FUNC getMachineId();  // used by the 'features' command
    static void MONGO_CLIENT_FUNC regenMachineId();
 
    // Timestamp is 4 bytes so we just use int32_t
    typedef int32_t Timestamp;
 
    // Wrappers so we can return stuff by value.
    struct InstanceUnique {
        static InstanceUnique MONGO_CLIENT_FUNC generate(SecureRandom& entropy);
        uint8_t bytes[kInstanceUniqueSize];
    };
 
    struct Increment {
    public:
        static Increment MONGO_CLIENT_FUNC next();
        uint8_t bytes[kIncrementSize];
    };
 
    void setTimestamp(Timestamp timestamp);
    void setInstanceUnique(InstanceUnique unique);
    void setIncrement(Increment inc);
 
    Timestamp getTimestamp() const;
    InstanceUnique getInstanceUnique() const;
    Increment getIncrement() const;
 
    ConstDataView view() const {
        return ConstDataView(_data);
    }
 
private:
    // Internal mutable view
    DataView _view() {
        return DataView(_data);
    }
 
    // When we are going to immediately overwrite the bytes, there is no point in zero
    // initializing the data first.
    struct no_initialize_tag {};
    explicit OID(no_initialize_tag) {}
 
    char _data[kOIDSize];
};
 
MONGO_CLIENT_API inline std::ostream& MONGO_CLIENT_FUNC operator<<(std::ostream& s, const OID& o) {
    return (s << o.toString());
}
 
MONGO_CLIENT_API inline StringBuilder& MONGO_CLIENT_FUNC
operator<<(StringBuilder& s, const OID& o) {
    return (s << o.toString());
}
 
enum JsonStringFormat {
    Strict,
    TenGen,
    JS
};
 
MONGO_CLIENT_API inline bool MONGO_CLIENT_FUNC operator==(const OID& lhs, const OID& rhs) {
    return lhs.compare(rhs) == 0;
}
MONGO_CLIENT_API inline bool MONGO_CLIENT_FUNC operator!=(const OID& lhs, const OID& rhs) {
    return lhs.compare(rhs) != 0;
}
MONGO_CLIENT_API inline bool MONGO_CLIENT_FUNC operator<(const OID& lhs, const OID& rhs) {
    return lhs.compare(rhs) < 0;
}
MONGO_CLIENT_API inline bool MONGO_CLIENT_FUNC operator<=(const OID& lhs, const OID& rhs) {
    return lhs.compare(rhs) <= 0;
}
 
}  // namespace mongo