Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/intern/src/symbol/symbols.rs')
| -rw-r--r-- | crates/intern/src/symbol/symbols.rs | 119 |
1 files changed, 101 insertions, 18 deletions
diff --git a/crates/intern/src/symbol/symbols.rs b/crates/intern/src/symbol/symbols.rs index db3c3c5200..ac6daaf006 100644 --- a/crates/intern/src/symbol/symbols.rs +++ b/crates/intern/src/symbol/symbols.rs @@ -1,15 +1,78 @@ //! Module defining all known symbols required by the rest of rust-analyzer. #![allow(non_upper_case_globals)] -use std::hash::{BuildHasher, BuildHasherDefault}; +use std::{ + hash::{BuildHasher, BuildHasherDefault}, + ptr::NonNull, +}; +use arrayvec::ArrayString; use dashmap::{DashMap, SharedValue}; use rustc_hash::FxHasher; use crate::{Symbol, symbol::TaggedArcPtr}; +macro_rules! last { + ( $($elems:literal,)+ ) => { + *[ $($elems,)* ].last().unwrap() + }; +} + +impl Integer { + #[inline] + pub fn as_uint(sym: &Symbol) -> Option<usize> { + if !sym.repr.is_arc() { + let elem_ref = sym.repr.pointer(); + // SAFETY: The types have the same layout. + let elem_ref = unsafe { std::mem::transmute::<NonNull<*const str>, &&str>(elem_ref) }; + Self::LIST.element_offset(elem_ref) + } else { + Self::as_uint_cold(sym) + } + } + + #[cold] + fn as_uint_cold(sym: &Symbol) -> Option<usize> { + sym.as_str().parse().ok() + } +} + macro_rules! define_symbols { - (@WITH_NAME: $($alias:ident = $value:literal,)* @PLAIN: $($name:ident,)*) => { + ( + @LISTS: $($list_type_name:ident = $list_prefix:literal + [ $($list_idx:literal,)+ ],)* + @WITH_NAME: $($alias:ident = $value:literal,)* + @PLAIN: $($name:ident,)* + ) => { + $( + pub enum $list_type_name {} + impl $list_type_name { + // Ensure we covered all numbers. + const LIST: &[&str; last!($($list_idx,)+) + 1] = { + static LIST: [&str; last!($($list_idx,)+) + 1] = [ $( concat!($list_prefix, $list_idx), )* ]; + &LIST + }; + + #[cold] + #[inline(never)] + fn create(idx: usize) -> Symbol { + use std::fmt::Write; + const MAX_LEN: usize = $list_prefix.len() + u64::MAX.ilog10() as usize + 1; + let mut s = ArrayString::<MAX_LEN>::new(); + s.push_str($list_prefix); + _ = write!(s, "{idx}"); + Symbol::intern(&s) + } + + #[inline] + pub fn get(idx: usize) -> Symbol { + match Self::LIST.get(idx) { + Option::Some(s) => Symbol { repr: TaggedArcPtr::non_arc(s) }, + Option::None => Self::create(idx), + } + } + } + )* + // The strings should be in `static`s so that symbol equality holds. $( pub const $name: Symbol = { @@ -32,6 +95,14 @@ macro_rules! define_symbols { let hash_one = |it_: &str| hasher_.hash_one(it_); { $( + for s in $list_type_name::LIST { + let hash_ = hash_one(s); + let shard_idx_ = dashmap_.determine_shard(hash_ as usize); + let symbol = Symbol { repr: TaggedArcPtr::non_arc(s) }; + dashmap_.shards_mut()[shard_idx_].get_mut().insert(hash_, (symbol, SharedValue::new(())), |(x, _)| hash_one(x.as_str())); + } + )* + $( let s = stringify!($name); let hash_ = hash_one(s); let shard_idx_ = dashmap_.determine_shard(hash_ as usize); @@ -49,25 +120,37 @@ macro_rules! define_symbols { }; } define_symbols! { + @LISTS: + Integer = "" + [ + 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, + 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, + 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, + 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, + 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, + 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, + 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, + 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, + 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, + 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, + 251, 252, 253, 254, 255, + ], + RaGeneratedName = "<ra@gennew>" + [ + 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, + 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, + 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, + 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, + 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, + 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, + 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, + 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, + 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, + 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, + 251, 252, 253, 254, 255, + ], + @WITH_NAME: dotdotdot = "...", - INTEGER_0 = "0", - INTEGER_1 = "1", - INTEGER_2 = "2", - INTEGER_3 = "3", - INTEGER_4 = "4", - INTEGER_5 = "5", - INTEGER_6 = "6", - INTEGER_7 = "7", - INTEGER_8 = "8", - INTEGER_9 = "9", - INTEGER_10 = "10", - INTEGER_11 = "11", - INTEGER_12 = "12", - INTEGER_13 = "13", - INTEGER_14 = "14", - INTEGER_15 = "15", __empty = "", unsafe_ = "unsafe", in_ = "in", |