ytDashBoard/ytDashBoardCore/websocketserver.cpp

#include "websocketserver.h"
#include "QtWebSockets/qwebsocketserver.h"
#include "QtWebSockets/qwebsocket.h"
#include "../ytDashBoardDog/dataprecess.h"
#include "infoclient.h"
#include "databoardmem.h"
#include "databasethread.h"



WebSocketServer::WebSocketServer(QObject *parent) : QObject(parent),
    m_pWebSocketServer(Q_NULLPTR)
{
    datalist.clear();
    count = 0;
    m_pWebSocketServer = new QWebSocketServer(QStringLiteral("WebSocket Server"), QWebSocketServer::NonSecureMode,this);
    timer = new QTimer(this);
    connect(timer, &QTimer::timeout, this, &WebSocketServer::timeout);

    connect(m_pWebSocketServer, &QWebSocketServer::newConnection, this, &WebSocketServer::onNewConnection);
    connect(m_pWebSocketServer, &QWebSocketServer::sslErrors, this, &WebSocketServer::onSslErrors);

}


WebSocketServer::~WebSocketServer()
{
    m_pWebSocketServer->close();
    qDeleteAll(clientList.begin(), clientList.end());
    emit closed();
}

void WebSocketServer::start()
{
    this->appendData("server listening start");
    if(m_pWebSocketServer->listen(QHostAddress::Any, 6001)){
        this->appendData("listening start on 6001");
        printf("listening start on 6001\n");
        timer->start(5000);
    }else{
        printf("EchoServer listening failed\n");
        exit(1);
    }

}

void WebSocketServer::onNewConnection()
{
    this->appendData("onNewConnection");
    QWebSocket *pSocket = m_pWebSocketServer->nextPendingConnection();
    connect(pSocket,&QWebSocket::textMessageReceived,this,&WebSocketServer::processTextMessage);
    connect(pSocket,&QWebSocket::disconnected,this,&WebSocketServer::socketDisconnected);

    clientList.append(new WebSocketClient("",pSocket));
}


void WebSocketServer::appendData(QString emitlog)
{
    emit weblog(emitlog);
}


