support chain in the interpreter

Author: hansihe

liblumen_alloc/src/erts/process/mailbox.rs

@@ -13,9 +13,40 @@ use crate::erts::term::Term;
 pub struct Mailbox {
     messages: VecDeque<Message>,
     seen: isize,
+
+    cursor: usize,
 }
 
 impl Mailbox {
+    // Start receive implementation for the eir interpreter
+    pub fn recv_start(&self) {
+        debug_assert!(self.cursor == 0);
+    }
+    /// Important to remember that this might return a term in a heap
+    /// fragment, and that it needs to be copied over to the process
+    /// heap before the message is removed from the mailbox.
+    pub fn recv_peek(&self) -> Option<Term> {
+        match self.messages.get(self.cursor) {
+            None => None,
+            Some(Message::Process(message::Process { data })) => Some(*data),
+            Some(Message::HeapFragment(message::HeapFragment { data, .. })) => Some(*data),
+        }
+    }
+    pub fn recv_last_off_heap(&self) -> bool {
+        match &self.messages[self.cursor - 1] {
+            Message::Process(_) => false,
+            Message::HeapFragment(_) => true,
+        }
+    }
+    pub fn recv_increment(&mut self) {
+        self.cursor += 1;
+    }
+    pub fn recv_finish(&mut self, proc: &ProcessControlBlock) {
+        self.remove(self.cursor - 1, proc);
+        self.cursor = 0;
+    }
+    // End receive implementation for the eir interpreter
+
     pub fn iter(&self) -> Iter<Message> {
         self.messages.iter()
     }
@@ -127,6 +158,7 @@ impl Default for Mailbox {
         Mailbox {
             messages: Default::default(),
             seen: -1,
+            cursor: 0,
         }
     }
 }

liblumen_eir_interpreter/Cargo.toml

@@ -16,13 +16,19 @@ cranelift-entity = "0.30.0"
 lazy_static = "1.3.0"
 
 # eirproject/eir crates
-libeir_diagnostics = { git = "https://github.com/eirproject/eir.git" }
-libeir_intern = { git = "https://github.com/eirproject/eir.git" }
-libeir_ir = { git = "https://github.com/eirproject/eir.git" }
-libeir_passes = { git = "https://github.com/eirproject/eir.git" }
-libeir_syntax_erl = { git = "https://github.com/eirproject/eir.git" }
+#libeir_diagnostics = { git = "https://github.com/eirproject/eir.git" }
+#libeir_intern = { git = "https://github.com/eirproject/eir.git" }
+#libeir_ir = { git = "https://github.com/eirproject/eir.git" }
+#libeir_passes = { git = "https://github.com/eirproject/eir.git" }
+#libeir_syntax_erl = { git = "https://github.com/eirproject/eir.git" }
+libeir_diagnostics = { path = "../../core_erlang_3/libeir_diagnostics" }
+libeir_intern = { path = "../../core_erlang_3/libeir_intern" }
+libeir_ir = { path = "../../core_erlang_3/libeir_ir" }
+libeir_passes = { path = "../../core_erlang_3/libeir_passes" }
+libeir_syntax_erl = { path = "../../core_erlang_3/libeir_syntax_erl" }
 
 # workspace crates
 liblumen_alloc = { path = "../liblumen_alloc" }
 lumen_runtime = { path = "../lumen_runtime" }
 
+[patch.'https://github.com/eirproject/eir.git']

liblumen_eir_interpreter/chain/Elixir.Chain.erl

@@ -0,0 +1,59 @@
+-file("lib/chain.ex", 1).
+
+-module('Elixir.Chain').
+
+-compile([no_auto_import]).
+
+-export(['__info__'/1,counter/1,create_processes/1,run/1]).
+
+-spec '__info__'(attributes |
+                 compile |
+                 functions |
+                 macros |
+                 md5 |
+                 module |
+                 deprecated) ->
+                    any().
+
+'__info__'(module) ->
+    'Elixir.Chain';
+'__info__'(functions) ->
+    [{counter,1},{create_processes,1},{run,1}];
+'__info__'(macros) ->
+    [];
+'__info__'(Key = attributes) ->
+    erlang:get_module_info('Elixir.Chain', Key);
+'__info__'(Key = compile) ->
+    erlang:get_module_info('Elixir.Chain', Key);
+'__info__'(Key = md5) ->
+    erlang:get_module_info('Elixir.Chain', Key);
+'__info__'(deprecated) ->
+    [].
+
+counter(_next_pid@1) ->
+    receive
+        _n@1 ->
+            erlang:send(_next_pid@1, _n@1 + 1)
+    end.
+
+create_processes(_n@1) ->
+    _last@1 =
+        'Elixir.Enum':reduce('Elixir.Range':new(1, _n@1),
+                             self(),
+                             fun(_, _send_to@1) ->
+                                    spawn('Elixir.Chain',
+                                          counter,
+                                          [_send_to@1])
+                             end),
+    erlang:send(_last@1, 0),
+    receive
+        _final_answer@1 when is_integer(_final_answer@1) ->
+            <<"Result is ",
+              ('Elixir.Kernel':inspect(_final_answer@1))/binary>>
+    end.
+
+run(_n@1) ->
+    'Elixir.IO':puts('Elixir.Kernel':inspect(timer:tc('Elixir.Chain',
+                                                      create_processes,
+                                                      [_n@1]))).
+