core/pkgs/by-name/je/jemalloc/rust.nix

{
  lib,
  stdenv,
  jemalloc,
  writeText,

  unprefixed ? false,
}:

let
  # On some platforms the unprefixed feature will be ignored:
  # https://github.com/tikv/jemallocator/blob/ab0676d77e81268cd09b059260c75b38dbef2d51/jemalloc-sys/src/env.rs
  unprefixed' =
    unprefixed
    && !stdenv.hostPlatform.isMusl
    && !stdenv.hostPlatform.isDarwin
    && !stdenv.hostPlatform.isAndroid;

in
jemalloc.overrideAttrs (oldAttrs: {
  configureFlags =
    oldAttrs.configureFlags
    ++ [ "--with-private-namespace=_rjem_" ]
    ++ lib.optionals (!unprefixed') [ "--with-jemalloc-prefix=_rjem_" ];

  setupHook = writeText "setup-hook.sh" ''
    export JEMALLOC_OVERRIDE="@out@/lib/libjemalloc${stdenv.hostPlatform.extensions.library}"
  '';
})