330 lines
9.4 KiB
C++
330 lines
9.4 KiB
C++
#include "AbstractSpectrumDataMessage.h"
|
||
#include <QFile>
|
||
|
||
AbstractSpectrumDataMessage::AbstractSpectrumDataMessage()
|
||
{
|
||
msg_items.append( "MSG_TYPE" ); // [0]
|
||
msg_items.append( "MSG_ID" ); // [1]
|
||
msg_items.append( "REF_ID" ); // [2]
|
||
msg_items.append( "PROD_ID" ); // [3]
|
||
msg_items.append( "DATA_TYPE" ); // [4]
|
||
|
||
data_types.append( "SAMPLEPHD" ); // [0]
|
||
data_types.append( "SPHDF" ); // [1]
|
||
data_types.append( "SPHDP" ); // [2]
|
||
data_types.append( "GASBKPHD" ); // [3]
|
||
data_types.append( "BLANKPHD" ); // [4]
|
||
data_types.append( "DETBKPHD" ); // [5]
|
||
data_types.append( "QCPHD" ); // [6]
|
||
data_types.append( "CALIBPHD" ); // [7]
|
||
data_types.append( "MET" ); // [8]
|
||
data_types.append( "RMSSOH" ); // [9]
|
||
data_types.append( "ALERT_FLOW" ); // [10]
|
||
data_types.append( "ALERT_SYSTEM" ); // [11]
|
||
data_types.append( "ALERT_TEMP" ); // [12]
|
||
data_types.append( "ALERT_UPS" ); // [13]
|
||
|
||
msg.verify_srid = false;
|
||
}
|
||
|
||
AbstractSpectrumDataMessage::~AbstractSpectrumDataMessage()
|
||
{
|
||
|
||
}
|
||
|
||
QString AbstractSpectrumDataMessage::GetSpectrumDataTypeFromFile(QString data_file)
|
||
{
|
||
QString ret_string;
|
||
|
||
QFile file(data_file);
|
||
if ( file.open(QIODevice::ReadOnly | QIODevice::Text) )
|
||
{
|
||
QTextStream content(&file);
|
||
if (!content.atEnd())
|
||
{
|
||
if ( AnalyseMessgeInfo(content, true) )
|
||
{
|
||
ret_string = msg.data_type;
|
||
if ( QLatin1String("SPHDP")==ret_string || QLatin1String("SPHDF")==ret_string )
|
||
{
|
||
ret_string = QLatin1String("SAMPLEPHD");
|
||
msg.data_type = ret_string;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
file.close();
|
||
return ret_string;
|
||
}
|
||
|
||
QString AbstractSpectrumDataMessage::GetSpectrumDataTypeFromMessage(QString data_msg)
|
||
{
|
||
QString ret_string;
|
||
|
||
QTextStream content(&data_msg, QIODevice::ReadOnly);
|
||
if (!content.atEnd())
|
||
{
|
||
if ( AnalyseMessgeInfo(content, true) )
|
||
{
|
||
ret_string = msg.msg_type;
|
||
if ( QLatin1String("SPHDP")==ret_string && QLatin1String("SPHDF")==ret_string )
|
||
{
|
||
ret_string = QLatin1String("SAMPLEPHD");
|
||
}
|
||
}
|
||
}
|
||
return ret_string;
|
||
}
|
||
|
||
bool AbstractSpectrumDataMessage::AnalyseFile(QString file_name)
|
||
{
|
||
bool bRet = true;
|
||
QFile file(file_name);
|
||
if (!file.open(QIODevice::ReadOnly | QIODevice::Text))
|
||
{
|
||
return bRet &= false;
|
||
}
|
||
|
||
QTextStream content(&file);
|
||
if (content.atEnd())
|
||
{
|
||
file.close();
|
||
return bRet &= false;
|
||
}
|
||
bRet &= AnalyseMessgeInfo(content);
|
||
/*
|
||
if (bRet)
|
||
{
|
||
bRet &= AnalyseMessgeBody(content);
|
||
}
|
||
*/
|
||
file.close();
|
||
/*
|
||
if ( bRet && msg.verify_srid )
|
||
{
|
||
ChangeOriginalFile(file_name);
|
||
}
|
||
*/
|
||
return bRet;
|
||
}
|
||
|
||
bool AbstractSpectrumDataMessage::AnalyseMsg(QString &msg_string)
|
||
{
|
||
bool bRet = true;
|
||
QTextStream content(&msg_string, QIODevice::ReadOnly);
|
||
if (content.atEnd())
|
||
{
|
||
return bRet &= false;
|
||
}
|
||
|
||
bRet &= AnalyseMessgeInfo(content);
|
||
/*
|
||
if (bRet)
|
||
{
|
||
bRet &= AnalyseMessgeBody(content);
|
||
}
|
||
*/
|
||
return bRet;
|
||
}
|
||
|
||
const MessageInfo &AbstractSpectrumDataMessage::GetMessageInfo()
|
||
{
|
||
return msg;
|
||
}
|
||
|
||
void AbstractSpectrumDataMessage::ClearData()
|
||
{
|
||
msg.data_type.clear();
|
||
msg.msg_id.clear();
|
||
msg.msg_type.clear();
|
||
msg.verify_srid = false;
|
||
}
|
||
|
||
bool AbstractSpectrumDataMessage::AnalyseMessgeInfo(QTextStream &content, bool only_analyse_msg)
|
||
{
|
||
bool bRet = true;
|
||
QString item_text = content.readLine();
|
||
if ( 0!=item_text.left(5).compare(QLatin1String("BEGIN")) &&
|
||
0!=item_text.mid(6).compare(QLatin1String("IMS2.0")) )
|
||
{
|
||
bRet &= false; //消息头的协议标示不对，消息无效
|
||
return bRet;
|
||
}
|
||
|
||
QString line = content.readLine();
|
||
while (bRet)
|
||
{
|
||
QTextStream line_stream(&line);
|
||
line_stream >> item_text;
|
||
if ( 0==item_text.compare(msg_items.at(0)) && bRet ) //MSG_TYPE
|
||
{
|
||
line_stream >> msg.msg_type;
|
||
if ( msg.msg_type!=(QLatin1String("DATA")) )
|
||
{
|
||
bRet &= false; //消息类型不是指定的“DATA”类型，消息过滤
|
||
break;
|
||
}
|
||
line = content.readLine();
|
||
}
|
||
else if ( 0==item_text.compare(msg_items.at(1)) && bRet ) //MSG_ID
|
||
{
|
||
for (int i=0; !line_stream.atEnd(); ++i)
|
||
{
|
||
if ( 0==i )
|
||
{
|
||
line_stream >> msg.msg_id;
|
||
}
|
||
else if (1==i)
|
||
{
|
||
line_stream >> msg.msg_src_code;
|
||
}
|
||
}
|
||
line = content.readLine();
|
||
}
|
||
else if ( 0==item_text.compare(msg_items.at(2)) && bRet ) //REF_ID
|
||
{
|
||
QString ref_id_after = line_stream.readLine();
|
||
QTextStream ref_id_after_stream(&ref_id_after);
|
||
for (int i=0; !ref_id_after_stream.atEnd(); ++i)
|
||
{
|
||
QString temp;
|
||
ref_id_after_stream >> temp;
|
||
if (0==i)
|
||
{
|
||
msg.ref_id_str = temp;
|
||
}
|
||
else if (1==i)
|
||
{
|
||
msg.ref_src_code = temp;
|
||
}
|
||
else if (2==1)
|
||
{
|
||
msg.seq_num = temp;
|
||
}
|
||
else if (3==1)
|
||
{
|
||
msg.tot_num = temp;
|
||
}
|
||
}
|
||
msg.verify_srid = true;
|
||
line = content.readLine();
|
||
}
|
||
else if ( 0==item_text.compare(msg_items.at(3)) && bRet ) //PROD_ID
|
||
{
|
||
line_stream >> msg.product_id >> msg.delivery_id;
|
||
msg.verify_srid = true;
|
||
line = content.readLine();
|
||
}
|
||
else if ( 0==item_text.compare(msg_items.at(4)) && bRet ) //DATA_TYPE
|
||
{
|
||
line_stream >> msg.data_type;
|
||
if ( data_types.contains(msg.data_type) && bRet )
|
||
{
|
||
if ( !only_analyse_msg )
|
||
{
|
||
bRet &= AnalyseMessgeBody(content);
|
||
}
|
||
return bRet;
|
||
}
|
||
else
|
||
{
|
||
line = content.readLine();
|
||
QTextStream data_line_stream(&line);
|
||
data_line_stream >> item_text;
|
||
while (!msg_items.contains(item_text) && !content.atEnd())
|
||
{
|
||
line = content.readLine();
|
||
QTextStream line_stream(&line);
|
||
line_stream >> item_text;
|
||
}
|
||
msg.verify_srid = true;
|
||
}
|
||
}
|
||
else
|
||
{
|
||
//bRet = false;
|
||
// bRet不置为false，表示针对消息头的有效检查持宽容态度，允许跟文件协议有能接受的出入
|
||
break;
|
||
}
|
||
|
||
if ( content.atEnd() || (content.status()!=QTextStream::Ok) )
|
||
{
|
||
bRet = false;
|
||
ClearData();
|
||
break;
|
||
}
|
||
}
|
||
|
||
return bRet;
|
||
}
|
||
|
||
void AbstractSpectrumDataMessage::SetMsgDataType(QString data_type)
|
||
{
|
||
msg.data_type = data_type;
|
||
}
|
||
|
||
void AbstractSpectrumDataMessage::ChangeOriginalFile(QString file_name)
|
||
{
|
||
QFile file(file_name);
|
||
if (file.open(QIODevice::ReadOnly | QIODevice::Text))
|
||
{
|
||
QString temp_file_name = file_name + QLatin1String(".temp");
|
||
QFile temp_file(temp_file_name);
|
||
if (!temp_file.open(QIODevice::WriteOnly | QIODevice::Text))
|
||
{
|
||
QString error = file.errorString();
|
||
return;
|
||
}
|
||
QTextStream in(&file);
|
||
QTextStream out(&temp_file);
|
||
|
||
bool bInBody = true;
|
||
QString line = in.readLine();
|
||
while ( !in.atEnd() && (in.status()==QTextStream::Ok) )
|
||
{
|
||
QString item_text;
|
||
QTextStream line_stream(&line);
|
||
line_stream >> item_text;
|
||
if ( 0==item_text.compare(msg_items.at(4)) && bInBody ) //DATA_TYPE
|
||
{
|
||
QString data_type;
|
||
line_stream >> data_type;
|
||
if ( data_types.contains(data_type) && bInBody )
|
||
{
|
||
bInBody = false;
|
||
}
|
||
else
|
||
{
|
||
line = in.readLine();
|
||
QTextStream data_line_stream(&line);
|
||
data_line_stream >> item_text;
|
||
while (!msg_items.contains(item_text) && !in.atEnd())
|
||
{
|
||
line = in.readLine();
|
||
QTextStream line_stream(&line);
|
||
line_stream >> item_text;
|
||
}
|
||
}
|
||
}
|
||
else if ( 0==item_text.compare(msg_items.at(2)) && bInBody ) //REF_ID
|
||
{
|
||
line = in.readLine();
|
||
}
|
||
else if ( 0==item_text.compare(msg_items.at(3)) && bInBody ) //PROD_ID
|
||
{
|
||
line = in.readLine();
|
||
}
|
||
else
|
||
{
|
||
out << line << "\n";
|
||
line = in.readLine();
|
||
}
|
||
}
|
||
temp_file.close();
|
||
file.close();
|
||
|
||
QFile::remove(file_name);
|
||
QFile::rename(temp_file_name, file_name);
|
||
}
|
||
}
|