Merge pull request #9 from asjir/main

Added functionality to query by node label

Author: joshday

Project.toml

@@ -1,7 +1,7 @@
 name = "SQLiteGraph"
 uuid = "de2defd0-d76e-464a-9029-1d71e199ae58"
 authors = ["joshday <[email protected]> and contributors"]
-version = "0.2.2"
+version = "0.3.0"
 
 [deps]
 DBInterface = "a10d1c49-ce27-4219-8d33-6db1a4562965"

README.md

@@ -80,14 +80,16 @@ replace!(db, Node(2, "Movie"; title="Forest Gump", genre="Drama"))
 
 - Use `getindex` to access elements.
 - If `:` is used as an index, an iterator is returned.
+- If `s::String` is used as an index, an iterator of nodes where `s` is *one* of the labels is returned.
 
 ```julia
-db[1]  # Node(2, "Movie"; title="Forest Gump", genre="Drama")
+db[1]  # Node(1, "Person", "Actor"; name="Tom Hanks")
 
 for node in db[:]
     println(node)
 end
 
+only(db["Movie"])  # Node(2, "Movie"; title="Forest Gump", genre="Drama")
 
 # (Pretend the graph is populated with many more items.  The following return iterators.)
 

src/SQLiteGraph.jl

@@ -38,7 +38,7 @@ function Base.show(io::IO, o::Node)
     !isempty(o.props) && print(io, "; "); print_props(io, o.props)
     print(io, ')')
 end
-args(n::Node) = (n.id, isempty(n.labels) ? "" : join(n.labels, ';'), JSON3.write(n.props))
+args(n::Node) = (n.id, isempty(n.labels) ? "" : ";"*join(n.labels, ';')*";", JSON3.write(n.props))
 
 
 
@@ -148,6 +148,7 @@ end
 #-----------------------------------------------------------------------------# getindex (Node)
 Base.getindex(db::DB, i::Integer) = Node(first(query(db, "*", "nodes", "id=$i")))
 Base.getindex(db::DB, ::Colon) = (Node(row) for row in query(db, "*", "nodes", "TRUE"))
+Base.getindex(db::DB, label::String) = (Node(row) for row in SQLiteGraph.query(db, "*", "nodes", "labels LIKE '%;$label;%'"))
 
 #-----------------------------------------------------------------------------# getindex (Edge)
 # all specified

test/runtests.jl

@@ -30,6 +30,9 @@ db = DB()
             @test n.id == i
         end
         @test length(collect(db[:])) == 5
+        @test length(collect(db["lab"])) == 3
+        @test length(collect(db["lab1"])) == 2
+        @test length(collect(db["lab2"])) == 2
     end
 end