about summary refs log tree commit diff
path: root/src/libexpr
diff options
context:
space:
mode:
authorEelco Dolstra <eelco.dolstra@logicblox.com>2015-07-24T00·31+0200
committerEelco Dolstra <eelco.dolstra@logicblox.com>2015-07-24T00·38+0200
commit2e8fd4c5cdeb59ae2bd403aa5069993303e8388e (patch)
treef9d51f5420dcd986b7f581324e068ea9a822c4a1 /src/libexpr
parentcb4320c1a0808fe88084d391341da86123b36e6a (diff)
Add concatStringsSep as a primop
This fixes the quadratic behaviour of concatStrings/concatStringsSep
in Nixpkgs.
Diffstat (limited to 'src/libexpr')
-rw-r--r--src/libexpr/primops.cc21
1 files changed, 21 insertions, 0 deletions
diff --git a/src/libexpr/primops.cc b/src/libexpr/primops.cc
index 7a4aad3cd57b..c320b4c39711 100644
--- a/src/libexpr/primops.cc
+++ b/src/libexpr/primops.cc
@@ -1514,6 +1514,26 @@ static void prim_match(EvalState & state, const Pos & pos, Value * * args, Value
 }
 
 
+static void prim_concatStringSep(EvalState & state, const Pos & pos, Value * * args, Value & v)
+{
+    PathSet context;
+
+    auto sep = state.forceString(*args[0], context, pos);
+    state.forceList(*args[1]);
+
+    string res;
+    res.reserve((args[1]->listSize() + 32) * sep.size());
+    bool first = true;
+
+    for (unsigned int n = 0; n < args[1]->listSize(); ++n) {
+        if (first) first = false; else res += sep;
+        res += state.coerceToString(pos, *args[1]->listElems()[n], context);
+    }
+
+    mkString(v, res, context);
+}
+
+
 /*************************************************************
  * Versions
  *************************************************************/
@@ -1720,6 +1740,7 @@ void EvalState::createBaseEnv()
     addPrimOp("__unsafeDiscardOutputDependency", 1, prim_unsafeDiscardOutputDependency);
     addPrimOp("__hashString", 2, prim_hashString);
     addPrimOp("__match", 2, prim_match);
+    addPrimOp("__concatStringsSep", 2, prim_concatStringSep);
 
     // Versions
     addPrimOp("__parseDrvName", 1, prim_parseDrvName);