Erlang 聊天室程序(八) 主题房间---supervisor 的使用

之前实现的功能相当于一个大厅，这里我们为这个聊天室程序添加“主题房间的功能”。客户端登陆后处于大厅中，然后获取目前所有的主题房间，再发送消息进入到某个房间内。同一房间的成员可见，发送的消息也只限本房间可见。

为了管理这些房间，Erlang Opt中的监控树是最合适的。

新建一个模块room_manager.erl 其行为遵守gen_supervisor，子进程为chat_room

init(Para)->
	{ok,
         {
        {one_for_one,3,5},
	 [
	   {the_room,
	    {chat_room,start_link,[]},
	    permanent,
	    10000,
	    worker
	   }	  
	  ]
          }
        }
.

再添加一个方法用于启动这个supervisor

start_link(Para)->
	supervisor:start_link({local,?MODULE},?MODULE, [Para])
.

如果现在就直接启动这个模块是没有问题的，但既然是主题房间，肯定不止一个吧。下面就为这个supervisor添加再启动主题房间的功能：

startNewRoom(Para)->
	supervisor:start_child(the_room, [Para])
.

就这么简单？ 肯定不是。还有以下几点需要考虑：

1.如何标识每个房间的信息，如名称、类型、编号、启动时间、当前用户数等。

2.如何维护每个房间的在线用户信息。

3.如何维护当前可用的主题房间的信息。

4.如何从大厅进入房间。

5.如何从房间退出到大厅。

6.如何将信息路由到自己所在的房间。

下面就一个一个地解决：

房间的标识：

如上代码启动主题房间的时候传入了Para参数，可以通过这个参数来初始化任何你需要的房间信息。PS: 这就是Erlang 的好处之一啊

修改chat_room.erl

start_link(Para)->
	gen_server:start_link(?MODULE, [Para],[]).

定义一个record叫roominfo

-record(roominfo,{id,name,type,unum,creationDate}).

用传入进来的参数初始化房间信息

