/*
- * Copyright 2014 Facebook, Inc.
+ * Copyright 2015 Facebook, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
#ifndef FOLLY_VARINT_H_
#define FOLLY_VARINT_H_
+#include <type_traits>
+#include <folly/Conv.h>
#include <folly/Range.h>
namespace folly {
/**
* Decode a value from a given buffer, advances data past the returned value.
*/
-uint64_t decodeVarint(ByteRange& data);
+template <class T>
+uint64_t decodeVarint(Range<T*>& data);
/**
* ZigZag encoding that maps signed integers with a small absolute value
return p - buf;
}
-inline uint64_t decodeVarint(ByteRange& data) {
+template <class T>
+inline uint64_t decodeVarint(Range<T*>& data) {
+ static_assert(
+ std::is_same<typename std::remove_cv<T>::type, char>::value ||
+ std::is_same<typename std::remove_cv<T>::type, unsigned char>::value,
+ "Only character ranges are supported");
+
const int8_t* begin = reinterpret_cast<const int8_t*>(data.begin());
const int8_t* end = reinterpret_cast<const int8_t*>(data.end());
const int8_t* p = begin;
}
if (p == end) {
throw std::invalid_argument("Invalid varint value. Too small: " +
- std::to_string(end - begin) + " bytes");
+ folly::to<std::string>(end - begin) +
+ " bytes");
}
val |= static_cast<uint64_t>(*p++) << shift;
}