天天看點
傳回

gen_server tasting 之超簡單名稱服務

          年假不能白休，時間不能浪費，看了 erlang 程式設計的 gen_server 章節，為了更好的了解、掌握于是上手寫一個名稱（鍵值）伺服器。這個 lzy_name_svc 伺服器是基于 otp gen_server 寫成的，在底層鍵值被儲存在了 erlang 的程序字典裡，并且用于存儲字典的程序是可以替換的，可以通過 lzy_name_svc:start/1 啟動服務時指定，預設情況儲存在“目前” erlang 程序中。閑話少叙，代碼貼上。

-module(lzy_name_svc).

-behaviour(gen_server).

-export([init/0, init/1, handle_call/3, handle_cast/2, handle_info/2, terminate/2, code_change/3]).
-export([start/0, start/1, stop/0, save/2, load/1, load_all/0, remove/1, remove_all/0]).

%% Interface functions.

start() ->
	gen_server:start_link({local, ?MODULE}, ?MODULE, [], []).

start(Args) ->
	gen_server:start_link({local, ?MODULE}, ?MODULE, Args, []).

stop() ->
	gen_server:call(?MODULE, stop).


%% @spec save(Key, Value) -> OldValue.
save(Key, Value) ->
	gen_server:call(?MODULE, {save, Key, Value}).

%% @spec load(Key) -> Value.
load(Key) ->
	gen_server:call(?MODULE, {load, Key}).

%% @spec load_all() -> [{Key, Value}].
load_all() ->
	gen_server:call(?MODULE, {load_all}).

%% @spec remove(Key) -> Value.
remove(Key) ->
	gen_server:call(?MODULE, {remove, Key}).

%% @spec remove_all() -> [{Key, Value}].
remove_all() ->
	gen_server:call(?MODULE, {remove_all}).


%%  Callback functions.

init([]) ->
	{ok, local};

init([{isolation, NameServer}]) ->
	{ok, {isolation, NameServer}}.

handle_call({save, Key, Value}, _From, NameServer) ->
	{reply, do_save(Key, Value, NameServer), NameServer};

handle_call({load, Key}, _From, NameServer) ->
	{reply, do_load(Key, NameServer), NameServer};

handle_call({load_all}, _From, NameServer) ->
    {reply, do_load_all(NameServer), NameServer};

handle_call({remove, Key}, _From, NameServer) ->
    {reply, do_remove(Key, NameServer), NameServer};

handle_call({remove_all}, _From, NameServer) ->
    {reply, do_remove_all(NameServer), NameServer};

handle_call({stop}, _From, NameServer) ->
    {stop, normal, stopped, NameServer}.

%% Default implement.

handle_cast(_Msg, State) ->
	{noreply, State}.
	
handle_info(_Info, State) ->
	{noreply, State}.

terminate(_Reason, _State) ->
	ok.

code_change(_OldVsn, State, _Extra) ->
	{ok, State}.

%% Private functions.

do_save(Key, Value, {isolation, NameServer}) ->
	NameServer ! {self(), save, Key, Value},
	receive
		Msg -> Msg
	end;

do_save(Key, Value, _) ->
	erlang:put(Key, Value).

do_load(Key, {isolation, NameServer}) ->
	NameServer ! {self(), load, Key},
	receive
		Msg -> Msg
	end;

do_load(Key, _) ->
	erlang:get(Key).

do_load_all({isolation, NameServer}) ->
	NameServer ! {self(), load_all},
	receive
		Msg -> Msg
	end;

do_load_all(_) ->
	erlang:get().

do_remove(Key, {isolation, NameServer}) ->
	NameServer ! {self(), remove, Key},
	receive
		Msg -> Msg
	end;

do_remove(Key, _) ->
	erlang:erase(Key).

do_remove_all({isolation, NameServer}) ->
	NameServer ! {self(), remove_all},
	receive
		Msg -> Msg
	end;

do_remove_all(_) ->
	erlang:erase().

上面這段代碼就是 lzy_name_svc 名稱服務了，有些地方寫得有點備援，呵呵。

          為了能夠替換字典程序來測試驗證名稱服務功能，還寫了一個超簡單的 foo_svc 服務，用來和 lzy_name_svc 通信完成程序字典存取。

-module(foo_svc).

-export([start/0, load_all/0, server_pid/0]).

start() ->
	register(fs, spawn(fun() -> loop() end)).

load_all() ->
	fs !  {self(), load_all},
	receive
		Msg -> Msg
	end.
	
server_pid() ->
	fs ! { self(), server_pid},
	receive
		Msg -> Msg
	end.
	
loop() ->
	receive
		{From, save, Key, Value} ->
			From ! erlang:put(Key, Value),
			loop();
		{From, load, Key} ->
			From ! erlang:get(Key),
			loop();
		{From, load_all} ->
			From ! erlang:get(),
			loop();
		{From, remove, Key} ->
			From ! erlang:erase(Key),
			loop();
		{From, remove_all} ->
			From ! erlang:erase(),
			loop();
		{From, server_pid} ->
			From ! self(),
			loop()
	end.

          下面的代碼就是建立和調用服務的相關代碼了，一起貼上來。第一段是以預設方式啟動了 lzy_name_svc 服務，并向存取 abc -> 123 名稱。

