PageRenderTime 47ms CodeModel.GetById 23ms RepoModel.GetById 1ms app.codeStats 0ms

/src/Modules/FileTransfer/TransferManager.h

https://gitlab.com/yoage/TTWinClient
C Header | 109 lines | 66 code | 12 blank | 31 comment | 0 complexity | 6b16bfec6e5b85291c66ef49865fa1e4 MD5 | raw file
Possible License(s): LGPL-3.0 
    

  1.  /*******************************************************************************
    2. 

  2.  *  @file      TransferManager.h 2014\9\3 11:19:29 $
    3. 

  3.  *  @author    大佛<dafo@mogujie.com>
    4. 

  4.  *  @brief     文件传输管理
    5. 

  5.  ******************************************************************************/
    6. 

  6. 

  7. #ifndef TRANSFERMANAGER_3046D4C9_3E8D_4C36_93B4_092651F5B66F_H__
    8. 

  8. #define TRANSFERMANAGER_3046D4C9_3E8D_4C36_93B4_092651F5B66F_H__
    9. 

  9. 

  10. #include "utility/TTAutoLock.h"
    11. 

  11. #include <list>
    12. 

  12. #include <map>
    13. 

  13. 

  14. /******************************************************************************/
    15. 

  15. class TransferFile;
    16. 

  16. class FileTransferUIThread;
    17. 

  17. 

  18. class TransferFileEntity
    19. 

  19. {
    20. 

  20. public:
    21. 

  21. 	TransferFileEntity();
    22. 

  22. public:
    23. 

  23. 	UInt32				nPort;
    24. 

  24. 	UInt32				nFileSize;
    25. 

  25. 	UInt32				nClientMode;			//CLIENT_REALTIME_SENDER = 1,CLIENT_REALTIME_RECVER,CLIENT_OFFLINE_UPLOAD,CLIENT_OFFLINE_DOWNLOAD
    26. 

  26. 	UInt32				nProgress;				//传输进度
    27. 

  27. 	TransferFile*		pFileObject;			//传输文件
    28. 

  28. 	std::string			sFromID;
    29. 

  29. 	std::string         sTaskID;				//文件唯一标示符
    30. 

  30. 	std::string         sToID;
    31. 

  31. 	std::string         sFileName;
    32. 

  32. 	std::string			sIP;
    33. 

  33. 	std::string         sPathOfflineFileOnSev;	//离线文件传输时，保存在文件服务器上的位置
    34. 

  34. 

  35. 	UInt32		        time;            //传文件发起时间
    36. 

  36. private:
    37. 

  37. 	CString				sSavePath;				//接收时，另存为的地址
    38. 

  38. 

  39. public:
    40. 

  40. 	CString getSaveFilePath();					//完整的保存文件路径
    41. 

  41. 	CString getSaveFloderFilePath();			//保存的文件夹位置
    42. 

  42. 	void setSaveFilePath(const CString& sPath);
    43. 

  43. 

  44. 	CString getRealFileName();
    45. 

  45. };
    46. 

  46. 

  47. typedef std::map<std::string, TransferFileEntity>     TransferFileMap;//群列表
    48. 

  48. 

  49. class TransferFileEntityManager
    50. 

  50. {
    51. 

  51. public:
    52. 

  52. 	~TransferFileEntityManager();
    53. 

  53. 	static TransferFileEntityManager* getInstance();
    54. 

  54. 	BOOL pushTransferFileEntity(IN  TransferFileEntity& FileInfo);
    55. 

  55. 	BOOL getFileInfoByTaskId(IN const std::string& sID, OUT TransferFileEntity& FileInfo);
    56. 

  56. 	BOOL DeleteFileInfoByTaskId(IN const std::string& sID);
    57. 

  57. 	void getAllTransferFileEntityFileID(std::list<std::string>& fileIdList);
    58. 

  58. 	BOOL updateFileInfoBysTaskID(IN const TransferFileEntity& info);
    59. 

  59. 	BOOL openFileByFileID(IN const std::string& sID);
    60. 

  60. 	BOOL openFileFolderByTaskID(IN const std::string& sID);
    61. 

  61. 	BOOL kickMapFileItemToVecFile(IN std::string& sfId);
    62. 

  62. 	void pushTransferFileEntityToVec(IN  TransferFileEntity& FileInfo);
    63. 

  63. 	BOOL checkIfIsSending(IN  CString sFilePath);//正在传输的文件不能再次传输
    64. 

  64. 

  65. public:
    66. 

  66. 	/**@name 包含多个file socket资源的线程相关*/
    67. 

  67. 	//@{
    68. 

  68. 	BOOL startup();
    69. 

  69. 	void shutdown();
    70. 

  70. 	/**
    71. 

  71. 	* 在UI thread中异步创建文件传输socket
    72. 

  72. 	*
    73. 

  73. 	* @param   std::string & taskId 文件传输任务id
    74. 

  74. 	* @return  void
    75. 

  75. 	* @exception there is no any exception to throw.
    76. 

  76. 	*/
    77. 

  77. 	void openFileSocketByTaskId(std::string& taskId);
    78. 

  78. 	/**
    79. 

  79. 	* 在UI thread中关闭文件传输socket
    80. 

  80. 	*
    81. 

  81. 	* @param   std::string & taskId 文件传输任务id
    82. 

  82. 	* @return  void
    83. 

  83. 	* @exception there is no any exception to throw.
    84. 

  84. 	*/
    85. 

  85. 	void closeFileSocketByTaskId(std::string& taskId);
    86. 

  86. 	/**
    87. 

  87. 	* 向文件传输服务器发送接收文件信令
    88. 

  88. 	*
    89. 

  89. 	* @param   std::string & taskId
    90. 

  90. 	* @return  BOOL
    91. 

  91. 	* @exception there is no any exception to throw.
    92. 

  92. 	*/
    93. 

  93. 	BOOL acceptFileTransfer(const std::string& taskId);//接受文件传输
    94. 

  94. 	BOOL rejectFileTransfer(const std::string& taskId);//拒绝文件传输
    95. 

  95. 	BOOL cancelFileTransfer(const std::string& taskId);//取消文件传输
    96. 

  96. 	FileTransferUIThread* getFileTransferThread();
    97. 

  97. 	//@}
    98. 

  98. 

  99. private:
    100. 

  100. 	TransferFileEntityManager();
    101. 

  101. 

  102. private:
    103. 

  103. 	util::TTFastLock					m_lock;
    104. 

  104. 	TransferFileMap						m_MapFile;
    105. 

  105. 	std::vector<TransferFileEntity>		m_VecFinishedFile;			//已经传输完的文件,任何操作过的文件都会被扔到这里
    106. 

  106. 	FileTransferUIThread*				m_fileUIThread;
    107. 

  107. };
    108. 

  108. /******************************************************************************/
    109. 

  109. #endif// TRANSFERMANAGER_3046D4C9_3E8D_4C36_93B4_092651F5B66F_H__
    110. 

  110.