void WebSocketServer::processTextMessage(QString message)
{

    QString data="";

    printf("%s\n",message.toUtf8().data());
    this->appendData(QString("processTextMessage message: %1").arg(message));

    QWebSocket *pClient = qobject_cast<QWebSocket *>(sender());  //几个用户同时连接服务器，获取某个时刻一个设备请求ID


    QList<WebSocketClient *>::iterator i;
    for(i=clientList.begin();i!=clientList.end();i++){
        printf("processTextMessage clientList size:%d\n",clientList.length());

        WebSocketClient *client = *i;
        if(client->m_client==pClient){ //判断某个时刻一个设备请求ID对应clientList的某个设备创建clientID

            QStringList Dat = message.split("\r\n");
            QByteArray cmd;
            bool found = false;
            for(int i=0;i<Dat.length();i++){
                if(Dat.at(i).indexOf("{")>=0)
                    found = true;
                if(found)
                    cmd.append(Dat.at(i).trimmed().toUtf8());
            }

            QJsonParseError json_error;
            QJsonDocument jsonDoc(QJsonDocument::fromJson(cmd,&json_error));
            if(json_error.error==QJsonParseError::NoError){
                QJsonObject root = jsonDoc.object();
                QString cmd = "";
                if((!root.value("CMD").isUndefined())&&(!root.value("CMD").isNull())){
                    if(root.value("CMD").isString())
                        cmd = root.value("CMD").toString();
                    else
                        cmd = QString("%1").arg(root.value("CMD").toInt());
                    //pClient->sendTextMessage(QString("CMD:%1").arg(cmd));
                }
                client->Cmd = cmd;

                QString stationid="";
                if((!root.value("StationID").isUndefined())&&(!root.value("StationID").isNull())){
                    if(root.value("StationID").isString())
                        stationid = root.value("StationID").toString();
                    else
                        stationid = QString("%1").arg(root.value("StationID").toInt());
                    //pClient->sendTextMessage(QString("StationID:%1").arg(stationid));
                }
                client->StationId=stationid;

                QString stationname="";
                if((!root.value("StationName").isUndefined())&&(!root.value("StationName").isNull())){
                    if(root.value("StationName").isString())
                        stationname = root.value("StationName").toString();
                    else
                        stationname = QString("%1").arg(root.value("StationName").toInt());
                    //pClient->sendTextMessage(QString("StationName:%1").arg(stationname));
                }
                client->StationName=stationname;

                QString deviceid="";
                if((!root.value("DeviceID").isUndefined())&&(!root.value("DeviceID").isNull())){
                    if(root.value("DeviceID").isString())
                        deviceid = root.value("DeviceID").toString();
                    else
                        deviceid = QString("%1").arg(root.value("DeviceID").toInt());
                    //pClient->sendTextMessage(QString("DeviceID:%1").arg(deviceid));
                }
                client->DeviceId=deviceid;

                QString busaddr="";
                if((!root.value("BusAddr").isUndefined())&&(!root.value("BusAddr").isNull())){
                    if(root.value("BusAddr").isString())
                        busaddr = root.value("BusAddr").toString();
                    else
                        busaddr = QString("%1").arg(root.value("BusAddr").toInt());
                    //pClient->sendTextMessage(QString("BusAddr:%1").arg(busaddr));
                }
                client->BusAddr=busaddr;

                QString count="";
                if((!root.value("Count").isUndefined())&&(!root.value("Count").isNull())){
                    if(root.value("Count").isString())
                        count = root.value("Count").toString();
                    else
                        count = QString("%1").arg(root.value("Count").toInt());
                    //pClient->sendTextMessage(QString("Count:%1").arg(count));
                }
                client->Count=count;

                QString start;
                if((!root.value("Start").isUndefined())&&(!root.value("Start").isNull())){
                    if(root.value("Start").isString())
                        start = root.value("Start").toString();
                    else
                        start = QString("%1").arg(root.value("Start").toInt());
                    //pClient->sendTextMessage(QString("Start:%1").arg(start));
                }
                client->Start=start;

                QString limit="";
                if((!root.value("Limit").isUndefined())&&(!root.value("Limit").isNull())){
                    if(root.value("Limit").isString())
                        limit = root.value("Limit").toString();
                    else
                        limit = QString("%1").arg(root.value("Limit").toInt());
                    //pClient->sendTextMessage(QString("Limit:%1").arg(limit));
                }
                client->Limit=limit;

                QString order;
                if((!root.value("Order").isUndefined())&&(!root.value("Order").isNull())){
                    if(root.value("Order").isString())
                        order = root.value("Order").toString();
                    else
                        order = QString("%1").arg(root.value("Order").toInt());
                    //pClient->sendTextMessage(QString("Order:%1").arg(order));
                }
                client->Order=order;

                QString time="";
                if((!root.value("Time").isUndefined())&&(!root.value("Time").isNull())){
                    if(root.value("Time").isString())
                        time = root.value("Time").toString();
                    else
                        time = QString("%1").arg(root.value("Time").toInt());
                    //pClient->sendTextMessage(QString("Time:%1").arg(time));
                }
                client->Time = time;
            }
            printf("client->Cmd = %s\n",client->Cmd.toUtf8().data());


            //sendtext to client  BEGIN
            if(QString::compare(client->Cmd,"getStationInfo")==0){
                QString stationStr;
                for(int i=0;i<16;i++)
                {
                    if(ytStationCount->station[i].Enabled == 0x01){
                        stationStr.append(QString("{\"StationID\":\"%1\",\"StationName\":\"%2\",\"StationStatus\":%3,\"StationStatusNote\":\"%4\",\"DeviceCount\":%5,\"DeviceAlarm\":%6,\"DeviceFault\":%7,\"LastCommTime\":\"%8\"},").arg(ytStationCount->station[i].StationID).arg(ytStationCount->station[i].StationName).arg(ytStationCount->station[i].StationStatus).arg(ytStationCount->station[i].StationStatusNote).arg(ytStationCount->station[i].DeviceCount).arg(ytStationCount->station[i].DeviceAlarm).arg(ytStationCount->station[i].DeviceFault).arg(ytStationCount->station[i].LastCommTime));
                    }
                }
                stationStr=stationStr.left(stationStr.length()-1);
                data=QString("{\"time\":\"%1\",\"CMD\":\"%2\",\"VER\":\"1.00\",\"TotalCount\":\"%3\",\"RESULT\":[%4],\"TimeStamp\":\"%5\",\"REPLY\":1}").arg(QDateTime::currentDateTime().toString("yyyy-MM-dd HH:mm:ss")).arg(client->Cmd).arg(ytStationCount->StationCount).arg(stationStr).arg(QDateTime::currentDateTime().toString("yyyy-MM-dd HH:mm:ss"));
                client->m_client->sendTextMessage(data);
                this->appendData(QString("getStationInfo sendtoclient: %1").arg(data));

            }else if(QString::compare(client->Cmd,"getStationDetail")==0){
                QString stationStr;
                for(int i=0;i<16;i++)
                {
                    if(QString::compare(client->StationId,ytStationCount->station[i].StationID)==0)
                    {
                        QString deviceStr;
                        for(int j=0;j<128;j++)
                        {
                            if(ytStationCount->station[i].StationDevice[j].Enabled == 0x01)
                            {
                                QString pointStr;
                                for(int k=0;k<256;k++){
                                    if((ytStationCount->station[i].StationDevice[j].DevicePoint[k].Enabled==0x01)&&(ytStationCount->station[i].StationDevice[j].DevicePoint[k].IsKeyPoint == 1)){
                                        QString pointvalueStr;

                                        pointvalueStr.append(QString("{\"AlarmStatus\":%1,\"Value\":%2,\"Time\":\"%3\"}").arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].pointvaluelist[0].AlarmStatus).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].pointvaluelist[0].Value).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].pointvaluelist[0].Time));

                                        pointStr.append(QString("{\"BusAddr\":%1,\"PointName\":\"%2\",\"PointType\":%3,\"AlarmType\":%4,\"AlarmStatus\":%5,\"ValueList\" :[%6],\"Time\":\"%7\"}").arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].BusAddr).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].PointName).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].PointType).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].AlarmType).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].AlarmStatus).arg(pointvalueStr).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].Time));
                                        break;
                                    }

                                }

                                deviceStr.append(QString("{\"DeviceID\":\"%1\",\"DeviceName\":\"%2\",\"DeviceTypeID\":\"%3\",\"DeviceType\":\"%4\",\"DeviceStatus\":%5,\"DeviceStatusNote\":\"%6\",\"KeyPoints\":[%7]},").arg(ytStationCount->station[i].StationDevice[j].DeviceID).arg(ytStationCount->station[i].StationDevice[j].DeviceName).arg(ytStationCount->station[i].StationDevice[j].DeviceTypeID).arg(ytStationCount->station[i].StationDevice[j].DeviceType).arg(ytStationCount->station[i].StationDevice[j].DeviceStatus).arg(ytStationCount->station[i].StationDevice[j].DeviceStatusNote).arg(pointStr));

                            }
                        }
                        deviceStr=deviceStr.left(deviceStr.length()-1);
                        stationStr.append(QString("{\"StationID\":\"%1\",\"StationName\":\"%2\",\"TotalCount\":%3,\"StationStatus\":%4,\"StationStatusNote\":\"%5\",\"LIST\":[%6]}").arg(ytStationCount->station[i].StationID).arg(ytStationCount->station[i].StationName).arg(ytStationCount->station[i].DeviceCount).arg(ytStationCount->station[i].StationStatus).arg(ytStationCount->station[i].StationStatusNote).arg(deviceStr));
                        break;
                    }
                }

                data=QString("{\"time\":\"%1\",\"CMD\":\"%2\",\"VER\":\"1.00\",\"RESULT\":%3,\"TimeStamp\":\"%4\",\"REPLY\":1}").arg(QDateTime::currentDateTime().toString("yyyy-MM-dd HH:mm:ss")).arg(client->Cmd).arg(stationStr).arg(QDateTime::currentDateTime().toString("yyyy-MM-dd HH:mm:ss"));
                client->m_client->sendTextMessage(data);
                this->appendData(QString("getStationDetail sendtoclient: %1").arg(data));

            }else if(QString::compare(client->Cmd,"getDeviceDetail")==0){
                QString stationStr;
                for(int i=0;i<16;i++)
                {
                    if(QString::compare(client->StationId,ytStationCount->station[i].StationID)==0)
                    {
                        QString deviceStr;
                        for(int j=0;j<128;j++)
                        {
                            if(QString::compare(client->DeviceId,ytStationCount->station[i].StationDevice[j].DeviceID)==0)
                            {
                                QString pointStr;
                                for(int k=0;k<256;k++)
                                {
                                    if(ytStationCount->station[i].StationDevice[j].DevicePoint[k].Enabled==0x01)
                                    {
                                        QString pointvalueStr;
                                        for(int m=0;m<10;m++)
                                        {
                                            pointvalueStr.append(QString("{\"AlarmStatus\":%1,\"Value\":%2,\"Time\":\"%3\"},").arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].pointvaluelist[m].AlarmStatus).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].pointvaluelist[m].Value).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].pointvaluelist[m].Time));

                                        }
                                        pointvalueStr=pointvalueStr.left(pointvalueStr.length()-1);
                                        pointStr.append(QString("{\"BusAddr\":%1,\"PointName\":\"%2\",\"PointType\":%3,\"AlarmType\":%4,\"AlarmStatus\":%5,\"Value\":%6,\"LIST\" :[%7],\"Time\":\"%8\"},").arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].BusAddr).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].PointName).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].PointType).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].AlarmType).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].AlarmStatus).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].Value).arg(pointvalueStr).arg(ytStationCount->station[i].StationDevice[j].DevicePoint[k].Time));


                                    }

                                }
                                pointStr=pointStr.left(pointStr.length()-1);
                                deviceStr.append(QString("{\"DeviceID\":\"%1\",\"DeviceName\":\"%2\",\"DeviceTypeID\":\"%3\",\"DeviceType\":\"%4\",\"DeviceStatus\":%5,\"DeviceStatusNote\":\"%6\",\"Points\":[%7]}").arg(ytStationCount->station[i].StationDevice[j].DeviceID).arg(ytStationCount->station[i].StationDevice[j].DeviceName).arg(ytStationCount->station[i].StationDevice[j].DeviceTypeID).arg(ytStationCount->station[i].StationDevice[j].DeviceType).arg(ytStationCount->station[i].StationDevice[j].DeviceStatus).arg(ytStationCount->station[i].StationDevice[j].DeviceStatusNote).arg(pointStr));
                                break;
                            }
                        }

                        stationStr.append(QString("{\"StationID\":\"%1\",\"StationName\":\"%2\",\"Device\":[%3]}").arg(ytStationCount->station[i].StationID).arg(ytStationCount->station[i].StationName).arg(deviceStr));
                        break;
                    }
                }

                data=QString("{\"time\":\"%1\",\"CMD\":\"%2\",\"VER\":\"1.00\",\"RESULT\":%3,\"TimeStamp\":\"%4\",\"REPLY\":1}").arg(QDateTime::currentDateTime().toString("yyyy-MM-dd HH:mm:ss")).arg(client->Cmd).arg(stationStr).arg(QDateTime::currentDateTime().toString("yyyy-MM-dd HH:mm:ss"));
                client->m_client->sendTextMessage(data);
                this->appendData(QString("getDeviceDetail sendtoclient: %1").arg(data));

            }else if(QString::compare(client->Cmd,"getPatrolRecord")==0){
//                QString patrolrecord;
//                db.open();
//                QSqlQuery qry;
//                qry=db.exec(QString("select v_datatime from yt_t_rec where uq_station_id = '%1' order by v_datatime %2 limit %3,%4").arg(client->StationId).arg(client->Order).arg(client->Start).arg(client->Limit));

//                QString timestr;
//                while(qry.next())
//                {
//                    timestr.append(QString("{\"Time\":\"%1\"},").arg(qry.value(0).toDateTime().toString("yyyy-MM-dd HH:mm:ss").toUtf8().data()));

//                }
//                timestr=timestr.left(timestr.length()-1);

//                QSqlQuery qry1;
//                QString stationname,totalcount;
//                qry1=db.exec(QString("select v_station_name,(select count(*) from yt_t_rec) from yt_t_station where uq_station_id = '%1'").arg(client->StationId));
//                while(qry1.next())
//                {
//                    stationname = qry1.value(0).toString().toUtf8().data();
//                    totalcount = qry1.value(1).toString().toUtf8().data();
//                }

//                patrolrecord=QString("{\"StationID\":\"%1\",\"StationName\":\"%2\",\"LIST\":[%3]}").arg(client->StationId).arg(stationname).arg(timestr);

//                db.close();
                QString patrolrecord;
                QString timestr;
                int clientstart = client->Start.toInt();
                int clientlimit = client->Limit.toInt() ;
                printf("client->Order = %s\n",client->Order.toUtf8().data());
                if(QString::compare(client->Order,"DESC")==0)
                {
                    //printf("testest11111");
                    counttemp=0;
                    for(int i=PatrolRecordList.size()-1;i>=0;i--){
                        //printf("testest22222");
                        if(QString::compare(PatrolRecordList.at(i).StationId,client->StationId)==0)
                        {
                            //printf("counttemp = %d\n",counttemp);
                            counttemp++;
                            if(counttemp >= clientstart && counttemp < clientstart+clientlimit)
                            {
                                printf("PatrolRecordList.at(i).DataTime = %s\n",PatrolRecordList.at(i).DataTime.toUtf8().data());
                                timestr.append(QString("{\"Time\":\"%1\"},").arg(PatrolRecordList.at(i).DataTime));
                            }

                        }
                    }
                }

                if(QString::compare(client->Order,"ASC")==0)
                {
                    counttemp=0;
                    for(int i=0;i<PatrolRecordList.size();i++){
                        if(QString::compare(PatrolRecordList.at(i).StationId,client->StationId)==0)
                        {
                            counttemp++;
                            if(counttemp >= clientstart && counttemp < clientstart+clientlimit)
                            {
                                timestr.append(QString("{\"Time\":\"%1\"},").arg(PatrolRecordList.at(i).DataTime));
                            }

                        }
                    }
                }
                timestr=timestr.left(timestr.length()-1);

                QString stationname;
                QString totalcount = QString::number(PatrolRecordList.size());
                for(int i=0;i<16;i++)
                {
                    if(QString::compare(client->StationId,ytStationCount->station[i].StationID)==0)
                    {
                        stationname = ytStationCount->station[i].StationName;
                    }
                }

                patrolrecord=QString("{\"StationID\":\"%1\",\"StationName\":\"%2\",\"LIST\":[%3]}").arg(client->StationId).arg(stationname).arg(timestr);


                data=QString("{\"time\":\"%1\",\"CMD\":\"%2\",\"VER\":\"1.00\",\"TotalCount\":%3,\"Count\":%4,\"RESULT\":%5,\"TimeStamp\":\"%6\",\"REPLY\":1}").arg(QDateTime::currentDateTime().toString("yyyy-MM-dd HH:mm:ss")).arg(client->Cmd).arg(totalcount).arg(client->Limit).arg(patrolrecord).arg(QDateTime::currentDateTime().toString("yyyy-MM-dd HH:mm:ss"));
                client->m_client->sendTextMessage(data);
                this->appendData(QString("getPatrolRecord sendtoclient: %1").arg(data));


            }else if(QString::compare(client->Cmd,"getPatrolDetail")==0){
//                db.open();
//                QSqlQuery qry;
//                qry=db.exec(QString("select t_values from yt_t_rec where uq_station_id = '%1' and v_datatime = '%2'").arg(client->StationId).arg(client->Time));
//                while(qry.next())
//                {
//                    data = qry.value(0).toString().toUtf8().data();
//                }
//                db.close();

                for(int i=0;i<PatrolRecordList.size();i++){

                    if(QString::compare(client->StationId,PatrolRecordList.at(i).StationId)==0 && QString::compare(client->Time,PatrolRecordList.at(i).DataTime)==0){
                        data = PatrolRecordList.at(i).Values;

                    }
                }

                client->m_client->sendTextMessage(data);
                this->appendData(QString("getPatrolDetail sendtoclient: %1").arg(data));

            }
            //sendtext to client  END

        }

    }
}

