fix(urn): Move UrnCoercer into corresponding Urn class

li-utils/src/main/java/com/linkedin/common/urn/CorpGroupUrn.java

@@ -1,5 +1,8 @@
 package com.linkedin.common.urn;
 
+import com.linkedin.data.template.Custom;
+import com.linkedin.data.template.DirectCoercer;
+import com.linkedin.data.template.TemplateOutputCastException;
 import java.net.URISyntaxException;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
@@ -42,4 +45,20 @@ public static CorpGroupUrn createFromUrn(Urn urn) throws URISyntaxException {
   public static CorpGroupUrn deserialize(String rawUrn) throws URISyntaxException {
     return createFromString(rawUrn);
   }
+
+  static {
+    Custom.registerCoercer(new DirectCoercer<CorpGroupUrn>() {
+      public Object coerceInput(CorpGroupUrn object) throws ClassCastException {
+        return object.toString();
+      }
+
+      public CorpGroupUrn coerceOutput(Object object) throws TemplateOutputCastException {
+        try {
+          return CorpGroupUrn.createFromString((String) object);
+        } catch (URISyntaxException e) {
+          throw new TemplateOutputCastException("Invalid URN syntax: " + e.getMessage(), e);
+        }
+      }
+    }, CorpGroupUrn.class);
+  }
 }

li-utils/src/main/java/com/linkedin/common/urn/CorpuserUrn.java

@@ -1,5 +1,8 @@
 package com.linkedin.common.urn;
 
+import com.linkedin.data.template.Custom;
+import com.linkedin.data.template.DirectCoercer;
+import com.linkedin.data.template.TemplateOutputCastException;
 import java.net.URISyntaxException;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
@@ -43,4 +46,21 @@ public static CorpuserUrn createFromUrn(Urn urn) throws URISyntaxException {
   public static CorpuserUrn deserialize(String rawUrn) throws URISyntaxException {
     return createFromString(rawUrn);
   }
+
+  static {
+    Custom.registerCoercer(new DirectCoercer<CorpuserUrn>() {
+      public Object coerceInput(CorpuserUrn object) throws ClassCastException {
+        return object.toString();
+      }
+
+      public CorpuserUrn coerceOutput(Object object) throws TemplateOutputCastException {
+        try {
+          return CorpuserUrn.createFromString((String) object);
+        } catch (URISyntaxException e) {
+          throw new TemplateOutputCastException("Invalid URN syntax: " + e.getMessage(), e);
+        }
+      }
+    }, CorpuserUrn.class);
+  }
+
 }

li-utils/src/main/java/com/linkedin/common/urn/DataPlatformUrn.java

@@ -1,5 +1,9 @@
 package com.linkedin.common.urn;
 
+import com.linkedin.data.template.Custom;
+import com.linkedin.data.template.DirectCoercer;
+import com.linkedin.data.template.TemplateOutputCastException;
+
 import java.net.URISyntaxException;
 
 
@@ -26,4 +30,20 @@ public static DataPlatformUrn createFromString(String rawUrn) throws URISyntaxEx
   public static DataPlatformUrn deserialize(String rawUrn) throws URISyntaxException {
     return createFromString(rawUrn);
   }
+
+  static {
+    Custom.registerCoercer(new DirectCoercer<DataPlatformUrn>() {
+      public Object coerceInput(DataPlatformUrn object) throws ClassCastException {
+        return object.toString();
+      }
+
+      public DataPlatformUrn coerceOutput(Object object) throws TemplateOutputCastException {
+        try {
+          return DataPlatformUrn.createFromString((String) object);
+        } catch (URISyntaxException e) {
+          throw new TemplateOutputCastException("Invalid URN syntax: " + e.getMessage(), e);
+        }
+      }
+    }, DataPlatformUrn.class);
+  }
 }

li-utils/src/main/java/com/linkedin/common/urn/DatasetUrn.java

@@ -1,6 +1,9 @@
 package com.linkedin.common.urn;
 
 import com.linkedin.common.FabricType;
+import com.linkedin.data.template.Custom;
+import com.linkedin.data.template.DirectCoercer;
+import com.linkedin.data.template.TemplateOutputCastException;
 import java.net.URISyntaxException;
 
 import static com.linkedin.common.urn.UrnUtils.*;
@@ -46,4 +49,20 @@ public static DatasetUrn createFromString(String rawUrn) throws URISyntaxExcepti
   public static DatasetUrn deserialize(String rawUrn) throws URISyntaxException {
     return createFromString(rawUrn);
   }