C:\Program Files\erl5.6.4\usr>..\bin\erl -sname server
Eshell V5.6.4  (abort with ^G)
([email protected])1> c(lzy_name_svc).
{ok,lzy_name_svc}
([email protected])2> c(foo_svc.erl).
{ok,foo_svc}
([email protected])3> lzy_name_svc:start().
{ok,<0.47.0>}
([email protected])4> lzy_name_svc:save(abc, 123).
undefined
([email protected])5> lzy_name_svc:load(abc).
123
([email protected])6> lzy_name_svc:load(efg).
undefined
([email protected])7> lzy_name_svc:load_all().
[{abc,123},
 {'$ancestors',[<0.35.0>]},
 {'$initial_call',{gen,init_it,
                       [gen_server,<0.35.0>,<0.35.0>,
                        {local,lzy_name_svc},
                        lzy_name_svc,[],[]]}}]
([email protected])8> lzy_name_svc:remove(abc).
123
([email protected])9> lzy_name_svc:load(abc).
undefined

下面這段是啟動 foo_svc 服務，用它建立的程序來專門存儲名稱資料，是通過 lzy_name_svc:start/1 傳入的 PID。

C:\Program Files\erl5.6.4\usr>..\bin\erl -sname server
Eshell V5.6.4  (abort with ^G)
([email protected])1> foo_svc:start().
true
([email protected])2> NameSvcPid = foo_svc:server_pid().
<0.37.0>
([email protected])3> lzy_name_svc:start([{isolation, NameSvcPid}]).
{ok,<0.40.0>}
([email protected])4> lzy_name_svc:save(abc, 123).
undefined
([email protected])5> lzy_name_svc:load(abc).
123
([email protected])6> foo_svc:load_all().
[{abc,123}]
([email protected])7> lzy_name_svc:remove_all().
[{abc,123}]
([email protected])8> foo_svc:load_all().
[]

          上邊的兩段都是在同一機器上的同一 erlang 節點上完成服務調用的，下面這段代碼是 lzy_name_svc 服務基于上邊狀态時，在同一機器的另外了個 erlang 節點上通過 rpc 庫完成服務調用的。

C:\Program Files\erl5.6.4\usr>..\bin\erl -sname client1
Eshell V5.6.4  (abort with ^G)
([email protected])1> rpc:call([email protected], lzy_name_svc, save, [abc, 123]).
undefined
([email protected])2> rpc:call([email protected], foo_svc, load_all, []).
[{abc,123}]

呵呵，挺入門的，就當做為學習過程的記錄吧。看好 erlang。

          在學習的過程中，有一個事情比較不解，就是對于 字典程序的 “熱替換” 。 我想本應該是可以通過 gen_server behaviour 用于“熱代碼替換”的 code_change 方法完成的，但試了幾次都達不到目的，服務倒是跑的正常，可是字典程序就是不能熱替換，code_change 正常傳回，可是名稱資料卻還是原有字典程序的。測試驗證代碼如下：

C:\Program Files\erl5.6.4\usr>..\bin\erl -sname server
Eshell V5.6.4  (abort with ^G)
([email protected])1> lzy_name_svc:start().
{ok,<0.37.0>}
([email protected])2> lzy_name_svc:save(abc, 123).
undefined
([email protected])3> lzy_name_svc:load_all().
[{abc,123},
 {'$ancestors',[<0.35.0>]},
 {'$initial_call',{gen,init_it,
                       [gen_server,<0.35.0>,<0.35.0>,
                        {local,lzy_name_svc},
                        lzy_name_svc,[],[]]}}]
([email protected])4> foo_svc:start().
true
([email protected])5> NameSvcPid = foo_svc:server_pid().
<0.41.0>
([email protected])6> foo_svc:load_all().
[]
([email protected])7> lzy_name_svc:code_change(foo, NameSvcPid, foo).
{ok,<0.41.0>}
([email protected])8> lzy_name_svc:load_all().
[{abc,123},
 {'$ancestors',[<0.35.0>]},
 {'$initial_call',{gen,init_it,
                       [gen_server,<0.35.0>,<0.35.0>,
                        {local,lzy_name_svc},
                        lzy_name_svc,[],[]]}}]

還請哪位 erlang guru 指點～

// 2009.02.07 22:52 添加

這裡提供了該名稱服務的新疊代版本。

gen_server tasting 之超簡單名稱服務（續）

添加了如下功能：

  1. 使用 otp 監控樹保證服務可靠性。
  2. 添加日志功能，記錄警告事件。
  3. 将名稱服務打包為 application。
  4. 開放 socket 服務，使用 vsns://verb /param 自定義協定對外提供通路支援。

// 2009.03.07 13:30 添加

作者：lzy.je

出處：http://lzy.iteye.com

本文版權歸作者所有，隻允許以摘要和完整全文兩種形式轉載，不允許對文字進行裁剪。未經作者同意必須保留此段聲明，且在文章頁面明顯位置給出原文連接配接，否則保留追究法律責任的權利。

Programing erlang c c++ c# socket
上一篇: NOI2021遊記
下一篇: DNN的輸入類型為什麼要用one-hot編碼?

繼續閱讀