Fixed segfault with RoutingModel

ankane · Sep 17, 2024 · 0d55326 · 0d55326
1 parent fc3d85c
commit 0d55326
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ext/or-tools/routing.cpp b/ext/or-tools/routing.cpp
@@ -309,7 +309,7 @@ void init_routing(Rice::Module& m) {
             return Rice::detail::From_Ruby<int64_t>().convert(callback.call("call", from_index));
           }
         );
-      })
+      }, Rice::Arg("callback").keepAlive())
     .define_method("register_transit_matrix", &RoutingModel::RegisterTransitMatrix)
     .define_method(
       "register_transit_callback",
@@ -319,7 +319,7 @@ void init_routing(Rice::Module& m) {
             return Rice::detail::From_Ruby<int64_t>().convert(callback.call("call", from_index, to_index));
           }
         );
-      })
+      }, Rice::Arg("callback").keepAlive())
     .define_method("add_dimension", &RoutingModel::AddDimension)
     .define_method("add_dimension_with_vehicle_transits", &RoutingModel::AddDimensionWithVehicleTransits)
     .define_method("add_dimension_with_vehicle_capacity", &RoutingModel::AddDimensionWithVehicleCapacity)