容易替换URL中查询字符串的参数值的代码

简单替换URL中查询字符串的参数值的代码
刚有个需求是要把URL中query部分的一个指定参数的值替换为新的，如果该参数不存在的话就将新的键值对加到query的末尾。前提是：
1、输入的字符串可能是空的，如果不为空则保证每个参数都是key=value形式的，其中key与等号一定存在；
2、输入的key不会带有奇怪的字符……（就避开了JavaScript的正则表达式没有提供escape功能的麻烦）

想了一下觉得用正则表达式做也短不到哪里去，就先给了一个普通版本：

function setQueryParamValue(queryString, key, newValue) {
  var newParam = key + '=' + newValue;
  if (!queryString) return newParam;
  
  var replaced = false;
  var params = queryString.split('&');
  for (var i = 0; i < params.length; i++) {
    if (params[i].split('=')[0] == key) {
      params[i] = newParam;
      replaced = true;
    }
  }
  if (replaced) return params.join('&');
  return queryString + '&' + newParam;
}

然后用正则表达式加上逻辑运算表达式再实现了一遍：

function setQueryParamValue(queryString, key, newValue) {
  var replaced = false;
  var newParam = key + '=' + newValue;
  var result = queryString.replace(new RegExp('(^|&)' + key + '=(.*?)(?=$|&)', 'g'), function (s, p1, p2) {
    replaced = true;
    return p1 + newParam;
  });
  return replaced && result || queryString && (queryString + '&' + newParam) || newParam;
}

（本来要安全的话这里应该对key做一次escape，不过有前提条件保证就不管了）
从代码的字符数量来说果然没短多少……核心替换的逻辑确实是“声明式”了一些，不过不知道多少人是比较喜欢后面这种写法的呢？

在同样的前提条件下要抽取参数的值就简单多了：

function getQueryParamValue(queryString, key) {
  var result = queryString.match(new RegExp('(?:^|&)' + key + '=(.*?)(?=$|&)'));
  return result && result[1];
}

这个我就不用非正则表达式的方式去做了。

要是有更好的办法用正则表达式来实现这个功能拜托告诉我哦～～