♻️ Use Core.Hashtbl in the lexer

Zeta611 · Jun 1, 2024 · f562090 · f562090
1 parent feb4904
commit f562090
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/lib/lexer.mll b/lib/lexer.mll
@@ -1,14 +1,13 @@
 {
+open Core
 open Lexing
 open Parser
 
 exception SyntaxError of string
 
-let keyword_tbl = Hashtbl.create 31
-
-let () =
-  List.iter
-    (fun (keyword, tok) -> Hashtbl.add keyword_tbl keyword tok)
+let keywords =
+  Hashtbl.of_alist_exn
+    (module String)
     [
       ("if", IF);
       ("then", THEN);
@@ -36,7 +35,7 @@ rule read =
   | newline   { new_line lexbuf; read lexbuf }
   | int as i  { INT (int_of_string i) }
   | real as r { REAL (float_of_string r) }
-  | id as s   { try Hashtbl.find keyword_tbl s with Not_found -> ID s }
+  | id as s   { match Hashtbl.find keywords s with Some s -> s | None -> ID s }
   | '#'       { comment lexbuf }
   | '+'       { PLUS }
   | "+."      { RPLUS }