+
+  static {
+    Custom.registerCoercer(new DirectCoercer<DatasetUrn>() {
+      public Object coerceInput(DatasetUrn object) throws ClassCastException {
+        return object.toString();
+      }
+
+      public DatasetUrn coerceOutput(Object object) throws TemplateOutputCastException {
+        try {
+          return DatasetUrn.createFromString((String) object);
+        } catch (URISyntaxException e) {
+          throw new TemplateOutputCastException("Invalid URN syntax: " + e.getMessage(), e);
+        }
+      }
+    }, DatasetUrn.class);
+  }
 }

li-utils/src/main/java/com/linkedin/common/urn/Urn.java

@@ -1,5 +1,9 @@
 package com.linkedin.common.urn;
 
+import com.linkedin.data.template.Custom;
+import com.linkedin.data.template.DirectCoercer;
+import com.linkedin.data.template.TemplateOutputCastException;
+
 import javax.annotation.Nonnull;
 import java.net.URISyntaxException;
 import java.util.regex.Matcher;
@@ -86,4 +90,21 @@ public static boolean isUrn(@Nonnull String urn) {
   public static Urn deserialize(String rawUrn) throws URISyntaxException {
     return createFromString(rawUrn);
   }
+
+  static {
+    Custom.registerCoercer(new DirectCoercer<Urn>() {
+      public Object coerceInput(Urn object) throws ClassCastException {
+        return object.toString();
+      }
+
+      public Urn coerceOutput(Object object) throws TemplateOutputCastException {
+        try {
+          return Urn.createFromString((String) object);
+        } catch (URISyntaxException e) {
+          throw new TemplateOutputCastException("Invalid URN syntax: " + e.getMessage(), e);
+        }
+      }
+
+    }, Urn.class);
+  }
 }

li-utils/src/main/pegasus/com/linkedin/common/CorpGroupUrn.pdsc

@@ -1,8 +1,7 @@
 {
   "ref" : "string",
   "java" : {
-    "class" : "com.linkedin.common.urn.CorpGroupUrn",
-    "coercerClass": "com.linkedin.common.urn.CorpGroupUrnCoercer"
+    "class" : "com.linkedin.common.urn.CorpGroupUrn"
   },
   "name" : "CorpGroupUrn",
   "namespace" : "com.linkedin.common",

li-utils/src/main/pegasus/com/linkedin/common/CorpuserUrn.pdsc

@@ -1,8 +1,7 @@
 {
   "ref" : "string",
   "java" : {
-    "class" : "com.linkedin.common.urn.CorpuserUrn",
-    "coercerClass": "com.linkedin.common.urn.CorpuserUrnCoercer"
+    "class" : "com.linkedin.common.urn.CorpuserUrn"
   },
   "name" : "CorpuserUrn",
   "namespace" : "com.linkedin.common",

li-utils/src/main/pegasus/com/linkedin/common/DataPlatformUrn.pdsc

@@ -1,8 +1,7 @@
 {
   "ref" : "string",
   "java" : {
-    "class" : "com.linkedin.common.urn.DataPlatformUrn",
-    "coercerClass": "com.linkedin.common.urn.DataPlatformUrnCoercer"
+    "class" : "com.linkedin.common.urn.DataPlatformUrn"
   },
   "name" : "DataPlatformUrn",
   "namespace" : "com.linkedin.common",

li-utils/src/main/pegasus/com/linkedin/common/DatasetUrn.pdsc

@@ -1,8 +1,7 @@
 {
   "ref" : "string",
   "java" : {
-    "class" : "com.linkedin.common.urn.DatasetUrn",
-    "coercerClass": "com.linkedin.common.urn.DatasetUrnCoercer"
+    "class" : "com.linkedin.common.urn.DatasetUrn"
   },
   "name" : "DatasetUrn",
   "namespace" : "com.linkedin.common",

li-utils/src/main/pegasus/com/linkedin/common/Urn.pdsc

@@ -4,7 +4,6 @@
   "namespace": "com.linkedin.common",
   "ref": "string",
   "java": {
-    "class": "com.linkedin.common.urn.Urn",
-    "coercerClass": "com.linkedin.common.urn.UrnCoercer"
+    "class": "com.linkedin.common.urn.Urn"
   }
 }

metadata-dao/src/main/java/com/linkedin/metadata/dao/utils/RecordUtils.java

@@ -254,7 +254,7 @@ private static <T> T invokeProtectedMethod(Object object, Method method, Object.
       method.setAccessible(true);
       return (T) method.invoke(object, args);
     } catch (IllegalAccessException | InvocationTargetException e) {
-      throw new RuntimeException(e);
+      throw new RuntimeException(e.getCause());
     } finally {
       method.setAccessible(false);
     }