ModuleObjcMessageTrace: also report method calls from protocol types

Author: nkcsgexi

lib/FrontendTool/LoadedModuleTrace.cpp

@@ -849,6 +849,8 @@ class ObjcMethodReferenceCollector: public SourceEntityWalker {
       return "interface_type";
     if (isa<clang::ObjCCategoryDecl>(clangD))
       return "category_type";
+    if (isa<clang::ObjCProtocolDecl>(clangD))
+      return "protocol_type";
     return "type";
   }
 public:

test/IDE/objc_send_collector.swift

@@ -7,16 +7,18 @@
 
 import Foo
 
-public func testProperties(_ x: FooClassBase) {
+public func testProperties(_ x: FooClassBase, _ y: FooProtocolBase) {
   _ = x.fooBaseInstanceFunc0()
   x.fooBaseInstanceFunc1(1.2)
   _ = FooClassBase.fooBaseClassFunc0()
+  y.fooProtoFunc()
 }
 
 // CHECK-DAG: "instance_method": "fooBaseInstanceFunc0"
 // CHECK-DAG: "instance_method": "fooBaseInstanceFunc1:"
 // CHECK-DAG: "class_method": "fooBaseClassFunc0"
 // CHECK-DAG: "interface_type": "FooClassBase"
+// CHECK-DAG: "protocol_type": "FooProtocolBase"
 // CHECK-DAG: "declared_at": "SOURCE_DIR/test/IDE/Inputs/mock-sdk/Foo.framework/Headers/Foo.h
 // CHECK-DAG: "referenced_at_file_id": 1
 // CHECK-DAG: "file_id": 1,