rabbitmq技術的一些感悟（二）

上一節文章主要是說了一下rabbitmq的安裝以及搭建好環境的一些命令，以及常用的api調用，其實自從google被封掉之後，我之前收藏的很多技術連接都已經被禁止訪問了，這個是多麼可悲的一件事情啊，說多了都是淚。

     首先，我先寫一段消費者的模塊，建立連接，初始化amq以及銷燬連接：

Comsumer::Comsumer(){

}
void Comsumer::init(){
    conn = amqp_new_connection();
    int sockfd = amqp_open_socket(g_logSrv.m_conf.m_hostname.c_str(), g_logSrv.m_conf.m_port);
    amqp_set_sockfd(conn,sockfd);
    LogServer::die_on_amqp_error(amqp_login(conn, g_logSrv.m_conf.m_vhosts.c_str(), 0, 131072, 0, AMQP_SASL_METHOD_PLAIN,  g_logSrv.m_conf.m_username.c_str(),  g_logSrv.m_conf.m_psw.c_str()),"Logging in");
    amqp_channel_open(conn, (amqp_channel_t)g_logSrv.m_conf.m_channel);
    LogServer::die_on_amqp_error(amqp_get_rpc_reply(conn), "Opening channel");
    for(list<ramqserverInfo>::iterator it = g_logSrv.m_conf.getAmqserverInfo.begin(); it != g_logSrv.m_conf.getAmqserverInfo.end(); it++){
        amqp_exchange_declare(conn, (amqp_channel_t)g_logSrv.m_conf.m_channel, amqp_cstring_bytes(it->exchange.c_str()), amqp_cstring_bytes("direct"), 0, 0, amqp_empty_table);
        LogServer::die_on_amqp_error(amqp_get_rpc_reply(conn), "Declaring exchange");
        amqp_queue_declare(conn, (amqp_channel_t)g_logSrv.m_conf.m_channel, amqp_cstring_bytes(it->queue.c_str()), 0, 1, 0, 1, amqp_empty_table); // durable && auto-delete
        LogServer::die_on_amqp_error(amqp_get_rpc_reply(conn), "Declaring queue");

        amqp_queue_bind(conn, (amqp_channel_t)g_logSrv.m_conf.m_channel, amqp_cstring_bytes(it->queue.c_str()), amqp_cstring_bytes(it->exchange.c_str()), amqp_cstring_bytes(it->route.c_str()), amqp_empty_table);
        LogServer::die_on_amqp_error(amqp_get_rpc_reply(conn), "Binding");

        amqp_basic_consume(conn, (amqp_channel_t)g_logSrv.m_conf.m_channel, amqp_cstring_bytes(it->queue.c_str()), amqp_empty_bytes, 0, 1, 0, amqp_empty_table);
        LogServer::die_on_amqp_error(amqp_get_rpc_reply(conn), "Consuming");
        connList.push_back(*it);
    }
}
Comsumer::~Comsumer(){
    if(conn == NULL){
        return;
    }
    LogServer::die_on_amqp_error(amqp_channel_close(conn, (amqp_channel_t)g_logSrv.m_conf.m_channel, AMQP_REPLY_SUCCESS), "Closing channel");
    LogServer::die_on_amqp_error(amqp_connection_close(conn, AMQP_REPLY_SUCCESS), "Closing connection");
    LogServer::die_on_error(amqp_destroy_connection(conn), "Ending connection");
}

 然後再寫一段消費者的隊列，他也是可持久化的隊列：

