slice.h 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. // Copyright (c) 2011 The LevelDB Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file. See the AUTHORS file for names of contributors.
  4. //
  5. // Slice is a simple structure containing a pointer into some external
  6. // storage and a size. The user of a Slice must ensure that the slice
  7. // is not used after the corresponding external storage has been
  8. // deallocated.
  9. //
  10. // Multiple threads can invoke const methods on a Slice without
  11. // external synchronization, but if any of the threads may call a
  12. // non-const method, all threads accessing the same Slice must use
  13. // external synchronization.
  14. #ifndef STORAGE_LEVELDB_INCLUDE_SLICE_H_
  15. #define STORAGE_LEVELDB_INCLUDE_SLICE_H_
  16. #include <assert.h>
  17. #include <stddef.h>
  18. #include <string.h>
  19. #include <string>
  20. #include "leveldb/export.h"
  21. namespace leveldb {
  22. class LEVELDB_EXPORT Slice {
  23. public:
  24. // Create an empty slice.
  25. Slice() : data_(""), size_(0) { }
  26. // Create a slice that refers to d[0,n-1].
  27. Slice(const char* d, size_t n) : data_(d), size_(n) { }
  28. // Create a slice that refers to the contents of "s"
  29. Slice(const std::string& s) : data_(s.data()), size_(s.size()) { }
  30. // Create a slice that refers to s[0,strlen(s)-1]
  31. Slice(const char* s) : data_(s), size_(strlen(s)) { }
  32. // Intentionally copyable.
  33. Slice(const Slice&) = default;
  34. Slice& operator=(const Slice&) = default;
  35. // Return a pointer to the beginning of the referenced data
  36. const char* data() const { return data_; }
  37. // Return the length (in bytes) of the referenced data
  38. size_t size() const { return size_; }
  39. // Return true iff the length of the referenced data is zero
  40. bool empty() const { return size_ == 0; }
  41. // Return the ith byte in the referenced data.
  42. // REQUIRES: n < size()
  43. char operator[](size_t n) const {
  44. assert(n < size());
  45. return data_[n];
  46. }
  47. // Change this slice to refer to an empty array
  48. void clear() { data_ = ""; size_ = 0; }
  49. // Drop the first "n" bytes from this slice.
  50. void remove_prefix(size_t n) {
  51. assert(n <= size());
  52. data_ += n;
  53. size_ -= n;
  54. }
  55. // Return a string that contains the copy of the referenced data.
  56. std::string ToString() const { return std::string(data_, size_); }
  57. // Three-way comparison. Returns value:
  58. // < 0 iff "*this" < "b",
  59. // == 0 iff "*this" == "b",
  60. // > 0 iff "*this" > "b"
  61. int compare(const Slice& b) const;
  62. // Return true iff "x" is a prefix of "*this"
  63. bool starts_with(const Slice& x) const {
  64. return ((size_ >= x.size_) &&
  65. (memcmp(data_, x.data_, x.size_) == 0));
  66. }
  67. private:
  68. const char* data_;
  69. size_t size_;
  70. };
  71. inline bool operator==(const Slice& x, const Slice& y) {
  72. return ((x.size() == y.size()) &&
  73. (memcmp(x.data(), y.data(), x.size()) == 0));
  74. }
  75. inline bool operator!=(const Slice& x, const Slice& y) {
  76. return !(x == y);
  77. }
  78. inline int Slice::compare(const Slice& b) const {
  79. const size_t min_len = (size_ < b.size_) ? size_ : b.size_;
  80. int r = memcmp(data_, b.data_, min_len);
  81. if (r == 0) {
  82. if (size_ < b.size_) r = -1;
  83. else if (size_ > b.size_) r = +1;
  84. }
  85. return r;
  86. }
  87. } // namespace leveldb
  88. #endif // STORAGE_LEVELDB_INCLUDE_SLICE_H_