class FDB::Transaction

def add_read_conflict_key(key)
  key = FDB.key_to_bytes(key)
  add_read_conflict_range(key, key + "\x00")
end

def add_read_conflict_range(bkey, ekey)
  bkey = FDB.key_to_bytes(bkey)
  ekey = FDB.key_to_bytes(ekey)
  FDBC.check_error FDBC.fdb_transaction_add_conflict_range(@tpointer, bkey, bkey.bytesize, ekey, ekey.bytesize, @@ConflictRangeType["READ"][0])
end

def add_write_conflict_key(key)
  key = FDB.key_to_bytes(key)
  add_write_conflict_range(key, key + "\x00")
end

def add_write_conflict_range(bkey, ekey)
  bkey = FDB.key_to_bytes(bkey)
  ekey = FDB.key_to_bytes(ekey)
  FDBC.check_error FDBC.fdb_transaction_add_conflict_range(@tpointer, bkey, bkey.bytesize, ekey, ekey.bytesize, @@ConflictRangeType["WRITE"][0])
end

def atomic_op(code, key, param)
  key = FDB.key_to_bytes(key)
  param = FDB.value_to_bytes(param)
  FDBC.fdb_transaction_atomic_op(@tpointer, key, key.bytesize, param, param.bytesize, code)
end

def cancel
  FDBC.fdb_transaction_cancel @tpointer
end

def clear(key)
  key = FDB.key_to_bytes(key)
  FDBC.fdb_transaction_clear(@tpointer, key, key.bytesize)
end

def clear_range(bkey, ekey)
  bkey = FDB.key_to_bytes(bkey)
  ekey = FDB.key_to_bytes(ekey)
  FDBC.fdb_transaction_clear_range(@tpointer, bkey || "", bkey && bkey.bytesize || 0, ekey || "", ekey && ekey.bytesize || 0)
end

def clear_range_start_with(prefix)
  prefix = FDB.key_to_bytes(prefix)
  prefix = prefix.dup.force_encoding "BINARY"
  clear_range(prefix, FDB.strinc(prefix))
end

def commit
  FutureNil.new(FDBC.fdb_transaction_commit(@tpointer))
end

def get_approximate_size
  Int64Future.new(FDBC.fdb_transaction_get_approximate_size @tpointer)
end

def get_committed_version
  version = FFI::MemoryPointer.new :int64
  FDBC.check_error FDBC.fdb_transaction_get_committed_version(@tpointer, version)
  version.read_long_long
end

def get_versionstamp
  Key.new(FDBC.fdb_transaction_get_versionstamp(@tpointer))
end

def initialize(tpointer, db)
  super(tpointer, db, 0)
  @snapshot = TransactionRead.new(tpointer, db, 1)
  @options = TransactionOptions.new lambda { |code, param|
    FDBC.check_error FDBC.fdb_transaction_set_option(@tpointer, code, param, param.nil? ? 0 : param.bytesize)
  }
  ObjectSpace.undefine_finalizer self
end

def on_error(e)
  raise e if !e.kind_of? Error
  FutureNil.new(FDBC.fdb_transaction_on_error(@tpointer, e.code))
end

def reset
  FDBC.fdb_transaction_reset @tpointer
end

def set(key, value)
  key = FDB.key_to_bytes(key)
  value = FDB.value_to_bytes(value)
  FDBC.fdb_transaction_set(@tpointer, key, key.bytesize, value, value.bytesize)
end

def set_read_version(version)
  FDBC.fdb_transaction_set_read_version(@tpointer, version)
end

def watch(key)
  key = FDB.key_to_bytes(key)
  FutureNil.new(FDBC.fdb_transaction_watch(@tpointer, key, key.bytesize))
end