📄 ecserverdlg.cpp
字号:
// ECSERVERDlg.cpp : 实现文件
//
#include "stdafx.h"
#include "ECSERVER.h"
#include "ECSERVERDlg.h"
#include ".\ecserverdlg.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
// 用于应用程序“关于”菜单项的 CAboutDlg 对话框
////////////////////////////////////////////////////////////
//全局或者静态变量初始化
int CECSERVERDlg::m_conn_num = 0;
CECSERVERDlg* pDlg;//窗口指针,便于线程执行体获得
bool ThreadRun[MAX_CONNECT_NUM];
class CAboutDlg : public CDialog
{
public:
CAboutDlg();
// 对话框数据
enum { IDD = IDD_ABOUTBOX };
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持
// 实现
protected:
DECLARE_MESSAGE_MAP()
};
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()
// CECSERVERDlg 对话框
CECSERVERDlg::CECSERVERDlg(CWnd* pParent /*=NULL*/)
: CDialog(CECSERVERDlg::IDD, pParent)
, m_pSet(NULL)
{
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
void CECSERVERDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
DDX_Control(pDX, IDC_EDIT1, m_message_list);
}
BEGIN_MESSAGE_MAP(CECSERVERDlg, CDialog)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
//}}AFX_MSG_MAP
ON_WM_CLOSE()
END_MESSAGE_MAP()
// CECSERVERDlg 消息处理程序
BOOL CECSERVERDlg::OnInitDialog()
{
CDialog::OnInitDialog();
// 将\“关于...\”菜单项添加到系统菜单中。
// IDM_ABOUTBOX 必须在系统命令范围内。
ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
ASSERT(IDM_ABOUTBOX < 0xF000);
CMenu* pSysMenu = GetSystemMenu(FALSE);
if (pSysMenu != NULL)
{
CString strAboutMenu;
strAboutMenu.LoadString(IDS_ABOUTBOX);
if (!strAboutMenu.IsEmpty())
{
pSysMenu->AppendMenu(MF_SEPARATOR);
pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
}
}
// 设置此对话框的图标。当应用程序主窗口不是对话框时,框架将自动
// 执行此操作
SetIcon(m_hIcon, TRUE); // 设置大图标
SetIcon(m_hIcon, FALSE); // 设置小图标
// TODO: 在此添加额外的初始化代码
pDlg = this;//维护窗口指针
ServerInit();//服务器初始化
for(int i=0;i<MAX_CONNECT_NUM;i++)
{
ThreadRun[i] = true;
}
try
{
CString sDSN="ODBC;DRIVER=Microsoft Access Driver (*.mdb);DSN='';DBQ=system.mdb";
m_database = new CDatabase;
if(!m_database->Open(NULL,FALSE,FALSE,sDSN,FALSE))
{
CDialog::EndDialog(0);
return FALSE;
}
//打开数据库的BOOK表设置记录集
CString strSQL="select * from BOOK";
m_pSet = new CRecordset(m_database);
m_pSet->Open(CRecordset::dynaset,strSQL);
if(!m_pSet->Requery()) AfxMessageBox("can not requery");
if(m_pSet->IsBOF() && m_pSet->IsEOF()) return -1;
}
catch(CDBException* pEx)
{
pEx->ReportError();
}
return TRUE; // 除非设置了控件的焦点,否则返回 TRUE
}
void CECSERVERDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
{
CAboutDlg dlgAbout;
dlgAbout.DoModal();
}
else
{
CDialog::OnSysCommand(nID, lParam);
}
}
// 如果向对话框添加最小化按钮,则需要下面的代码
// 来绘制该图标。对于使用文档/视图模型的 MFC 应用程序,
// 这将由框架自动完成。
void CECSERVERDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // 用于绘制的设备上下文
SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
// 使图标在工作矩形中居中
int cxIcon = GetSystemMetrics(SM_CXICON);
int cyIcon = GetSystemMetrics(SM_CYICON);
CRect rect;
GetClientRect(&rect);
int x = (rect.Width() - cxIcon + 1) / 2;
int y = (rect.Height() - cyIcon + 1) / 2;
// 绘制图标
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialog::OnPaint();
}
}
//当用户拖动最小化窗口时系统调用此函数取得光标显示。
HCURSOR CECSERVERDlg::OnQueryDragIcon()
{
return static_cast<HCURSOR>(m_hIcon);
}
int CECSERVERDlg::ServerInit(void)//服务器初始化
{
m_psocket = new TCPSocket();
m_psocket->Create(PORT_LISTEN);//在端口PORT_LISTEN上创建套接字
m_psocket->Listen(MAX_CONNECT_NUM);//设置最大连接数为CONNECT_NUM
//创建接受客户连接的工作线程
AfxBeginThread(ThreadAccept,this->GetSafeHwnd(),THREAD_PRIORITY_NORMAL);
WriteStr("服务器初始化成功!");
return 0;
}
UINT CECSERVERDlg::ThreadAccept(LPVOID lpram)//接收用户连接线程
{
while(m_conn_num<MAX_CONNECT_NUM)
{
pDlg->m_psocket->Accept(&(pDlg->clientsocket[m_conn_num]));//接受新的连接,都由clientsocket[i]来处理
int socketid = m_conn_num;
AfxBeginThread(ThreadProc,(LPVOID)&socketid,THREAD_PRIORITY_NORMAL);//新创建线程,专门接受发送过来的数据
m_conn_num++;
}
AfxEndThread(0);
return 0;
}
UINT CECSERVERDlg::ThreadProc(LPVOID lpram)
{
char* buf = new char[DATAPACK_SIZE];
DataPack* m_pdatapack = new DataPack();
int* sockid =(int*)lpram ;
while(ThreadRun[*sockid])//运行参数
{
memset(buf,0,DATAPACK_SIZE);
memset(m_pdatapack,0,DATAPACK_SIZE);
pDlg->clientsocket[*sockid].Recv(buf,DATAPACK_SIZE);
m_pdatapack = (DataPack*)buf;
switch(m_pdatapack->flag)
{
case LOGIN://用户登陆处理
{
//查找用户
if((pDlg->GetUser(m_pdatapack->DataInfo.user))==-1)
break;
else
{
CString str;
str.Format("用户:%d登陆成功!",m_pdatapack->DataInfo.user.id);
pDlg->WriteStr(str);
//通知用户登陆成功,并把个人信息发送给用户
m_pdatapack->flag = LOGINSUCCESS;
pDlg->clientsocket[*sockid].Send((const char*)m_pdatapack,DATAPACK_SIZE);
//把书目信息发送给用户
pDlg->SendBookInfo(*sockid);
//把该用户的购书信息发送给用户
pDlg->SendBuyBookInfo(*sockid,m_pdatapack->DataInfo.user.id);
}
break;
}
case BUYBOOK://用户购书
{
CString str;
str.Format("用户:%d购书成功",m_pdatapack->DataInfo.buyitem.user.id);
pDlg->WriteStr(str);
//用户购书处理
pDlg->BuyBook(m_pdatapack->DataInfo.buyitem);
//通知用户购书成功
m_pdatapack->flag = BUYBOOK_SUCCESS;
pDlg->clientsocket[*sockid].Send((const char*)m_pdatapack,DATAPACK_SIZE);
break;
}
case DROPBOOK://用户退书
{
CString str;
str.Format("用户:%d退购成功",m_pdatapack->DataInfo.buyitem.user.id);
pDlg->WriteStr(str);
//用户退购处理
if(pDlg->DropBook(m_pdatapack->DataInfo.buyitem)==0)
{
m_pdatapack->flag = DROPBOOK_SUCCESS;
pDlg->clientsocket[*sockid].Send((const char*)m_pdatapack,DATAPACK_SIZE);
}
break;
}
case LOGOUT:
{//用户退出
CString str;
str.Format("用户:%d退出",m_pdatapack->DataInfo.buyitem.user.id);
pDlg->WriteStr(str);
pDlg->clientsocket[*sockid].Close;
ThreadRun[*sockid] = false;
break;
}
}
}
AfxEndThread(0);//结束线程
return 0;
}
void CECSERVERDlg::OnClose()
{
// TODO: 在此添加消息处理程序代码和/或调用默认值
CDialog::OnClose();
}
//本函数用于从数据库中根据用户ID:userin.id查找用户user,找到的话把其余信息放在user中,否则直接注册
int CECSERVERDlg::GetUser(User& user)//
{
CString m_errmsg;
CDBVariant m_value;
//安全性处理
if(user.id==0||strcmp(user.name,"")==0||strcmp(user.password,"")==0)
{
return -1;
}
TRY
{
CString strSQL;
strSQL.Format("select * from User where USER_ID=%d AND USER_PASSWORD='%s'",user.id,user.password);
CRecordset* m_pSet_tmp = new CRecordset(m_database);
m_pSet_tmp->Open(CRecordset::dynaset,strSQL);
if(m_pSet_tmp->IsBOF())//没有这个id,插入新纪录注册这个用户
{
CString strSQL ;
strSQL.Format("insert into User(USER_ID,USER_NAME,USER_PASSWORD) values(%d,'%s',%s)",user.id,user.name,user.password);
m_database->ExecuteSQL(strSQL);
return 0;
}
//得到对应id的用户名
m_pSet_tmp->GetFieldValue("USER_NAME",m_value);
strcpy(user.name,m_value.m_pstring->GetBuffer(0));
m_pSet_tmp->Close();
}
CATCH_ALL(e)
{
e->GetErrorMessage(m_errmsg.GetBuffer(128),128);
AfxMessageBox(m_errmsg);
return -1;
}
END_CATCH_ALL
return 0;
}
int CECSERVERDlg::SendBookInfo(int socketid)//向用户发送书目信息
{
CString temp;
DataPack datapack;
memset(&datapack,0,DATAPACK_SIZE);
datapack.flag = BOOK;
//依次把书目信息发送给用户
m_pSet->MoveFirst();
while( !m_pSet->IsEOF() )
{
m_pSet->GetFieldValue("BOOK_ID",temp);
datapack.DataInfo.book.book_id = atoi(temp.GetBuffer(0));
m_pSet->GetFieldValue("BOOK_NAME",temp);
strcpy(datapack.DataInfo.book.book_name,temp.GetBuffer(0));
m_pSet->GetFieldValue("BOOK_AUTHOR",temp);
strcpy(datapack.DataInfo.book.book_author,temp.GetBuffer(0));
m_pSet->GetFieldValue("PUBLISHER",temp);
strcpy(datapack.DataInfo.book.publisher,temp.GetBuffer(0));
m_pSet->GetFieldValue("PUBLISH_TIME",temp);
strcpy(datapack.DataInfo.book.publish_time,temp.GetBuffer(0));
m_pSet->GetFieldValue("PRICE",temp);
strcpy(datapack.DataInfo.book.price,temp.GetBuffer(0));
m_pSet->GetFieldValue("LEFT_NUM",temp);
datapack.DataInfo.book.left_num = atoi(temp.GetBuffer(0));
m_pSet->GetFieldValue("PRIVIEW",temp);
strcpy(datapack.DataInfo.book.priview,temp.GetBuffer(0));
//向用户发送书目信息
clientsocket[socketid].Send((const char*)&datapack,DATAPACK_SIZE);
m_pSet->MoveNext();
}
return 0;
}
int CECSERVERDlg::BuyBook(BuyItem& buyitem)//购书处理
{
CString m_errmsg;
TRY
{//向表BUY_ITEM插入购书记录
CString strSQL ;
strSQL.Format("insert into BUY_ITEM(BOOK_ID,BOOK_NAME,USER_ID,BUY_NUM) values(%d,'%s',%d,%d)",buyitem.book.book_id,buyitem.book.book_name,buyitem.user.id,buyitem.buy_num);
m_database->ExecuteSQL(strSQL);
}
CATCH_ALL(e)
{
e->GetErrorMessage(m_errmsg.GetBuffer(128),128);
AfxMessageBox(m_errmsg);
return -1;
}
END_CATCH_ALL
return 0;
}
int CECSERVERDlg::SendBuyBookInfo(int sockid, int user_id)//发送用户的购书信息
{
CString m_errmsg;
CString temp;
DataPack datapack;
memset(&datapack,0,DATAPACK_SIZE);
datapack.flag = BUYBOOK_SUCCESS;
TRY
{
//从表BUY_ITEM中查找用户的购书记录
CString strSQL;
strSQL.Format("select * from BUY_ITEM where USER_ID=%d",user_id);
CRecordset* m_pSet_tmp = new CRecordset(m_database);
m_pSet_tmp->Open(CRecordset::dynaset,strSQL);
if(m_pSet_tmp->IsBOF())
return -1;
//发送用户的购书记录
while(!m_pSet_tmp->IsEOF())
{
m_pSet_tmp->GetFieldValue("BOOK_ID",temp);
datapack.DataInfo.buyitem.book.book_id = atoi(temp.GetBuffer(0));
m_pSet_tmp->GetFieldValue("BOOK_NAME",temp);
strcpy(datapack.DataInfo.buyitem.book.book_name,temp);
m_pSet_tmp->GetFieldValue("USER_ID",temp);
datapack.DataInfo.buyitem.user.id = atoi(temp.GetBuffer(0));
m_pSet_tmp->GetFieldValue("BUY_NUM",temp);
datapack.DataInfo.buyitem.buy_num = atoi(temp.GetBuffer(0));
clientsocket[sockid].Send((const char*)&datapack,DATAPACK_SIZE);
m_pSet_tmp->MoveNext();
}
}
CATCH_ALL(e)
{
e->GetErrorMessage(m_errmsg.GetBuffer(128),128);
AfxMessageBox(m_errmsg);
return -1;
}
END_CATCH_ALL
return 0;
}
int CECSERVERDlg::DropBook(BuyItem& buyitem)//退购处理
{
CString m_errmsg;
TRY
{//删除用户的购书记录
CString strSQL ;
strSQL.Format("delete * from BUY_ITEM where USER_ID = %d AND BOOK_ID = %d",buyitem.user.id,buyitem.book.book_id);
m_database->ExecuteSQL(strSQL);
}
CATCH_ALL(e)
{
e->GetErrorMessage(m_errmsg.GetBuffer(128),128);
AfxMessageBox(m_errmsg);
return -1;
}
END_CATCH_ALL
return 0;
}
int CECSERVERDlg::WriteStr(CString str)//向界面上输出状态信息
{
CString strTemp = str;
strTemp += _T("\r\n");
m_message_list.ReplaceSel(strTemp);
return 0;
}
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -