From f163e20d0aa57c47d1465228f543dba1ea4da8b0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=80=81=E4=B9=9D?= <366193849@qq.com>
Date: Tue, 2 Nov 2021 22:51:54 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BD=BF=E7=94=A8Task?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 FastGithub.UI/FastGithub.UI.csproj            |  6 ++--
 FastGithub.UI/MainWindow.xaml                 |  2 +-
 .../{LogListBox.xaml => UdpLogListBox.xaml}   | 18 +++++++---
 ...gListBox.xaml.cs => UdpLogListBox.xaml.cs} | 36 +++++++++++++------
 4 files changed, 43 insertions(+), 19 deletions(-)
 rename FastGithub.UI/{LogListBox.xaml => UdpLogListBox.xaml} (68%)
 rename FastGithub.UI/{LogListBox.xaml.cs => UdpLogListBox.xaml.cs} (62%)

diff --git a/FastGithub.UI/FastGithub.UI.csproj b/FastGithub.UI/FastGithub.UI.csproj
index fec018e..0908950 100644
--- a/FastGithub.UI/FastGithub.UI.csproj
+++ b/FastGithub.UI/FastGithub.UI.csproj
@@ -82,7 +82,7 @@
       <Generator>MSBuild:Compile</Generator>
       <SubType>Designer</SubType>
     </Page>
-    <Page Include="LogListBox.xaml">
+    <Page Include="UdpLogListBox.xaml">
       <SubType>Designer</SubType>
       <Generator>MSBuild:Compile</Generator>
     </Page>
@@ -98,8 +98,8 @@
     <Compile Include="FlowChart.xaml.cs">
       <DependentUpon>FlowChart.xaml</DependentUpon>
     </Compile>
-    <Compile Include="LogListBox.xaml.cs">
-      <DependentUpon>LogListBox.xaml</DependentUpon>
+    <Compile Include="UdpLogListBox.xaml.cs">
+      <DependentUpon>UdpLogListBox.xaml</DependentUpon>
     </Compile>
     <Compile Include="MainWindow.xaml.cs">
       <DependentUpon>MainWindow.xaml</DependentUpon>
diff --git a/FastGithub.UI/MainWindow.xaml b/FastGithub.UI/MainWindow.xaml
index 97b193a..97b6371 100644
--- a/FastGithub.UI/MainWindow.xaml
+++ b/FastGithub.UI/MainWindow.xaml
@@ -152,7 +152,7 @@
 
             <TabItem Style="{StaticResource TabItemExWithUnderLineStyle}" Header="日志记录" Height="40" Width="100" Margin="5 0" FontSize="18">
                 <Grid Background="#f7f7f7">
-                    <local:LogListBox Margin="16"></local:LogListBox>
+                    <local:UdpLogListBox/>
                 </Grid>
             </TabItem>
 
diff --git a/FastGithub.UI/LogListBox.xaml b/FastGithub.UI/UdpLogListBox.xaml
similarity index 68%
rename from FastGithub.UI/LogListBox.xaml
rename to FastGithub.UI/UdpLogListBox.xaml
index 9a241c7..cb459ed 100644
--- a/FastGithub.UI/LogListBox.xaml
+++ b/FastGithub.UI/UdpLogListBox.xaml
@@ -1,4 +1,4 @@
-<UserControl x:Class="FastGithub.UI.LogListBox"
+<UserControl x:Class="FastGithub.UI.UdpLogListBox"
              xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
              xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
              xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006" 
@@ -8,23 +8,33 @@
              d:DesignHeight="450" d:DesignWidth="800">
     <Grid>
         <ListBox 
-            x:Name="listBox"
+            x:Name="listBox" 
             ItemsSource="{Binding LogList}"
             VirtualizingPanel.IsVirtualizing="True"
             ScrollViewer.CanContentScroll="True"
             ScrollViewer.HorizontalScrollBarVisibility="Disabled" 
             BorderThickness="0"
             Background="#f7f7f7">
+            <ListBox.ItemContainerStyle>
+                <Style TargetType="{x:Type ListBoxItem}">
+                    <Setter Property="FocusVisualStyle" Value="{x:Null}" />
+                    <Style.Triggers>
+                        <Trigger Property="IsSelected" Value="True">
+                            <Setter Property="FocusManager.FocusedElement" Value="{Binding RelativeSource={RelativeSource Self}}" />
+                        </Trigger>
+                    </Style.Triggers>
+                </Style>
+            </ListBox.ItemContainerStyle>
             <ListBox.ItemTemplate>
                 <DataTemplate>