Producer::Producer(){
}
void Producer::init(){
    conn.conn = amqp_new_connection();
    int sockfd = amqp_open_socket(g_logSrv.m_conf.m_hostname.c_str(), g_logSrv.m_conf.m_port);
    amqp_set_sockfd(conn.conn,sockfd);
    LogServer::die_on_amqp_error(amqp_login(conn.conn, g_logSrv.m_conf.m_vhosts.c_str(), 0, 131072, 0, AMQP_SASL_METHOD_PLAIN,  g_logSrv.m_conf.m_username.c_str(),  g_logSrv.m_conf.m_psw.c_str()),"Logging in");
    amqp_channel_open(conn.conn, (amqp_channel_t)g_logSrv.m_conf.m_channel);
    LogServer::die_on_amqp_error(amqp_get_rpc_reply(conn.conn), "Opening channel");
    for(list<ramqserverInfo>::iterator it = g_logSrv.m_conf.commonAmqserver.begin(); it != g_logSrv.m_conf.commonAmqserver.end(); it++){
        amqp_exchange_declare(conn.conn, (amqp_channel_t)g_logSrv.m_conf.m_channel, amqp_cstring_bytes(it->exchange.c_str()), amqp_cstring_bytes("direct"), 0, 0, amqp_empty_table);
        LogServer::die_on_amqp_error(amqp_get_rpc_reply(conn.conn), "Declaring exchange");
        amqp_queue_declare(conn.conn, (amqp_channel_t)g_logSrv.m_conf.m_channel, amqp_cstring_bytes(it->queue.c_str()), 0, 1, 0, 1, amqp_empty_table); // durable && auto-delete
        LogServer::die_on_amqp_error(amqp_get_rpc_reply(conn.conn), "Declaring queue");
        amqp_queue_bind(conn.conn, (amqp_channel_t)g_logSrv.m_conf.m_channel, amqp_cstring_bytes(it->queue.c_str()), amqp_cstring_bytes(it->exchange.c_str()), amqp_cstring_bytes(it->route.c_str()), amqp_empty_table);
        LogServer::die_on_amqp_error(amqp_get_rpc_reply(conn.conn), "Binding");
        connList.push_back(*it);
    }
    conn.props._flags = AMQP_BASIC_CONTENT_TYPE_FLAG | AMQP_BASIC_DELIVERY_MODE_FLAG;
    conn.props.content_type = amqp_cstring_bytes("text/plain");
    conn.props.delivery_mode = 2;
}
Producer::~Producer(){
    if(conn.conn == NULL){
        return;
    }
    LogServer::die_on_amqp_error(amqp_channel_close(conn.conn, (amqp_channel_t)g_logSrv.m_conf.m_channel, AMQP_REPLY_SUCCESS), "Closing channel");
    LogServer::die_on_amqp_error(amqp_connection_close(conn.conn, AMQP_REPLY_SUCCESS), "Closing connection");
    LogServer::die_on_error(amqp_destroy_connection(conn.conn), "Ending connection");
}

最後就是如何向rabbitmq中寫消息和取消息了：

void LogServer::WriteToAmqQueue(bool isQuick, string str){
    if(isQuick){
        uint8_t size = m_produce.connQuickList.size();
        uint8_t index = rand()%size;
        LogServer::die_on_error(amqp_basic_publish(m_produce.connQuick.conn, (amqp_channel_t)g_logSrv.m_conf.m_channel, amqp_cstring_bytes(m_produce.connQuickList[index].exchange.c_str()), amqp_cstring_bytes(m_produce.connQuickList[index].route.c_str()), 0, 0, &m_produce.connQuick.props, amqp_cstring_bytes(str.c_str())), "Publishing");
    }else{
        uint8_t size = m_produce.connList.size();
        uint8_t index = rand()%size;
        LogServer::die_on_error(amqp_basic_publish(m_produce.conn.conn, (amqp_channel_t)g_logSrv.m_conf.m_channel, amqp_cstring_bytes(m_produce.connList[index].exchange.c_str()), amqp_cstring_bytes(m_produce.connList[index].route.c_str()), 0, 0, &m_produce.conn.props, amqp_cstring_bytes(str.c_str())), "Publishing");
    }
}
void LogServer::runAmqConsumer(){
    m_produce.init();
    m_consumer.init();
    uint32_t tickTime = time(0);
    while(1){
        amqp_rpc_reply_t ret;
        amqp_envelope_t envelope;
        memset(&envelope,0,sizeof(envelope));
        envelope.message.body.bytes={0};
        amqp_maybe_release_buffers(m_consumer.conn);
        ret = amqp_consume_message(m_consumer.conn, &envelope, NULL, 0);

        if (ret.reply_type == AMQP_RESPONSE_NORMAL ){
            char* str = (char*)envelope.message.body.bytes;
            if(str == NULL){
                return;
            }
            str[envelope.message.body.len-1] = '\0';
            LOG(DEBUG)("recv from amq:len:%d,%s\n",envelope.message.body.len,str);
            vector<string> vectStr;
            Util::strSplit(str, "|", vectStr);
            if(vectStr.size() > 8 && (g_logSrv.m_conf.quickWords.find(vectStr[8]) != g_logSrv.m_conf.quickWords.end())){
                WriteToAmqQueue(true,str);
            }
            else
            {
                WriteToAmqQueue(false,str);
            }
            amqp_destroy_envelope(&envelope);
        //  pushMsgQueue(str);
            str = NULL;
        }
        else{
            amqp_destroy_envelope(&envelope);
            return;
        }

        if(tickTime + FAMETICK < time(0) && m_conf.loadConfig() != 0){
            tickTime = time(0);
            if(LogServer::m_stop == true){
                m_produce.init();
                m_consumer.init();
                stop();
            }
            continue;
        }
    }
}