diff options
Diffstat (limited to 'src/shared_string.hh')
| -rw-r--r-- | src/shared_string.hh | 32 |
1 files changed, 30 insertions, 2 deletions
diff --git a/src/shared_string.hh b/src/shared_string.hh index 8af9820c..782f1674 100644 --- a/src/shared_string.hh +++ b/src/shared_string.hh @@ -14,6 +14,9 @@ struct StringStorage : UseMemoryDomain<MemoryDomain::SharedString> int refcount; int length; + StringStorage() = default; + constexpr StringStorage(int refcount, int length) : refcount(refcount), length(length) {} + [[gnu::always_inline]] char* data() { return reinterpret_cast<char*>(this + 1); } [[gnu::always_inline]] @@ -40,8 +43,17 @@ struct StringStorage : UseMemoryDomain<MemoryDomain::SharedString> StringStorage::operator delete(s, sizeof(StringStorage) + s->length + 1); } - friend void inc_ref_count(StringStorage* s, void*) { ++s->refcount; } - friend void dec_ref_count(StringStorage* s, void*) { if (--s->refcount == 0) StringStorage::destroy(s); } + friend void inc_ref_count(StringStorage* s, void*) + { + if (s->refcount != -1) + ++s->refcount; + } + + friend void dec_ref_count(StringStorage* s, void*) + { + if (s->refcount != -1 and --s->refcount == 0) + StringStorage::destroy(s); + } }; inline RefPtr<StringStorage> operator"" _ss(const char* ptr, size_t len) @@ -49,6 +61,22 @@ inline RefPtr<StringStorage> operator"" _ss(const char* ptr, size_t len) return StringStorage::create({ptr, (int)len}); } +template<size_t len> +struct StaticStringStorage : StringStorage +{ + template<size_t... I> constexpr + StaticStringStorage(const char (&literal)[len], IndexSequence<I...>) + : StringStorage{-1, len}, data{literal[I]...} {} + + const char data[len]; +}; + +template<size_t len> +constexpr StaticStringStorage<len> static_storage(const char (&literal)[len]) +{ + return { literal, make_index_sequence<len>() }; +} + class SharedString : public StringView { public: |
