writeext feature

.github/workflows/linux-cross.yaml

@@ -27,6 +27,7 @@ jobs:
         CFLAGS: "-O2 -fno-plt"
         LDFLAGS: "-O2 -flto -Wl,--as-needed"
         CARGO_UNSTABLE_SPARSE_REGISTRY: "true"
+        ORJSON_ENABLE_WRITEEXT: "1"
       with:
         maturin-version: v0.14.14
         target: ${{ matrix.target.arch }}

.github/workflows/manylinux2014.yaml

@@ -19,6 +19,7 @@ jobs:
       CFLAGS: "-O2 -fno-plt"
       LDFLAGS: "-O2 -flto -Wl,--as-needed"
       CARGO_UNSTABLE_SPARSE_REGISTRY: "true"
+      ORJSON_ENABLE_WRITEEXT: "1"
     container:
       image: quay.io/pypa/manylinux2014_x86_64:latest
       options: --user 0

.github/workflows/manylinux_2_28.yaml

@@ -20,6 +20,7 @@ jobs:
       LDFLAGS: "-O2 -flto=thin -fuse-ld=lld -Wl,--as-needed"
       RUSTFLAGS: "-C linker=clang -C link-arg=-fuse-ld=lld"
       CARGO_UNSTABLE_SPARSE_REGISTRY: "true"
+      ORJSON_ENABLE_WRITEEXT: "1"
     container:
       image: quay.io/pypa/manylinux_2_28_x86_64:latest
       options: --user 0

build.rs

@@ -1,13 +1,17 @@
 // SPDX-License-Identifier: (Apache-2.0 OR MIT)
 
+use std::env;
+
 fn main() {
     println!("cargo:rerun-if-changed=build.rs");
     println!("cargo:rerun-if-changed=include/yyjson/*");
     println!("cargo:rerun-if-env-changed=CC");
     println!("cargo:rerun-if-env-changed=CFLAGS");
     println!("cargo:rerun-if-env-changed=LDFLAGS");
     println!("cargo:rerun-if-env-changed=RUSTFLAGS");
+    println!("cargo:rerun-if-env-changed=ORJSON_DISABLE_WRITEEXT");
     println!("cargo:rerun-if-env-changed=ORJSON_DISABLE_YYJSON");
+    println!("cargo:rerun-if-env-changed=ORJSON_ENABLE_WRITEEXT");
 
     let py_cfg = pyo3_build_config::get();
     py_cfg.emit_pyo3_cfgs();
@@ -20,8 +24,16 @@ fn main() {
         println!("cargo:rustc-cfg=feature=\"optimize\"");
     }
 
-    if std::env::var("ORJSON_DISABLE_YYJSON").is_ok() {
-        if std::env::var("CARGO_FEATURE_YYJSON").is_ok() {
+    if env::var("ORJSON_DISABLE_WRITEEXT").is_ok() {
+    } else if env::var("ORJSON_ENABLE_WRITEEXT").is_ok()
+        || env::var("CARGO_CFG_TARGET_OS").unwrap() == "macos"
+        || env::var("CARGO_CFG_TARGET_ENV").unwrap() == "gnu"
+    {
+        println!("cargo:rustc-cfg=feature=\"writeext\"");
+    }
+
+    if env::var("ORJSON_DISABLE_YYJSON").is_ok() {
+        if env::var("CARGO_FEATURE_YYJSON").is_ok() {
             panic!("ORJSON_DISABLE_YYJSON and --features=yyjson both enabled.")
         }
     } else {
@@ -36,7 +48,7 @@ fn main() {
                 println!("cargo:rustc-cfg=feature=\"yyjson\"");
             }
             Err(_) => {
-                if std::env::var("CARGO_FEATURE_YYJSON").is_ok() {
+                if env::var("CARGO_FEATURE_YYJSON").is_ok() {
                     panic!("yyjson was enabled but the build failed. To build with a different backend do not specify the feature.")
                 }
             }

ci/azure-macos.yml

@@ -20,6 +20,7 @@ steps:
     LDFLAGS: "-O2 -flto=thin -fuse-ld=lld -Wl,--as-needed"
     RUSTFLAGS: "-C linker=clang -C target-feature=+sse4.2"
     CARGO_UNSTABLE_SPARSE_REGISTRY: "true"
+    ORJSON_ENABLE_WRITEEXT: "1"
   displayName: build
 - bash: pip install target/wheels/orjson*.whl
   displayName: install
@@ -37,6 +38,7 @@ steps:
     LDFLAGS: "-O2 -flto=thin -fuse-ld=lld -Wl,--as-needed"
     RUSTFLAGS: "-C linker=clang"
     CARGO_UNSTABLE_SPARSE_REGISTRY: "true"
+    ORJSON_ENABLE_WRITEEXT: "1"
   displayName: build universal2
 - bash: pip install --force-reinstall target/wheels/orjson*universal2.whl
   displayName: install universal2

src/serialize/mod.rs

@@ -18,6 +18,7 @@ mod tuple;
 mod uuid;
 mod writer;
 
+#[cfg(feature = "writeext")]
 mod json;
 
 pub use serializer::serialize;

src/serialize/serializer.rs

@@ -21,6 +21,10 @@ use serde::ser::{Serialize, SerializeMap, Serializer};
 use std::io::Write;
 use std::ptr::NonNull;
 
+#[cfg(not(feature = "writeext"))]
+use serde_json::{to_writer, to_writer_pretty};
+
+#[cfg(feature = "writeext")]
 use crate::serialize::json::{to_writer, to_writer_pretty};
 
 pub const RECURSION_LIMIT: u8 = 255;