用ICE实现一个简单的聊天室

用ICE实现一个简单的聊天室

    聊天室是一种典型的网络应用，这个程序演示了ICE框架的基本结构。

1）定义SLICE接口。
module Chat {

interface  MessageReceiver {
     void  onMessage( string  sender,  string  msg);
};

dictionary < string , MessageReceiver *>  UserList;
 
interface  Room {
     bool  login( string  user, MessageReceiver *  receiver);
     void  logout( string  user);
     void  sendMessage( string  user,  string  message);
};

};
定义两个接口，Room用于服务器端实现，MessageReceiver用于客户端实现。
使用slice2cpp生成C++文件: chat.h, chat.cpp.

2）服务器端实现。
#include  < iostream >
#include  < Ice / Ice.h >
#include  " chat.h "

using  std:: string ;

class  ChatRoomI :  public  Chat::Room {
public :
     virtual   bool  login( const   string & ,  const  Chat::MessageReceiverPrx & ,  const  Ice::Current & );
     virtual   void  logout( const   string & ,  const  Ice::Current & );
     virtual   void  sendMessage( const   string & ,  const   string & ,  const  Ice::Current & );
private :
     void  broadcast( const   string &  user,  const   string &  message);
     bool  notify( const  Chat::MessageReceiverPrx &  receiver,  const   string &  sender,  const   string &  message);
    
    Chat::UserList    users_;
};

bool  ChatRoomI::login( const   string &  user,  const  Chat::MessageReceiverPrx &  receiver,  const  Ice::Current & ) {
     if  (users_.find(user)  !=  users_.end()) {
        return   false ;
    }
    users_.insert(Chat::UserList::value_type(user, receiver));
    broadcast(user,  " ---login--- " );
     return   true ;
}

void  ChatRoomI::logout( const   string &  user,  const  Ice::Current & ) {
    users_.erase(user);
    broadcast(user,  " ===logout=== " );
}

void  ChatRoomI::sendMessage( const   string &  user,  const   string &  message,  const  Ice::Current & ) {
    broadcast(user, message);
}

void  ChatRoomI::broadcast( const   string &  user,  const   string &  message) {
    Chat::UserList::iterator it  =  users_.begin(), end  =  users_.end();
     while  (it  !=  end) {
         if  (user  !=  it -> first  &&   ! notify(it -> second, it -> first, message))
            users_.erase(it ++ );
         else
             ++ it;
    }
}

bool  ChatRoomI::notify( const  Chat::MessageReceiverPrx &  receiver,  const   string &  sender,  const   string &  message) {
     bool  ret  =   true ;   
     try  {
        receiver -> onMessage(sender, message);
    }  catch ( const  std::exception &  ex) {
        ret  =   false ;
    }
    return  ret;
}

class  Server :  public  Ice::Application {
public :
     virtual   int  run( int  argc,  char *  argv[]) {
        Ice::ObjectAdapterPtr adapter  =  communicator() -> createObjectAdapterWithEndpoints(
                 " Chat.RoomAdapter " ,  " default -p 10000 " );
        Chat::RoomPtr room  =   new  ChatRoomI;
        adapter -> add(room, communicator() -> stringToIdentity( " Chat.Room " ));
        adapter -> activate();
        communicator() -> waitForShutdown();
         return   0 ;
    }
};

int  main( int  argc,  char *  argv[]) {
    Server app;
     return  app.main(argc, argv);
}
3）客户端实现。
#include <iostream>
#include <Ice/Ice.h>
#include "chat.h"

using std::string;

class ChatUserI : public Chat::MessageReceiver {
public:
    virtual void onMessage(const string& user, const string& message, const Ice::Current&) {
        std::cout << user << " :  " << message << std::endl;
    }
};

class Client : public Ice::Application {
public:
    virtual int run(int argc, char* argv[]) {
        Chat::RoomPrx chatRoom = Chat::RoomPrx::checkedCast(
                communicator()->stringToProxy("Chat.Room:default -p 10000"));
        if (!chatRoom)  {
            std::cout << "Proxy not existed!\n";
            return -1;
        }
    
        Ice::ObjectAdapterPtr adapter = communicator()->createObjectAdapterWithEndpoints(
                "Chat.UserAdapter", "default");
        Chat::MessageReceiverPtr user = new ChatUserI;
        adapter->add(user, communicator()->stringToIdentity("Chat.User"));
        adapter->activate();
        
        Chat::MessageReceiverPrx userPrx = Chat::MessageReceiverPrx::uncheckedCast(
                adapter->createProxy(communicator()->stringToIdentity("Chat.User")));
        
        string name;
        std::cout << "Please input user name: ";
        std::cin >> name;
        if (!chatRoom->login(name, userPrx)) {
            std::cout << "login error: User Name already exist!\n";
            return 0;
        }
        std::cout << "login OK!" << std::endl;
        
        string message;
        while (std::getline(std::cin, message) && message != "quit") {
            chatRoom->sendMessage(name, message);
        }

        chatRoom->logout(name);

        return 0;
    }
};

int main(int argc, char* argv[]) {
    Client app;
    return app.main(argc, argv);
}