init([Para])->
 id_generator:start_link(),
 %ets:new(clientinfo,[public,
 %      ordered_set,
 %      named_table,
 %      {keypos,#clientinfo.id}
 %      ]),
 client_manager:init(),
 RoomId=id_generator:getnewid("room"),
 case is_record(Para,roominfo)of
  true->
   {ok,#roominfo{unum=0}=Para};
  Els->  
            %maybe we should throw exception here     
  {ok,#roominfo{id=RoomId,unum=0,name="room"++integer_to_list(RoomId),type="def"}}
 end
.

每个房间的在线成员信息维护：

还记得client_manager这个模块否？这个模块就是用来专门管理房间对应的在线成员的，这里需要对其进行修改。

在前面的代码中client_manager这个模块的调用是与chat_room处于同一进程内的，也就是顺序性调用。

每次调用时所操纵的表是写死的，在这里需要改为动态的，在每次操纵时传入要操纵的表。

所以应该在chat_room中保存自己拥有的表名。

修改roominfo的定义添加tablename，并在初始化chat_room时生成好。

-record(roominfo,{id,name,type,unum,tablename,creationDate}).

init([Para])->
	id_generator:start_link(),
	%ets:new(clientinfo,[public,
	%					 ordered_set,
	%					 named_table,
	%					 {keypos,#clientinfo.id}
	%					 ]),	
	RoomId=id_generator:getnewid("room"),
	ClientTableName=list_to_atom(?PRIFIX++integer_to_list(RoomId)), 
	client_manager:init(ClientTableName),
	case is_record(Para,roominfo)of
		true->
			{ok,#roominfo{unum=0}=Para};
		Els->						
	        {ok,#roominfo{id=RoomId,unum=0,name="room"++integer_to_list(RoomId),type="def"}}
	end
.

修改client_manger.erl为每个方法都添加table参数。

%% Author: Administrator
%% Created: 2012-2-28
%% Description: TODO: response for managing clientinfo
-module(client_manager).
%%
%% Include files
%%
-include("clientinfo.hrl").
-include("message.hrl").
%%
%% Exported Functions
%%
-export([init/1,addClient/2,updateClient/2,getNextClient/2,removeClient/2,getClient/2,getNick/2]).
%%
%% API Functions
%%
%%create clientinfo talbe here
init(Tab)->
	ets:new(Tab,[public,
	         ordered_set,
			 named_table,
			{keypos,#clientinfo.id}
		])
.

%%
%% Local Functions
%%
%add new clientinfo
addClient(Tab,ClientInfo)->
	ets:insert(Tab, ClientInfo)
.
%update clientinfo
updateClient(Tab,Message)->
	#message{content=Info,from=Key}=Message,
	io:format("content of message is:~p~n",[Info]),
	%TODO:we should formart content from json to record #clientinfo
	Rec =util_SetInfoParas:paraElements(Info),
	io:format("parased record is:~p~n",[Rec]),
	if is_record(Rec,clientinfo) ->
		   #clientinfo{nick=Nick,sex=Sex,age=Age,province=Province,
					   city=City,posx=Px,posy=Py}=Rec,
		   io:format("here Key is:~p~n",[Key]),
		   case ets:update_element(Tab, Key, [{#clientinfo.nick,Nick},
												{#clientinfo.sex,Sex},
												{#clientinfo.age,Age},
												{#clientinfo.province,Province},
					   							{#clientinfo.city,City},
												{#clientinfo.posx,Px},
												{#clientinfo.posy,Py}]) of
			    true->
			        {ok,Rec};  	   
			    false->
					{fals,Rec}
          end;
	   true->
		io:format("wrong format of clientinfo"),
		{false,Rec}
	end
.
%remove clientinfo
removeClient(Tab,Key)->	
	[TheKey]=Key,
	case ets:lookup(Tab, TheKey) of
		[Record]->
			io:format("record tobe delete from clientinfo is:~p~n",[Record]);
		[]->
            io:format("no record found tobe delete from clientinfo~n")			
	end,
	io:format("delete  clientinfo from table key is:~p~n",[TheKey]),
	ets:delete(Tab, TheKey)
.
getNick(Tab,Key)->
	case ets:lookup(Tab, Key) of
		[Record]->
			#clientinfo{nick=Nick}=Record,
			case Nick of
				undefined ->
					"client"++integer_to_list(Key);
			    Nick->
                     Nick
            end;
		[]->
			"client"++integer_to_list(Key)
    end
.
getClient(Tab,Key)->
	ets:lookup(Tab, Key)
.
getNextClient(Tab,[Key])->	
	case ets:next(Tab, Key) of
		'$end_of_table'->
		 io:format("getNext1,no key found~n"),
			[];
		Next->
			io:format("getNext1,key found~p~n",[Key]),
		    ets:lookup(Tab, Next)
	end
;
getNextClient(Tab,[])->    
	case ets:first(Tab) of
		'$end_of_table'->
			io:format("getNext,no key found~n"),
			[];
		Key->
			io:format("getNext,key found~p~n",[Key]),
			ets:lookup(Tab, Key)
	end
.	

再修改chat_room.erl中调用部分，为每个调用都添加table参数。

 

handle_call({remove_clientinfo,Ref},From,State)->
	Key=Ref#clientinfo.id,
	#roominfo{tablename=Table}=State,
	client_manager:removeClient(Table,Key),
	{reply,ok,State}
	;
handle_call({sendmsg,Msg},From,State)->
	%Key=ets:first(clientinfo),
	%io:format("feching talbe key is ~p~n",[Key]),
	%sendMsg(Key,Msg),
	#roominfo{tablename=Table}=State,
	sendMsg(Table,Msg,[]),
	{reply,ok,State}
;
handle_call({get_member,Msg},From,State)->
 #message{content=Content}=Msg,
 #roominfo{tablename=Table}=State,
 case binary_to_list(Content) of
  "all"->
   List=getClient(Table,[],[]),  
   #message{from=Id}=Msg,   
   TheFrom=client_manager:getNick(Table,Id),
   Message=Msg#message{type="result",from=TheFrom,subject="clientinfo",content=List},
   {Pid,_}=From,
   Pid!{get_member,Message}; 
  _Els->
   ok
 end,
 {reply,ok,State}
;
handle_call({add_clientinfo,ClientInfo},From,State)-> 
 %store clientinfo to ets
 %ets:insert(clientinfo, NewRec)
 #roominfo{tablename=Table}=State,
 client_manager:addClient(Table,ClientInfo),
 {reply,ok,State}
;
handle_call({set_clientinfo,Message},From,State)->
 %we can send result message to client
 %or send a broadcast message to all client 
 #message{from=Id}=Message,
 #roominfo{tablename=Table}=State,
 [Client]=client_manager:getClient(Table,Id),
 #clientinfo{pid=Pid}=Client,
 TheNick=client_manager:getNick(Table,Id), 
 case client_manager:updateClient(Table,Message)of
  {ok,Rec}->   
   #clientinfo{nick=Nick}=Rec,   
   Pid!{setinfo,Rec},     
   {Content}={TheNick ++ " has change nickname:" ++ Nick},
    NewMessage=#message{id="0",
        from=Id,
        to="",
        content=Content,
        type="msg",
        subject="chat",
        time=Message#message.time},
   io:format("Notice Message is:~p~n",[NewMessage]),
   sendMsg(Table,NewMessage,[]);
  {false,Rec}->
 ok
 end,
 {reply,ok,State}
.
sendMsg(Table,Msg,Key)->
 io:format("broading casting msg~p~n",[Key]), 
 case client_manager:getNextClient(Table,Key)of
  [Record]->   
   Next=Record#clientinfo.id,
   Pid=Record#clientinfo.pid,
   Pid!{dwmsg,Msg},
   sendMsg(Table,Msg,[Next]);
  []->
   ok
 end
.
getClient(Table,Key,UList)->
 case client_manager:getNextClient(Key) of
  [Record]->
   Next=Record#clientinfo.id,
   io:format("record is:~p~n",[Record]),
   Json=util_SetInfoParas:deparaElement(Record),
   io:format("clientSession found:~p~n",[Record]),
   NewUList=[Json|UList],
   getClient(Table,[Next],NewUList);
  []->
   UList
 end
.

如此即可。注：client_manager有一个getNick/1 函数，负责返回对应ID用户的昵称。如何从正确的表格中找出用户的昵称？这个问题将留在用户消息路由部分去处理。

暂时到这。