From 55dc940abb02786e47fc01da3aa209d640c1860a Mon Sep 17 00:00:00 2001
From: PMExtra <pm@jubeat.net>
Date: Mon, 30 Mar 2020 17:10:04 +0800
Subject: [PATCH] GetUserProperty extension.

---
 .../Extensions/UserPropertyExtension.cs       | 18 ++++++++++++
 .../MqttApplicationMessage_Tests.cs           | 29 +++++++++++++++++++
 2 files changed, 47 insertions(+)
 create mode 100644 Source/MQTTnet/Extensions/UserPropertyExtension.cs
 create mode 100644 Tests/MQTTnet.Core.Tests/MqttApplicationMessage_Tests.cs

diff --git a/Source/MQTTnet/Extensions/UserPropertyExtension.cs b/Source/MQTTnet/Extensions/UserPropertyExtension.cs
new file mode 100644
index 00000000..273be252
--- /dev/null
+++ b/Source/MQTTnet/Extensions/UserPropertyExtension.cs
@@ -0,0 +1,18 @@
+using System;
+using System.Linq;
+
+namespace MQTTnet.Extensions
+{
+    public static class UserPropertyExtension
+    {
+        public static string GetUserProperty(this MqttApplicationMessage message, string propertyName, StringComparison comparisonType = StringComparison.OrdinalIgnoreCase)
+        {
+            return message.UserProperties.SingleOrDefault(up => up.Name.Equals(propertyName, comparisonType))?.Value;
+        }
+
+        public static T GetUserProperty<T>(this MqttApplicationMessage message, string propertyName, StringComparison comparisonType = StringComparison.OrdinalIgnoreCase)
+        {
+            return (T) Convert.ChangeType(GetUserProperty(message, propertyName, comparisonType), typeof(T));
+        }
+    }
+}
diff --git a/Tests/MQTTnet.Core.Tests/MqttApplicationMessage_Tests.cs b/Tests/MQTTnet.Core.Tests/MqttApplicationMessage_Tests.cs
new file mode 100644
index 00000000..073e4c43
--- /dev/null
+++ b/Tests/MQTTnet.Core.Tests/MqttApplicationMessage_Tests.cs
@@ -0,0 +1,29 @@
+using System.Collections.Generic;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using MQTTnet.Extensions;
+using MQTTnet.Packets;
+
+namespace MQTTnet.Tests
+{
+    [TestClass]
+    public class MqttApplicationMessage_Tests
+    {
+        [TestMethod]
+        public void GetUserProperty_Test()
+        {
+            var message = new MqttApplicationMessage
+            {
+                UserProperties = new List<MqttUserProperty>
+                {
+                    new MqttUserProperty("foo", "bar"),
+                    new MqttUserProperty("value", "1011"),
+                    new MqttUserProperty("CASE", "insensitive")
+                }
+            };
+
+            Assert.AreEqual("bar", message.GetUserProperty("foo"));
+            Assert.AreEqual(1011, message.GetUserProperty<int>("value"));
+            Assert.AreEqual("insensitive", message.GetUserProperty("case"));
+        }
+    }
+}