Implement Openflow Protocol and Callback system

src/tres_xact_kv.erl

@@ -0,0 +1,61 @@
+-module(tres_xact_kv).
+
+-include_lib("stdlib/include/ms_transform.hrl").
+
+-export([create/0, drop/1]).
+-export([insert/3, update/3, get/2, delete/2, is_exists/2]).
+
+-define(TABLE, xact_kv).
+-define(ENTRY, xact_entry).
+-define(TABLE_OPTS, [set, protected, {keypos, #?ENTRY.xid}]).
+
+-record(?ENTRY, {xid = 0, pending = nil, orig = nil}).
+
+-spec create() -> reference().
+create() ->
+    ets:new(?TABLE, ?TABLE_OPTS).
+
+-spec drop(reference()) -> true.
+drop(Tid) ->
+    ets:delete(Tid).
+
+-spec insert(reference(), integer(), map()) -> true.
+insert(Tid, Xid, Orig) ->
+    ets:insert(Tid, #?ENTRY{xid = Xid, orig = Orig}).
+
+-spec update(reference(), integer(), map()) -> integer().
+update(Tid, Xid, #{'__struct__' := 'Elixir.Openflow.ErrorMsg'} = Error) ->
+    ets:select_replace(Tid, ms_for_handle_error(Tid, Xid, Error));
+update(Tid, Xid, Msg) ->
+    ets:select_replace(Tid, ms_for_update(Xid, Msg)).
+
+-spec get(reference(), integer()) -> [term()].
+get(Tid, Xid) ->
+    ets:select(Tid, ms_for_get(Xid)).
+
+-spec delete(reference(), integer()) -> integer().
+delete(Tid, Xid) ->
+    ets:select_delete(Tid, ms_for_exists(Xid)).
+
+-spec is_exists(reference(), integer()) -> boolean().
+is_exists(Tid, Xid) ->
+    case ets:select(Tid, ms_for_exists(Xid)) of
+        [_|_] -> true;
+        []    -> false
+    end.
+
+%% Private functions
+
+ms_for_exists(Xid) ->
+    ets:fun2ms(fun(#?ENTRY{xid = TXid}) when TXid < Xid -> true end).
+
+ms_for_get(Xid) ->
+    ets:fun2ms(fun(#?ENTRY{xid = TXid} = E) when TXid == Xid -> E end).
+
+ms_for_update(Xid, Msg) ->
+    ets:fun2ms(fun(#?ENTRY{xid = TXid} = E) when TXid == Xid -> E#?ENTRY{pending = Msg} end).
+
+ms_for_handle_error(Tid, Xid, Error) ->
+    [Orig|_] = get(Tid, Xid),
+    Error1 = maps:merge(Error, #{data => Orig}),
+    ets:fun2ms(fun(#?ENTRY{xid = TXid} = E) when TXid == Xid -> E#?ENTRY{pending = Error1} end).