void WebSocketServer::patrolrecord(QList<QStringList> PatrolList)
{
    PatrolRecordList.clear();
    for(int i=0;i<PatrolList.size();i++){
        ReceivePatrolRecord patrolrecord;
        patrolrecord.DataTime = PatrolList.at(i).at(0);
        patrolrecord.StationId = PatrolList.at(i).at(1);
        patrolrecord.Values = PatrolList.at(i).at(2);

        PatrolRecordList.append(patrolrecord);

    }

}


void WebSocketServer::socketDisconnected()
{
    this->appendData("socketDisconnected");
    QWebSocket *pClient = qobject_cast<QWebSocket *>(sender());
    if(pClient){
        QList<WebSocketClient *>::iterator i;
        for(i=clientList.begin();i!=clientList.end();i++){
            WebSocketClient *client = *i;
            if(client->m_client==pClient){
                clientList.removeAll(client);
                pClient->deleteLater();
                return;
            }
        }
    }
}

void WebSocketServer::onSslErrors(const QList<QSslError> &)
{
    printf("Ssl errors occurred\n");
    this->appendData("Ssl errors occurred");
}

void WebSocketServer::timeout()
{
    count++;
    if(count>5){
        QList<WebSocketClient *>::iterator i;
        for(i=clientList.begin();i!=clientList.end();i++){
            WebSocketClient *client = *i;
            if(client->m_client){
                client->m_client->sendTextMessage(QString("{\"Cmd\":\"HeartBeat\",\"time\":\"%1\"}").arg(QDateTime::currentDateTime().toString("yyyy-MM-dd HH:mm:ss")));
                this->appendData(QString("{\"Cmd\":\"HeartBeat\",\"time\":\"%1\"}").arg(QDateTime::currentDateTime().toString("yyyy-MM-dd HH:mm:ss")));
            }
        }
        count=0;
    }
}