一个基于socket的资源共享平台的实现（三）

需要共享资源，则需要探测本地资源分布情况。

此处我们用的算法比较2……不过还是说说吧。

 

下面使用MFC的CFileFind实现一个本地文件递归收集器，以jason格式存储文件路径和文件大小：

 

#ifndef COLLECTER_H_ #define COLLECTER_H_ #include<iostream> #include<vector> #include "StdioFileEx/StdioFileEx.h" #define TargetFile "result.txt" #define CfgFileName "path.txt" using namespace std; struct FileStruct { CString path; __int64 size; }; vector<FileStruct> files; void findfile(CString dir) { CFileFind finder; BOOL bWorking = finder.FindFile(dir + _T("//*.*")); while (bWorking) { bWorking = finder.FindNextFile(); //列举文件 if( finder.IsDots() == false && finder.IsDirectory() == false ) { FileStruct file; CString tmp = dir + _T("//") + CString( finder.GetFileName() ); //只传送文件名 //CString tmp = CString( finder.GetFileName() ); file.path = tmp; __int64 iFileLenInByte = (__int64)finder.GetLength(); __int64 size = iFileLenInByte; file.size = size; files.push_back( file ); } //递归访问子目录 if( finder.IsDirectory() && finder.IsDots() == false ) findfile( dir + _T("//") + CString( finder.GetFileName() ) ); } } /* [{"f_name":"文件名","f_hash":"文件哈希，32字符","f_length":null/片长，数字不要加引号,"f_type":null/"文件类型就后缀吧先","f_time":null/"年代字符串 加双引号","f_intro":"简介，类似年代"},{}...{}] */ void WriteResult() { CStdioFileEx fp; fp.Open(_T(TargetFile),CFile::modeWrite|CFile::modeCreate); fp.WriteString( _T("[")); for( vector<FileStruct>::iterator iter = files.begin(); iter != files.end(); ++iter) { if( iter != files.begin() ) fp.WriteString(_T(",")); CString MyFileSize; MyFileSize.Format(_T("%ld"),iter->size); fp.WriteString(_T("{/"fname/":/"")); fp.WriteString( iter->path ); fp.WriteString(_T("/",")); fp.WriteString(_T("/"fsize/":/"")); fp.WriteString( MyFileSize ); fp.WriteString(_T("/"}")); //fp.WriteString( iter->size ); } fp.WriteString( _T("]")); fp.Close(); } void ReadCfg(CString& path) { CStdioFileEx fp; fp.Open(_T(CfgFileName),CFile::modeRead); fp.ReadString( path ); fp.Close(); } void StartCollect() { CString SearchPath; ReadCfg( SearchPath ); cout<<"searching please wait."<<endl;; findfile( SearchPath ); WriteResult(); cout<<"finish"<<endl; system("pause"); } #endif

 

然后定期收集，对收集结果MD5，若发生变化，则上传服务器。

服务器端使用一个数据结构维护所有资源站资源，对于用户的搜索，我们此处使用最简单的比较方法，看是否包含子串，此处就不详加介绍了。

 

服务器对所有的客户端进行集中管理，需要维护一张在线用户列表。

关于服务器如何探测资源站是否在线，TCP主动的的连接与断开可以检测到，但是若是被动的断线等，是无法正确响应的。所以此处我们还是使用心跳包模型——客户端每隔一段时间就向服务器发送心跳包，证明自己还“活着”，服务器若长时间没有收到某个客户端的心跳，则认为它“死掉”了，就从在线列表里将其KICK。

 

服务器端我们是使用的CONSOLE管理，实际就是单开一个control线程，可以由管理员输入命令进行管理。如查询在线列表、查询资源、踢出用户、广播消息等，这里也不加详细介绍了。

 

服务器需要维护LOG文件，如下所示：

 

