Sora オブジェクトに tp_traverse をつける（リークするけど）

Author: melpon

src/sora.cpp

@@ -208,6 +208,14 @@ std::shared_ptr<SoraConnection> Sora::CreateConnection(
   if (video_frame_transformer) {
     conn->SetVideoSenderFrameTransformer(video_frame_transformer);
   }
+
+  weak_connections_.erase(
+      std::remove_if(
+          weak_connections_.begin(), weak_connections_.end(),
+          [](std::weak_ptr<SoraConnection> w) { return w.expired(); }),
+      weak_connections_.end());
+  weak_connections_.push_back(conn);
+
   return conn;
 }
 

src/sora.h

@@ -159,6 +159,8 @@ class Sora : public DisposePublisher {
    */
   SoraVideoSource* CreateVideoSource();
 
+  std::vector<std::weak_ptr<SoraConnection>> weak_connections_;
+
  private:
   /**
    * Python で渡された値を boost::json::value に変換します。

src/sora_sdk_ext.cpp

@@ -288,6 +288,38 @@ PyType_Slot connection_slots[] = {
     {Py_tp_clear, (void*)connection_tp_clear},
     {0, nullptr}};
 
+int sora_tp_traverse(PyObject* self, visitproc visit, void* arg) {
+  if (!nb::inst_ready(self)) {
+    return 0;
+  }
+
+  Sora* sora = nb::inst_ptr<Sora>(self);
+  for (auto wc : sora->weak_connections_) {
+    auto conn = wc.lock();
+    if (conn) {
+      nb::object conn_obj = nb::find(conn);
+      Py_VISIT(conn_obj.ptr());
+    }
+  }
+
+  return 0;
+}
+
+int sora_tp_clear(PyObject* self) {
+  if (!nb::inst_ready(self)) {
+    return 0;
+  }
+
+  Sora* sora = nb::inst_ptr<Sora>(self);
+  sora->weak_connections_.clear();
+
+  return 0;
+}
+
+PyType_Slot sora_slots[] = {{Py_tp_traverse, (void*)sora_tp_traverse},
+                            {Py_tp_clear, (void*)sora_tp_clear},
+                            {0, nullptr}};
+
 /**
  * Python で利用するすべてのクラスと定数は以下のように定義しなければならない
  */
@@ -529,7 +561,7 @@ NB_MODULE(sora_sdk_ext, m) {
       .def("__del__", &SoraVideoFrameTransformer::Del)
       .def_rw("on_transform", &SoraVideoFrameTransformer::on_transform_);
 
-  nb::class_<Sora>(m, "Sora")
+  nb::class_<Sora>(m, "Sora", nb::type_slots(sora_slots))
       .def(nb::init<std::optional<bool>, std::optional<std::string>>(),
            "use_hardware_encoder"_a = nb::none(), "openh264"_a = nb::none())
       .def("create_connection", &Sora::CreateConnection, "signaling_urls"_a,

test_with_llvm.py

@@ -8,7 +8,9 @@ def test(debugger, command, result, internal_dict):
     debugger.HandleCommand("settings set target.process.follow-fork-mode child")
 
     target = debugger.CreateTargetWithFileAndArch("uv", lldb.LLDB_ARCH_DEFAULT)
-    process = target.LaunchSimple(["run", "pytest", "tests", "-s"], None, None)
+    process = target.LaunchSimple(
+        ["run", "pytest", "tests/test_sora_disconnect.py", "-s"], None, None
+    )
 
     if not process:
         print("Error: could not launch process")