Merge pull request #164 from Pivot-Studio/feat/make_db_parallel

feat: make db parallel

Author: Chronostasys

Cargo.lock

@@ -1,4 +1,4 @@
-use std::{cell::RefCell, path::PathBuf, rc::Rc};
+use std::path::PathBuf;
 
 use lsp_types::{
     CompletionItem, DocumentSymbol, GotoDefinitionResponse, Hover, InlayHint, Location,
@@ -11,7 +11,7 @@ use super::diag::PLDiag;
 pub struct Diagnostics((String, Vec<PLDiag>));
 
 #[salsa::accumulator]
-pub struct PLReferences(Rc<RefCell<Vec<Location>>>);
+pub struct PLReferences(Vec<Location>);
 
 #[salsa::accumulator]
 pub struct GotoDef(GotoDefinitionResponse);

src/ast/accumulators.rs

@@ -4,7 +4,7 @@
 use std::{
     cell::{Cell, RefCell},
     path::Path,
-    rc::Rc,
+    sync::Arc,
 };
 
 use inkwell::{
@@ -114,19 +114,19 @@ pub fn create_llvm_deps<'ctx>(
 
 #[derive(Clone)]
 pub struct LLVMBuilder<'a, 'ctx> {
-    handle_table: Rc<RefCell<FxHashMap<ValueHandle, AnyValueEnum<'ctx>>>>,
-    handle_reverse_table: Rc<RefCell<FxHashMap<AnyValueEnum<'ctx>, ValueHandle>>>,
-    block_table: Rc<RefCell<FxHashMap<BlockHandle, BasicBlock<'ctx>>>>,
-    block_reverse_table: Rc<RefCell<FxHashMap<BasicBlock<'ctx>, BlockHandle>>>,
+    handle_table: Arc<RefCell<FxHashMap<ValueHandle, AnyValueEnum<'ctx>>>>,
+    handle_reverse_table: Arc<RefCell<FxHashMap<AnyValueEnum<'ctx>, ValueHandle>>>,
+    block_table: Arc<RefCell<FxHashMap<BlockHandle, BasicBlock<'ctx>>>>,
+    block_reverse_table: Arc<RefCell<FxHashMap<BasicBlock<'ctx>, BlockHandle>>>,
     context: &'ctx Context,                // llvm context
     builder: &'a Builder<'ctx>,            // llvm builder
     module: &'a Module<'ctx>,              // llvm module
     dibuilder: &'a DebugInfoBuilder<'ctx>, // debug info builder
     diunit: &'a DICompileUnit<'ctx>,       // debug info unit
     targetmachine: &'a TargetMachine,      // might be used in debug info
     discope: Cell<DIScope<'ctx>>,          // debug info scope
-    ditypes_placeholder: Rc<RefCell<FxHashMap<String, RefCell<Vec<MemberType<'ctx>>>>>>, // hold the generated debug info type place holder
-    ditypes: Rc<RefCell<FxHashMap<String, DIType<'ctx>>>>, // hold the generated debug info type
+    ditypes_placeholder: Arc<RefCell<FxHashMap<String, RefCell<Vec<MemberType<'ctx>>>>>>, // hold the generated debug info type place holder
+    ditypes: Arc<RefCell<FxHashMap<String, DIType<'ctx>>>>, // hold the generated debug info type
 }
 
 pub fn get_target_machine(level: OptimizationLevel) -> TargetMachine {
@@ -167,12 +167,12 @@ impl<'a, 'ctx> LLVMBuilder<'a, 'ctx> {
             diunit,
             targetmachine: tm,
             discope: Cell::new(diunit.get_file().as_debug_info_scope()),
-            ditypes: Rc::new(RefCell::new(FxHashMap::default())),
-            ditypes_placeholder: Rc::new(RefCell::new(FxHashMap::default())),
-            handle_table: Rc::new(RefCell::new(FxHashMap::default())),
-            handle_reverse_table: Rc::new(RefCell::new(FxHashMap::default())),
-            block_table: Rc::new(RefCell::new(FxHashMap::default())),
-            block_reverse_table: Rc::new(RefCell::new(FxHashMap::default())),
+            ditypes: Arc::new(RefCell::new(FxHashMap::default())),
+            ditypes_placeholder: Arc::new(RefCell::new(FxHashMap::default())),
+            handle_table: Arc::new(RefCell::new(FxHashMap::default())),
+            handle_reverse_table: Arc::new(RefCell::new(FxHashMap::default())),
+            block_table: Arc::new(RefCell::new(FxHashMap::default())),
+            block_reverse_table: Arc::new(RefCell::new(FxHashMap::default())),
         }
     }
     fn get_llvm_value_handle(&self, value: &AnyValueEnum<'ctx>) -> ValueHandle {
@@ -580,7 +580,7 @@ impl<'a, 'ctx> LLVMBuilder<'a, 'ctx> {
     fn get_or_add_global_value(
         &self,
         name: &str,
-        pltype: Rc<RefCell<PLType>>,
+        pltype: Arc<RefCell<PLType>>,
         ctx: &mut Ctx<'a>,
     ) -> PointerValue<'ctx> {
         let global = self.get_global_var_handle(name);
@@ -776,7 +776,7 @@ impl<'a, 'ctx> IRBuilder<'a, 'ctx> for LLVMBuilder<'a, 'ctx> {
     fn get_or_add_global(
         &self,
         name: &str,
-        pltype: Rc<RefCell<PLType>>,
+        pltype: Arc<RefCell<PLType>>,
         ctx: &mut Ctx<'a>,
     ) -> ValueHandle {
         self.get_llvm_value_handle(
@@ -796,9 +796,9 @@ impl<'a, 'ctx> IRBuilder<'a, 'ctx> for LLVMBuilder<'a, 'ctx> {
         &self,
         range: Range,
         v: ValueHandle,
-        tp: Rc<RefCell<PLType>>,
+        tp: Arc<RefCell<PLType>>,
         ctx: &mut Ctx<'a>,
-    ) -> Result<(ValueHandle, Rc<RefCell<PLType>>), PLDiag> {
+    ) -> Result<(ValueHandle, Arc<RefCell<PLType>>), PLDiag> {
         let handle = v;
         let v = self.get_llvm_value(handle).unwrap();
         if !v.is_pointer_value() {
@@ -1335,7 +1335,7 @@ impl<'a, 'ctx> IRBuilder<'a, 'ctx> for LLVMBuilder<'a, 'ctx> {
     fn add_global(
         &self,
         name: &str,
-        pltype: Rc<RefCell<PLType>>,
+        pltype: Arc<RefCell<PLType>>,
         ctx: &mut Ctx<'a>,
         line: u32,
         pltp: &PLType,

src/ast/builder/llvmbuilder.rs

@@ -3,7 +3,7 @@
 /// 2. 所有涉及llvm类型的函数（包括参数或返回值）都应该是private的
 pub mod llvmbuilder;
 pub mod no_op_builder;
-use std::{cell::RefCell, path::Path, rc::Rc};
+use std::{cell::RefCell, path::Path, sync::Arc};
 
 use enum_dispatch::enum_dispatch;
 use inkwell::{values::BasicValueEnum, FloatPredicate, IntPredicate};
@@ -35,7 +35,7 @@ pub trait IRBuilder<'a, 'ctx> {
     fn add_global(
         &self,
         name: &str,
-        pltype: Rc<RefCell<PLType>>,
+        pltype: Arc<RefCell<PLType>>,
         ctx: &mut Ctx<'a>,
         line: u32,
         pltp: &PLType,
@@ -116,17 +116,17 @@ pub trait IRBuilder<'a, 'ctx> {
     fn get_or_add_global(
         &self,
         name: &str,
-        pltype: Rc<RefCell<PLType>>,
+        pltype: Arc<RefCell<PLType>>,
         ctx: &mut Ctx<'a>,
     ) -> ValueHandle;
     fn build_load(&self, ptr: ValueHandle, name: &str) -> ValueHandle;
     fn try_load2var(
         &self,
         range: Range,
         v: ValueHandle,
-        tp: Rc<RefCell<PLType>>,
+        tp: Arc<RefCell<PLType>>,
         ctx: &mut Ctx<'a>,
-    ) -> Result<(ValueHandle, Rc<RefCell<PLType>>), PLDiag>;
+    ) -> Result<(ValueHandle, Arc<RefCell<PLType>>), PLDiag>;
     fn get_function(&self, name: &str) -> Option<ValueHandle>;
     fn build_call(&self, f: ValueHandle, args: &[ValueHandle]) -> Option<ValueHandle>;
     fn add_function(

src/ast/builder/mod.rs

@@ -1,3 +1,5 @@
+use std::sync::Arc;
+
 use crate::ast::{ctx::Ctx, pltype::PLType};
 
 use super::{IRBuilder, ValueHandle};
@@ -39,7 +41,7 @@ impl<'a, 'ctx> IRBuilder<'a, 'ctx> for NoOpBuilder {
     fn add_global(
         &self,
         _name: &str,
-        _pltype: std::rc::Rc<std::cell::RefCell<crate::ast::pltype::PLType>>,
+        _pltype: Arc<std::cell::RefCell<crate::ast::pltype::PLType>>,
         _ctx: &mut crate::ast::ctx::Ctx<'a>,
         _line: u32,
         _pltp: &crate::ast::pltype::PLType,
@@ -228,7 +230,7 @@ impl<'a, 'ctx> IRBuilder<'a, 'ctx> for NoOpBuilder {
     fn get_or_add_global(
         &self,
         _name: &str,
-        _pltype: std::rc::Rc<std::cell::RefCell<crate::ast::pltype::PLType>>,
+        _pltype: Arc<std::cell::RefCell<crate::ast::pltype::PLType>>,
         _ctx: &mut crate::ast::ctx::Ctx<'a>,
     ) -> super::ValueHandle {
         0
@@ -242,12 +244,12 @@ impl<'a, 'ctx> IRBuilder<'a, 'ctx> for NoOpBuilder {
         &self,
         _range: crate::ast::range::Range,
         _v: super::ValueHandle,
-        tp: std::rc::Rc<std::cell::RefCell<crate::ast::pltype::PLType>>,
+        tp: Arc<std::cell::RefCell<crate::ast::pltype::PLType>>,
         _ctx: &mut crate::ast::ctx::Ctx<'a>,
     ) -> Result<
         (
             super::ValueHandle,
-            std::rc::Rc<std::cell::RefCell<crate::ast::pltype::PLType>>,
+            Arc<std::cell::RefCell<crate::ast::pltype::PLType>>,
         ),
         crate::ast::diag::PLDiag,
     > {

src/ast/builder/no_op_builder.rs

@@ -113,7 +113,16 @@ pub fn compile_dry(db: &dyn Db, docs: MemDocsInput) -> Option<ModWrapper> {
     if input.is_none() {
         return None;
     }
-    compile_dry_file(db, input.unwrap())
+    let re = compile_dry_file(db, input.unwrap());
+    if let Some(res) = db.get_ref_str() {
+        re.and_then(|plmod| {
+            plmod
+                .plmod(db)
+                .get_refs(&res, db, &mut FxHashSet::default());
+            Some(())
+        });
+    }
+    re
 }
 
 #[salsa::tracked]