【2010-01-25 9:0:24】开启服务器
【2010-01-25 9:0:31】127.0.0.1登入服务器
【2010-01-25 9:0:31】127.0.0.1申请搜索，关键字 =
【2010-01-25 9:1:32】127.0.0.1长时间未响应，从服务器移除
【2010-01-25 9:3:20】开启服务器
【2010-01-25 9:3:23】0.0.0.0申请搜索，关键字 =
【2010-01-25 9:3:30】开启服务器
【2010-01-25 9:3:34】127.0.0.1登入服务器
【2010-01-25 9:3:34】127.0.0.1申请搜索，关键字 =
【2010-01-25 9:3:37】0.0.0.0申请搜索，关键字 =
【2010-01-25 9:3:42】0.0.0.0申请搜索，关键字 = 1
【2010-01-25 9:4:14】127.0.0.1申请搜索，关键字 =
【2010-01-25 9:5:19】127.0.0.1广播消息：sss
【2010-01-25 9:5:21】127.0.0.1广播消息：ahaha
【2010-01-25 9:5:23】127.0.0.1申请搜索，关键字 =
【2010-01-25 9:6:0】127.0.0.1长时间未响应，从服务器移除
【2010-01-25 9:7:9】127.0.0.1登入服务器
【2010-01-25 9:7:9】127.0.0.1申请搜索，关键字 =
【2010-01-25 9:7:46】127.0.0.1长时间未响应，从服务器移除
【2010-01-25 9:9:25】127.0.0.1登入服务器
【2010-01-25 9:9:26】127.0.0.1申请搜索，关键字 =
【2010-01-25 9:14:39】127.0.0.1登入服务器
【2010-01-25 9:15:15】127.0.0.1长时间未响应，从服务器移除
【2010-01-25 9:17:33】127.0.0.1登入服务器
【2010-01-25 9:17:33】127.0.0.1申请搜索，关键字 =
【2010-01-25 9:17:42】127.0.0.1申请搜索，关键字 =
【2010-01-25 9:18:41】127.0.0.1长时间未响应，从服务器移除
【2010-01-25 9:19:7】127.0.0.1登入服务器
【2010-01-25 9:19:8】127.0.0.1申请搜索，关键字 =
【2010-01-25 9:19:45】127.0.0.1长时间未响应，从服务器移除
【2010-01-25 9:20:6】127.0.0.1登入服务器
【2010-01-25 9:20:7】127.0.0.1申请搜索，关键字 =
【2010-01-25 9:21:56】127.0.0.1长时间未响应，从服务器移除
【2010-03-10 14:24:49】开启服务器
【2010-03-10 14:25:50】59.175.191.118登入服务器
【2010-03-10 14:28:18】127.0.0.1登入服务器
【2010-03-10 14:28:18】127.0.0.1申请搜索，关键字 =
【2010-03-10 14:29:32】系统广播消息:test
【2010-03-10 14:29:32】系统广播消息:test
【2010-03-10 14:30:44】127.0.0.1长时间未响应，从服务器移除
【2010-03-10 14:30:53】123.118.15.209登入服务器
【2010-03-10 14:30:54】123.118.15.209申请搜索，关键字 =

 

这里是我的服务器日志类。

 

需要注意的是由于多线程写日志，需要线程互斥访问log文件，注意 is_busy 变量。

#ifndef NETSHARE_LOG_H_ #define NETSHARE_LOG_H_ #include <string> #include <vector> #include <fstream> #include <ctime> using namespace std; //服务器日志管理类 class NS_Log { public: NS_Log( string filename ) : logfile( filename.c_str(), ios::app ) , is_busy( false ) {} ~NS_Log() { logfile.close(); } void Write( string dat ) { while( is_busy ) //互斥 { Sleep( 20 ); } is_busy = true; string now_time = GetNowTime(); logfile<<"【"<<now_time<<"】"<<dat<<endl; is_busy = false; } private: //获取当前时间 string GetNowTime() { char pTime[1000]; int year=0,mon=0,day=0,hour=0,min=0,sec=0; time_t now_time; now_time = time(NULL); struct tm *m_tm; m_tm = gmtime(&(time_t)now_time); year = m_tm->tm_year+1900; mon = m_tm->tm_mon+1; day = m_tm->tm_mday; hour = m_tm->tm_hour; min = m_tm->tm_min; sec = m_tm->tm_sec; sprintf(pTime, "%04d-%02d-%02d %d:%d:%d", year, mon, day, hour, min, sec); return string( pTime ); } fstream logfile; bool is_busy; }; #endif

 

未完待续……