Merge pull request #41 from banhbio/dev

support multithreading
banhbio · Jan 30, 2023 · 99e04b3 · 99e04b3 · banhbio · Jan 30, 2023
2 parents 429e9a5 + bf08ae3
commit 99e04b3
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "Taxonomy"
 uuid = "10bb16a4-4bfc-432e-9419-4ca69c1a753c"
 authors = ["banhbio <ban@kuicr.kyoto-u.ac.jp>"]
-version = "0.4.2"
+version = "0.4.3"
 
 [deps]
 AbstractTrees = "1520ce14-60c1-5f80-bbc7-55ef81b5835c"

diff --git a/src/database.jl b/src/database.jl
@@ -18,9 +18,11 @@ struct DB
     function DB(nodes_dmp::String, names_dmp::String)
         @assert isfile(nodes_dmp)
         @assert isfile(names_dmp)
+        _nodes = Threads.@spawn importnodes(nodes_dmp)
+        _names = Threads.@spawn importnames(names_dmp)
 
-        parents, ranks = importnodes(nodes_dmp)
-        names = importnames(names_dmp)
+        parents, ranks = fetch(_nodes)
+        names = fetch(_names)
 
         db = new(nodes_dmp, names_dmp, Dict(parents), Dict(ranks), Dict(names))
         current_db!(db)

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -15,7 +15,7 @@ else
     @warn "Start test with existing database"
 end
 
-@testset "dabase.jl" begin
+@testset "database.jl" begin
     @test_throws ErrorException isnothing(current_db())
 
     db = Taxonomy.DB("db/nodes.dmp", "db/names.dmp")