Feature/issue 7350 more ddl telemetry (#7351) (#905)

* enhancement ddl telemetry * clang-format
timeplus-io · Feb 20, 2025 · 0e9cc36 · 0e9cc36
1 parent 008c656
commit 0e9cc36
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 12 deletions.
diff --git a/src/Interpreters/CollectCreateTelemetry.cpp b/src/Interpreters/CollectCreateTelemetry.cpp
@@ -68,9 +68,18 @@ void collectCreateTelemetry(const StoragePtr & storage, ContextPtr context)
             builder.isExternalTarget(!materialized_view->getExternalTargetTableID().empty());
 
             if (auto target = materialized_view->tryGetTargetTable())
-                builder.withTargetStorageType(target->getName());
+            {
+                if (auto * target_external_stream = target->as<StorageExternalStream>())
+                    builder.withTargetStorageType(target_external_stream->getType());
+                else if (auto * target_external_table = target->as<StorageExternalTable>())
+                    builder.withTargetStorageType(target_external_table->getType());
+                else
+                    builder.withTargetStorageType(target->getName());
+            }
             else
+            {
                 builder.withTargetStorageType("Stream");
+            }
         });
     }
     else if (auto random_stream = storage->as<StorageRandom>())

diff --git a/src/Interpreters/TelemetryElement.h b/src/Interpreters/TelemetryElement.h
@@ -51,6 +51,7 @@ class TelemetryElementBuilderBase
         getObject()->set("properties", properties);
 
         withServerUUID();
+        withEdition();
     }
 
     std::shared_ptr<TelemetryElement> build() { return std::move(element); }
@@ -73,6 +74,8 @@ class TelemetryElementBuilderBase
         setProperty("server_id", server_uuid_str);
     }
 
+    void withEdition() { setProperty("edition", EDITION); }
+
 private:
     std::shared_ptr<TelemetryElement> element;
     Poco::JSON::Object::Ptr properties;
@@ -84,11 +87,7 @@ class TelemetryStatsElementBuilder final : public TelemetryElementBuilderBase
 public:
     using Self = TelemetryStatsElementBuilder;
 
-    TelemetryStatsElementBuilder() : TelemetryElementBuilderBase("proton_ping")
-    {
-        setEdition();
-        setVersion();
-    }
+    TelemetryStatsElementBuilder() : TelemetryElementBuilderBase("proton_ping") { setVersion(); }
 
     Self & useCPU(unsigned int cpu_)
     {
@@ -163,12 +162,6 @@ class TelemetryStatsElementBuilder final : public TelemetryElementBuilderBase
     }
 
 private:
-    Self & setEdition()
-    {
-        setProperty("edition", EDITION);
-        return *this;
-    }
-
     Self & setVersion()
     {
         setProperty("version", VERSION_STRING);