-                    <StackPanel Margin="0 5">
+                    <StackPanel Margin="16 5">
                         <StackPanel.ContextMenu>
                             <ContextMenu>
                                 <MenuItem Header="复制本项" Click="MenuItem_Copy_Click" />
                                 <MenuItem Header="清除所有" Click="MenuItem_Clear_Click" />
                             </ContextMenu>
                         </StackPanel.ContextMenu>
-                        <TextBlock TextWrapping="Wrap" Text="{Binding Timestamp, StringFormat={}{0:yyyy-MM-dd HH:mm:ss.fff}}"/>
+                        <TextBlock TextWrapping="Wrap" FontSize="12" Text="{Binding Timestamp, StringFormat={}{0:yyyy-MM-dd HH:mm:ss.fff}}"/>
                         <TextBlock TextWrapping="Wrap" FontSize="12" Margin="0 5" Text="{Binding SourceContext}"/>
                         <TextBlock TextWrapping="Wrap" FontSize="14" FontWeight="Light" Text="{Binding Message}">
                             <TextBlock.Foreground>
diff --git a/FastGithub.UI/LogListBox.xaml.cs b/FastGithub.UI/UdpLogListBox.xaml.cs
similarity index 62%
rename from FastGithub.UI/LogListBox.xaml.cs
rename to FastGithub.UI/UdpLogListBox.xaml.cs
index 8835fb8..be1421b 100644
--- a/FastGithub.UI/LogListBox.xaml.cs
+++ b/FastGithub.UI/UdpLogListBox.xaml.cs
@@ -3,43 +3,57 @@ using System.Collections.ObjectModel;
 using System.Net;
 using System.Net.Sockets;
 using System.Text;
+using System.Threading.Tasks;
 using System.Windows.Controls;
 
 namespace FastGithub.UI
 {
     /// <summary>
-    /// LogListBox.xaml 的交互逻辑
+    /// UdpLogListBox.xaml 的交互逻辑
     /// </summary>
-    public partial class LogListBox : UserControl
+    public partial class UdpLogListBox : UserControl
     {
         private readonly byte[] buffer = new byte[ushort.MaxValue];
         private readonly Socket socket = new Socket(SocketType.Dgram, ProtocolType.Udp);
 
         public ObservableCollection<UdpLog> LogList { get; } = new ObservableCollection<UdpLog>();
 
-        public LogListBox()
+        public UdpLogListBox()
         {
             InitializeComponent();
             DataContext = this;
 
-            this.socket.Bind(new IPEndPoint(IPAddress.Loopback, UdpLoggerPort.Value));
-            this.BeginReceiveFrom();
+            this.InitUdpLoggerAsync();
         }
 
-        private void BeginReceiveFrom()
+        private async void InitUdpLoggerAsync()
+        {            
+            this.socket.Bind(new IPEndPoint(IPAddress.Loopback, UdpLoggerPort.Value));
+            while (true)
+            {
+                var log = await this.GetUdpLogAsync();
+                this.LogList.Add(log);
+            }
+        }
+
+        private async Task<UdpLog> GetUdpLogAsync()
         {
             EndPoint remoteEP = new IPEndPoint(IPAddress.Any, 0);
-            this.socket.BeginReceiveFrom(buffer, 0, buffer.Length, SocketFlags.None, ref remoteEP, this.EndReceiveFrom, null);
+            var taskCompletionSource = new TaskCompletionSource<int>();
+            this.socket.BeginReceiveFrom(buffer, 0, buffer.Length, SocketFlags.None, ref remoteEP, this.EndReceiveFrom, taskCompletionSource);
+            var length = await taskCompletionSource.Task;
+
+            var json = Encoding.UTF8.GetString(buffer, 0, length);
+            return Newtonsoft.Json.JsonConvert.DeserializeObject<UdpLog>(json);
         }
 
+
         private void EndReceiveFrom(IAsyncResult ar)
         {
             EndPoint remoteEP = new IPEndPoint(IPAddress.Any, 0);
             var length = this.socket.EndReceiveFrom(ar, ref remoteEP);
-            var json = Encoding.UTF8.GetString(buffer, 0, length);
-            var log = Newtonsoft.Json.JsonConvert.DeserializeObject<UdpLog>(json);
-            this.Dispatcher.Invoke(() => this.LogList.Add(log));
-            this.BeginReceiveFrom();
+            var taskCompletionSource = (TaskCompletionSource<int>)ar.AsyncState;
+            taskCompletionSource.TrySetResult(length);
         }
 
         private void MenuItem_Copy_Click(object sender, System.Windows.RoutedEventArgs e)