rrqm / touchsocket Goto Github PK

static void Main(string[] args)
{

    var service = new HttpService();
    service.Setup(new TouchSocketConfig()//加载配置
        .SetListenIPHosts(7789)
        .SetRegistrator(new CanAotContainer())
        .ConfigureContainer(a =>//容器的配置顺序应该在最前面
        {
            a.AddConsoleLogger();//添加一个控制台日志注入（注意：在maui中控制台日志不可用）
            a.RegisterSingleton(service);//将服务器以单例注入。便于插件或其他地方获取。
        })
        .ConfigurePlugins(a =>
        {
            a.UseWebSocket()//添加WebSocket功能
                .SetWSUrl("/ws")//设置url直接可以连接。
                .SetVerifyConnection(VerifyConnection)
                .UseAutoPong();//当收到ping报文时自动回应pong

            a.Add<CloseExceptionPlugin>();
            a.Add<MessageRelayPlugin>();
            a.Add<DisconnectedPlugin>();
        }));

    service.Start();
    service.Logger.Info("服务器已启动");

    Console.ReadKey();
}

        tcpClient.Disconnected = (client, e) => {
            Console.WriteLine($"{DateTime.Now}：中控服务器、断开连接");
        };//从服务器断开连接，当连接不成功时不会触发。

        tcpClient.Received = (client, byteBlock, requestInfo) =>
        {

        };

        tcpClient.Setup(new TouchSocketConfig().
            SetRemoteIPHost($"{ip}:{port}").
            ConfigurePlugins((a) =>
        {

            a.UseReconnection(-1, false, 1000 * 5, (e) =>
            {
                Console.WriteLine($"{DateTime.Now}：中控服务器、重新连接成功");
            });

        }));

    protected Func<object, object[], object> CreateExpressionInvoker(MethodInfo method)
    {
        var instance = Expression.Parameter(typeof(object), "instance");
        var parameters = Expression.Parameter(typeof(object[]), "parameters");

        var instanceCast = method.IsStatic ? null : Expression.Convert(instance, method.DeclaringType);
        var parametersCast = method.GetParameters()
                                   .Select((p, i) =>
                                   {
                                       var parameter = Expression.ArrayIndex(parameters, Expression.Constant(i));
                                       return Expression.Convert(parameter, p.ParameterType.GetRefOutType()); // 这里需要使用GetRefOutType()，否则如果rpc函数有ref 参数时，使用表达式树生成代码时会报ByRef错误
                                   });
        var body = Expression.Call(instanceCast, method, parametersCast);

    TcpService service = new();
    service.IDChanged = (client, e) =>
    {
        client.Logger.Warning($"{client.GetIPPort()}ID变更: {e.OldID},{e.NewID}");
    };
    service.Received = (client, byteBlock, requestInfo) =>  //从客户端收到信息
    {
        try
        {
            TLVDataFrame buf = (TLVDataFrame)requestInfo;
            if (buf.Tag == 10)
            {
                var idsA = Encoding.UTF8.GetString(buf.Value, 0, buf.Length).SplitFirst('-');
                if (idsA[1] != "0")                          //处理待连接的客户端ID
                {
                    try
                    {
                        var idnew = idsA[1] + "-" + idsA[0];  //待连接的客户端的新ID
                        var idold = client.GetOtherIDs().Single((id) => id.StartsWith(idsA[1] + "-"));      //老ID
                        if (string.IsNullOrEmpty(idold))    //没有找到待连接的客户端
                        {
                            idsA[1] = "0";
                        }
                        else if (idold != idnew)            //待连接的客户端有变动
                        {
                            service.ResetID(idold, idnew);
                            service.Send(idnew, new ValueTLVDataFrame(10, Encoding.UTF8.GetBytes(idnew)));

                            var idsB = idold.SplitFirst('-');
                            if (idsB[1] != "0")             //原先还有连接关系。抢占，需要断开原有关系
                            {
                                idold = idsB[1] + "-" + idsB[0];
                                idnew = idsB[1] + "-0";
                                service.ResetID(idold, idnew);
                                service.Send(idnew, new ValueTLVDataFrame(10, Encoding.UTF8.GetBytes(idnew)));
                            }
                        }
                    }
                    catch
                    {
                        idsA[1] = "0";
                    }
                }

                if (idsA[0] != "0")                 //处理自身的ID关系
                {
                    var idnew = idsA[0] + "-" + idsA[1];
                    var idold = client.ID;
                    client.ResetID(idnew);
                    client.Send(new ValueTLVDataFrame(10, Encoding.UTF8.GetBytes(idnew)));
                    client.Logger.Warning($"{client.GetIPPort()}变更ID:{idnew},在线客户数{service.Count}");

                    var idsC = idold.SplitFirst('-');
                    if (idsC[1] != "0")             //原先还有连接关系。需要放弃，断开原有关系
                    {
                        idold = idsC[1] + "-" + idsC[0];
                        idnew = idsC[1] + "-0";
                        service.ResetID(idold, idnew);
                        service.Send(idnew, new ValueTLVDataFrame(10, Encoding.UTF8.GetBytes(idnew)));
                    }
                }
            }
            else                    //正常需要转发的数据
            {
                var idsD = client.ID.SplitFirst('-');
                if (idsD[1] != "0")
                {
                    var rmid = idsD[1] + "-" + idsD[0];
                    service.Send(rmid, buf);
                    client.Logger.Info($"{client.GetIPPort()} 透传 {rmid},{buf.Tag},{buf.Length}");
                }
            }
        }
        catch (Exception ex)
        {
            client.Logger.Error(ex.ToString());
        }
    };

private void button_SendFile_Click(object sender, EventArgs e)
{
          foreach (SocketClient Socket in Server.GetClients())
          {
                  Socket.SendFile(fPath);
          }
}

public void SendFile(string path)
 {
        Task T = new Task(() =>
        {
                byte[] buffer = new byte[1024*64];
                using (var reader = FilePool.GetReader(path))
                {
                        int seq = 0;//文件过大，需要切片发送
                        while (true)
                        {
                                int r = reader.Read(buffer, 0, buffer.Length);
                                if (r == 0) { break; }

                                byte[] packet = new RequestInfo(buffer).BuildAsBytes();
                                base.Send(packet, 0, packet.Length);
                                seq++;
                        }
                        //Thread.Sleep(5000);//如果让每个Task结束发送后都等待一段时间，就不会发生该问题
                }

        }); TStart();
}