c# socket執行緒池實現
伺服器端:
PoolServer.cs類
using System;
using System.Collections.Generic;
using System.Text;
using System.Net;
using System.Net.Sockets;
using System.IO;
using System.Threading;
namespace Net.Common.Pool
{
/// <summary>
/// 伺服器
/// 1.呼叫構造方法:
/// (1)載入引數:埠、最大活動數量
/// (2)啟用PooledConnectionHandler執行緒,根據最大活動數量建立執行緒,並且全部啟動
/// PooledConnectionHandler執行緒,讀取pool池,判斷池內是否為空,如果不為空,則讓執行緒安全的讀取池內的第一個,並且進行業務處理
/// 2.開啟服務
/// (1)將TcpListener物件初始化
/// (2)等待客戶機的請求,呼叫AcceptConnections()方法
///
/// @author layicr
/// </summary>
public class PoolServer
{
/// <summary>
/// 最大活動數量
/// </summary>
private int maxConnections;
/// <summary>
/// 最大活動數量
/// </summary>
public int MaxConnections
{
get { return maxConnections; }
set { maxConnections = value; }
}
/// <summary>
/// 埠
/// </summary>
private int port;
/// <summary>
/// 埠
/// </summary>
public int Port
{
get { return port; }
set { port = value; }
}
/// <summary>
/// 第幾張網絡卡,預設為第一個,值為0
/// </summary>
private int addressIndex = 0;
/// <summary>
/// 第幾張網絡卡,預設為第一個,值為0
/// </summary>
public int AddressIndex
{
get { return addressIndex; }
set { addressIndex = value; }
}
/// <summary>
/// 是否開啟服務
/// </summary>
private bool reLease = false;
/// <summary>
/// 執行緒集合
/// </summary>
private List<Thread> listConnectionThread = new List<Thread>();
private TcpListener listener = null;
private TcpClient client = null;
/// <summary>
///
/// </summary>
public PoolServer()
{
}
/// <summary>
///
/// </summary>
/// <param name="port">埠</param>
/// <param name="maxConnections">最大活動數量</param>
public PoolServer(int port, int maxConnections)
{
this.port = port;
this.maxConnections = maxConnections;
}
/// <summary>
///
/// </summary>
/// <param name="addressIndex">第幾張網絡卡</param>
/// <param name="port">埠</param>
/// <param name="maxConnections">最大活動數量</param>
public PoolServer(int addressIndex, int port, int maxConnections)
{
this.addressIndex = addressIndex;
this.port = port;
this.maxConnections = maxConnections;
}
/// <summary>
/// 啟動服務
/// </summary>
public void ServerStart()
{
this.reLease = true;
IPHostEntry IpEntry = Dns.GetHostEntry(Dns.GetHostName()); //得到主機IP
IPAddress iPAddress = IpEntry.AddressList[addressIndex];
listener = new TcpListener(iPAddress, this.port);
listener.Start();
//開啟執行緒
SetUpHandlers();
// 允許客戶機連線到伺服器,等待客戶機請求
AcceptConnections();
}
/// <summary>
/// 停止服務
/// </summary>
public void ServerStop()
{
if (listener != null)
{
try
{
listener.Stop();
listener = null;
//清空池資料
PoolConnectionHandler.pool.Clear();
reLease = false;
//讓PooledConnectionHandler執行緒停止
foreach(Thread item in listConnectionThread)
{
//停止
item.Abort();
}
//清空
listConnectionThread.Clear();
}
catch (Exception e)
{
Console.WriteLine(e.Message);
}
}
}
/// <summary>
/// 允許客戶機連線到伺服器,等待客戶機請求
/// 1.開啟一個執行緒
/// </summary>
private void AcceptConnections()
{
new Thread(ListenerStart).Start();
}
/// <summary>
/// 允許客戶機連線到伺服器,等待客戶機請求
/// 2.判斷當前reLease(是否啟用服務)的值是否為true
/// 如果沒有啟用服務,則執行緒關閉
/// 如果啟用服務,則開始等待客戶機請求,如果受到了客戶機的請求,則加入池中
/// </summary>
private void ListenerStart()
{
while (reLease)
{
try
{
client = listener.AcceptTcpClient();
HandleConnection(client);
}
catch (Exception e)
{
//
Console.WriteLine(e.Message);
}
}
}
/// <summary>
/// 將請求加入池中
/// </summary>
/// <param name="client"></param>
private void HandleConnection(TcpClient client)
{
PoolConnectionHandler.ProcessRequest(client);
}
/// <summary>
/// 開啟PooledConnectionHandler執行緒
/// </summary>
private void SetUpHandlers()
{
Thread thread = null;
for (int i = 0; i < maxConnections; i++)
{
PoolConnectionHandler currentHandler = new PoolConnectionHandler();
thread = new Thread(new ThreadStart(currentHandler.Run));
thread.Name = "Thread(" + i+")";
thread.Start();
listConnectionThread.Add(thread);
}
}
}
}
PoolConnectionHandler.cs類
using System;
using System.Collections.Generic;
using System.Text;
using System.Net;
using System.Net.Sockets;
using System.IO;
using System.Threading;
namespace Net.Common.Pool
{
public class PoolConnectionHandler
{
/// <summary>
/// 池
/// </summary>
public static LinkedList<TcpClient> pool = new LinkedList<TcpClient>();
/// <summary>
///
/// </summary>
protected TcpClient client = null;
/// <summary>
///
/// </summary>
public PoolConnectionHandler() { }
/// <summary>
/// 業務處理
/// </summary>
public void HandleConnection()
{
try
{
//這裡呼叫, 業務層的方法
HandleMain.Handle(client);
}
catch
{ }
}
/// <summary>
/// 將未處理的請求加入池中
/// </summary>
/// <param name="requestToHandle"></param>
public static void ProcessRequest(TcpClient requestToHandle)
{
//對池進行加鎖
lock (pool)
{
//把未處理的請求加入池
pool.AddLast(requestToHandle);
//將等待執行緒被喚醒
Monitor.PulseAll(pool);
}
}
/// <summary>
/// 一直接讀取pool(池),並且對pool進行加鎖,判斷其集合是否為空
/// 如果pool為空,則一直的讀取
/// 如要pool不為空,則取出第一個值,到了這一步鎖解除(儲存池取值的過程是安全,取出的值是唯一),接著執行handleConnection
/// </summary>
public void Run()
{
while (true)
{
try
{
lock (pool)
{
while (pool.Count == 0)
{
Monitor.Wait(pool);
}
//取第一個值
client = pool.First.Value;
//移除
pool.RemoveFirst();
}
HandleConnection();
}
catch (Exception e)
{
Console.WriteLine(e.Message);
}
}
}
}
}
HandleMain.cs業務類:
using System;
using System.Collections.Generic;
using System.Text;
using System.Net;
using System.Net.Sockets;
using System.IO;
namespace Net.Common.Pool
{
/// <summary>
/// 業務類,測試類
/// </summary>
public class HandleMain
{
public static void Handle(TcpClient client)
{
Console.WriteLine("當前執行緒:" + System.Threading.Thread.CurrentThread.Name);
NetworkStream networkStream = null;
//輸入
StreamReader streamReader = null;
FileStream fileStream = null;
byte[] buffer = null;
int length = 3*1024;
try
{
networkStream = client.GetStream();
streamReader = new StreamReader(networkStream);
//檔名
string fileName = streamReader.ReadLine();
fileStream = new FileStream(@"E:\1\" + fileName, FileMode.Create, FileAccess.Write);
int readLength = 0;
// int sumLength = 0;
// DateTime myDtStart = DateTime.Now;
buffer = new byte[length];
while ((readLength = networkStream.Read(buffer, 0, length)) > 0)
{
fileStream.Write(buffer, 0, readLength);
fileStream.Flush();
//總量
// sumLength += readLength;
//double total = sumLength / 1000;
// TimeSpan ts = DateTime.Now - myDtStart;
//當前速度
// double speed = total / ts.TotalSeconds;
}
fileStream.Flush();
}
catch (Exception ex)
{
Console.WriteLine(ex.Message);
}
finally
{
if (fileStream != null)
{
fileStream.Close();
}
if (streamReader != null)
{
streamReader.Close();
}
if (networkStream != null)
{
networkStream.Close();
}
if (client != null)
{
client.Close();
}
}
}
}
}
執行伺服器端:
int port = 3848;
PoolServer poolServer = new PoolServer(port, 10);
poolServer.ServerStart();
客戶端測試:
using System.Net;
using System.IO;
using System.Net.Sockets;
public void Send(){
string serverIP = "192.168.1.138";//伺服器IP地址
int port = 3848;//埠號
string filePath = "c:\\1.jpg";//要傳輸的檔案
int length = 2048;
TcpClient client = null;
NetworkStream networkStream = null;
FileStream fileStream = null;
StreamWriter streamWriter = null;
byte[] buffer = null;
int readLength = 0;
String msg = null;//傳輸訊息
string fileExt = null; //檔案的字尾
while (true)
{
System.Threading.Thread.Sleep(900);//很重要/////
try
{
//連線伺服器
client = new TcpClient(serverIP, port);
client.SendTimeout = 60 * 1000;//傳送超時值,以毫秒為單位
networkStream = client.GetStream();
//輸出
streamWriter = new StreamWriter(networkStream);
//輸入
// StreamReader sr = new StreamReader(networkStream);
fileStream = new FileStream(filePath, FileMode.Open, FileAccess.Read, FileShare.None);
//得到字尾,包括"."
fileExt = Path.GetExtension(filePath).ToLower();
//遠端伺服器生成的檔名稱
msg = System.Guid.NewGuid().ToString() + fileExt;
//傳輸訊息
streamWriter.WriteLine(msg);
streamWriter.Flush();
buffer = new byte[length];
while ((readLength = fileStream.Read(buffer, 0, length)) > 0)
{
networkStream.Write(buffer, 0, readLength);
networkStream.Flush();
}
networkStream.Flush();
}
catch (Exception ex)
{
MessageBox.Show(ex.Message);
}
finally
{
if (fileStream != null)
{
fileStream.Close();
}
if (streamWriter != null)
{
streamWriter.Close();
}
if (networkStream != null)
{
networkStream.Close();
}
if (client != null)
{
client.Close();
}
}
}
}