Skip to content

Commit 1e1e527

Browse files
committed
chore: use auto_segment instead of segment
1 parent 898065c commit 1e1e527

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

agent/lib_mongodb.c

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,9 @@ NR_PHP_WRAPPER_END
179179

180180
NR_PHP_WRAPPER(nr_mongodb_operation_before) {
181181
(void)wraprec;
182-
nr_segment_start(NRPRG(txn), NULL, NULL);
182+
nr_segment_t* segment;
183+
segment = nr_segment_start(NRPRG(txn), NULL, NULL);
184+
segment->wraprec = auto_segment->wraprec;
183185
}
184186
NR_PHP_WRAPPER_END
185187

@@ -189,7 +191,6 @@ NR_PHP_WRAPPER(nr_mongodb_operation_after) {
189191
zval* database = NULL;
190192
zval* server = NULL;
191193
zval* this_var = NULL;
192-
nr_segment_t* segment = NULL;
193194
nr_datastore_instance_t instance = {
194195
.host = NULL,
195196
.port_path_or_id = NULL,
@@ -234,8 +235,7 @@ NR_PHP_WRAPPER(nr_mongodb_operation_after) {
234235
nr_mongodb_get_host_and_port_path_or_id(server, &instance.host,
235236
&instance.port_path_or_id);
236237

237-
segment = nr_txn_get_current_segment(NRPRG(txn), NULL);
238-
nr_segment_datastore_end(&segment, &params);
238+
nr_segment_datastore_end(&auto_segment, &params);
239239

240240
leave:
241241
nr_php_arg_release(&server);

0 commit comments

Comments
 (0)