mirror of
https://github.com/fish-shell/fish-shell.git
synced 2024-11-29 13:23:53 +08:00
40 lines
1.1 KiB
C++
40 lines
1.1 KiB
C++
// Library for pooling common strings.
|
|
#include "config.h" // IWYU pragma: keep
|
|
|
|
#include "intern.h"
|
|
|
|
#include <stddef.h>
|
|
|
|
#include <algorithm>
|
|
#include <cwchar>
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include "common.h"
|
|
#include "fallback.h" // IWYU pragma: keep
|
|
|
|
bool string_less_than_string(const wchar_t *a, const wchar_t *b) { return std::wcscmp(a, b) < 0; }
|
|
|
|
/// The table of intern'd strings.
|
|
owning_lock<std::vector<const wchar_t *>> string_table;
|
|
|
|
static const wchar_t *intern_with_dup(const wchar_t *in, bool dup) {
|
|
if (!in) return nullptr;
|
|
|
|
auto table = string_table.acquire();
|
|
|
|
const wchar_t *result;
|
|
auto iter = std::lower_bound(table->begin(), table->end(), in, string_less_than_string);
|
|
if (iter != table->end() && std::wcscmp(*iter, in) == 0) {
|
|
result = *iter;
|
|
} else {
|
|
result = dup ? wcsdup(in) : in;
|
|
table->insert(iter, result);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
const wchar_t *intern(const wchar_t *in) { return intern_with_dup(in, true); }
|
|
|
|
const wchar_t *intern_static(const wchar_t *in) { return intern_with_dup(in, false); }
|