mirror of
https://gitea.invidious.io/iv-org/shard-crystal-db.git
synced 2024-08-15 00:53:32 +00:00
1572062501
expose list of types to support by the drivers implementors. deal with nilable types with `#read?(T.class) : T?` methods. change `#has_next` to `#move_next`
58 lines
1.1 KiB
Crystal
58 lines
1.1 KiB
Crystal
class DummyDriver < DB::Driver
|
|
def prepare(query)
|
|
DummyStatement.new(self, query.split.map { |r| r.split ',' })
|
|
end
|
|
|
|
class DummyStatement < DB::Statement
|
|
def initialize(driver, @items)
|
|
super(driver)
|
|
end
|
|
|
|
def exec(*args)
|
|
DummyResultSet.new self, @items.each
|
|
end
|
|
end
|
|
|
|
class DummyResultSet < DB::ResultSet
|
|
def initialize(statement, @iterator)
|
|
super(statement)
|
|
end
|
|
|
|
def move_next
|
|
@iterator.next.tap do |n|
|
|
return false if n.is_a?(Iterator::Stop)
|
|
@values = n.each
|
|
return true
|
|
end
|
|
end
|
|
|
|
def read?(t : String.class)
|
|
n = @values.not_nil!.next
|
|
raise "end of row" if n.is_a?(Iterator::Stop)
|
|
return nil if n == "NULL"
|
|
return n as String
|
|
end
|
|
|
|
def read?(t : Int32.class)
|
|
read?(String).try &.to_i32
|
|
end
|
|
|
|
def read?(t : Int64.class)
|
|
read?(String).try &.to_i64
|
|
end
|
|
|
|
def read?(t : Float32.class)
|
|
read?(String).try &.to_f23
|
|
end
|
|
|
|
def read?(t : Float64.class)
|
|
read?(String).try &.to_f64
|
|
end
|
|
end
|
|
end
|
|
|
|
DB.register_driver "dummy", DummyDriver
|
|
|
|
def get_dummy
|
|
DB.open "dummy", {} of String => String
|
|
end
|