阅读量:0
在C#中实现实时消息推送，通常可以使用WebSocket技术。WebSocket提供了一个全双工通信通道，允许服务器主动向客户端推送消息。以下是一个简单的示例，展示了如何使用C#和ASP.NET Core实现WebSocket实时消息推送。
1. 创建ASP.NET Core项目
首先，创建一个新的ASP.NET Core项目。在Visual Studio中，可以选择“Web Application”模板，然后选择“API”作为项目类型。
2. 添加WebSocket支持
在Startup.cs文件中，添加对WebSocket的支持。首先，确保已经安装了Microsoft.AspNetCore.Http.Extensions.WebSocket包。如果没有安装，可以使用以下命令进行安装：
dotnet add package Microsoft.AspNetCore.Http.Extensions.WebSocket 然后在Startup.cs的ConfigureServices方法中添加WebSocket中间件：
public void ConfigureServices(IServiceCollection services) { services.AddControllers(); services.AddWebSocket(options => { options.AllowedOrigins.Add("*"); }); } 在Configure方法中配置WebSocket中间件：
public void Configure(IApplicationBuilder app, IWebHostEnvironment env) { if (env.IsDevelopment()) { app.UseDeveloperExceptionPage(); } app.UseRouting(); app.UseEndpoints(endpoints => { endpoints.MapControllers(); endpoints.MapWebSocket("/ws", options => { options.AllowedOrigins.Add("*"); }); }); } 3. 实现WebSocket连接处理
创建一个新的类WebSocketHandler来处理WebSocket连接：
public class WebSocketHandler { private readonly WebSocketContext _context; private readonly ConcurrentQueue<string> _messages = new ConcurrentQueue<string>(); public WebSocketHandler(WebSocketContext context) { _context = context; } public async Task RunAsync() { var buffer = new byte[1024 * 4]; WebSocketReceiveResult result; do { result = await _context.WebSocket.ReceiveAsync(new WebSocketReceiveMessage { Buffer = buffer, MessageType = WebSocketMessageType.Text, Count = buffer.Length }); if (result.MessageType == WebSocketMessageType.Text) { var message = Encoding.UTF8.GetString(buffer, 0, result.Count); _messages.Enqueue(message); } } while (!result.CloseStatus.HasValue); await _context.WebSocket.CloseAsync(result.CloseStatus.Value, result.CloseStatusDescription, CancellationToken.None); } public async Task SendMessageAsync(string message) { if (_context.WebSocket.State == WebSocketState.Open) { await _context.WebSocket.SendAsync(new WebSocketSendMessage { MessageType = WebSocketMessageType.Text, Data = Encoding.UTF8.GetBytes(message) }); } } public async Task<IEnumerable<string>> GetMessagesAsync() { while (_messages.TryDequeue(out var message)) { yield return message; } } } 4. 创建WebSocket控制器
创建一个新的控制器WebSocketController来处理WebSocket连接和消息：
[ApiController] [Route("api/[controller]")] public class WebSocketController : ControllerBase { private readonly WebSocketHandler _webSocketHandler; public WebSocketController(WebSocketHandler webSocketHandler) { _webSocketHandler = webSocketHandler; } [HttpPost] public async Task<IActionResult> Connect() { var webSocket = await HttpContext.WebSockets.AcceptWebSocketAsync(); var context = new WebSocketContext(webSocket); _webSocketHandler = new WebSocketHandler(context); await _webSocketHandler.RunAsync(); return Ok(); } [HttpPost] public async Task<IActionResult> SendMessage(string message) { await _webSocketHandler.SendMessageAsync(message); return Ok(); } [HttpGet] public async Task<IEnumerable<string>> GetMessages() { var messages = new List<string>(); foreach (var message in await _webSocketHandler.GetMessagesAsync()) { messages.Add(message); } return Ok(messages); } } 5. 测试WebSocket连接
现在你可以使用任何支持WebSocket的客户端来测试你的WebSocket服务器。例如，你可以使用浏览器中的JavaScript代码来连接到/ws端点并发送接收消息：
const socket = new WebSocket('ws://localhost:5000/ws'); socket.addEventListener('open', (event) => { console.log('WebSocket connection opened:', event); socket.send('Hello Server!'); }); socket.addEventListener('message', (event) => { console.log('Message from server:', event.data); }); socket.addEventListener('close', (event) => { console.log('WebSocket connection closed:', event); }); 通过以上步骤，你可以在C#中实现一个简单的实时消息推送系统。你可以根据需要扩展这个示例，例如添加身份验证、消息持久化等功能。
