Experimenting with rb_str_reserve.

ioquatix · Mar 20, 2017 · e6b164e · e6b164e
1 parent 335c54e
commit e6b164e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/ext/trenni/extconf.rb b/ext/trenni/extconf.rb
@@ -6,6 +6,7 @@
 # Needed by Ruby 2.1
 have_func("rb_sym2str")
 have_func("rb_str_cat_cstr")
+have_func("rb_str_reserve")
 
 gem_name = File.basename(__dir__)
 extension_name = 'trenni'

diff --git a/ext/trenni/trenni.h b/ext/trenni/trenni.h
@@ -16,6 +16,15 @@
 #define RB_IMMEDIATE_P IMMEDIATE_P
 #endif
 
+#ifndef HAVE_RB_STR_RESERVE
+inline VALUE rb_str_reserve(VALUE string, long extra) {
+	long actual = RSTRING_LEN(string);
+	rb_str_resize(string, actual + extra);
+	rb_str_set_len(string, actual);
+	return string;
+}
+#endif
+
 extern VALUE rb_Trenni, rb_Trenni_Markup, rb_Trenni_Tag, rb_Trenni_MarkupString, rb_Trenni_MarkupString_EMPTY, rb_Trenni_Native, rb_Trenni_ParseError;
 
 extern ID id_cdata, id_open_tag_begin, id_open_tag_end, id_attribute, id_close_tag, id_text, id_doctype, id_comment, id_instruction, id_read, id_expression, id_key_get, id_new, id_name, id_attributes, id_closed, id_to_s, id_is_a;