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用户的昵称。如何从正确的表格中找出用户的昵称？这个问题将留在用户消息路由部分去处理。

        暂时到这。