CODEWITHSUNDEEP
javascriptlodash
Andreas Köberle
Andreas Köberle

Reputation: 110892

Create chain in lodash with custom functions

Is there way to get my own custom function in a chain of lodash. So for example like this:

var l = [1,2,3]
var add = function(a, b){return a+b}

var r =_.chain(l).find(function(a){return a>1}).add(5).value()

=>r = 7

Upvotes: 10

Views: 9349

Answers (4)

michele thefirst
michele thefirst

Reputation: 1

Maybe it's too late but _.tap is another chance

Upvotes: 0

Aurelio
Aurelio

Reputation: 25772

Another option is to just drop chaining and leverage function composition through _.flow.

From the DOCS:

[Flow] Creates a function that returns the result of invoking the given functions with the this binding of the created function, where each successive invocation is supplied the return value of the previous.

This means that every function inside Flow will receive as input the output of the previous one. In practice this means we are not limited to using only Lodash API methods, but we can mix and match whatever function we fancy, as long as the next one is able to handle that return value.

var l = [1,2,3]
var add = _.curry((a, b) => a + b);

_.flow(
  _.find(a => a > 1),
  add(5),
)(l);

// => 7

NB - This example is using the Functional version of Lodash, if you don't want or can't use that one you can still achieve the same result, check my other answer to another question about Lodash.

Upvotes: 6

Andreas Köberle
Andreas Köberle

Reputation: 110892

After @stride anwswer I came up with a more generic solution using _.mixin:

function add(a, b, c) {
  return a + b + c
}

function sub(a, b, c) {
  return a - b - c
}

_.mixin({
  run: function (v, f) {
    var args = Array.prototype.slice.call(arguments, 2)
    args.unshift(v)
    return f.apply(this, args)
  }
})
var r = _.chain(1).run(add, 1, 1).run(sub, 2, 2).value()

console.log(r) ->  -1
1 + 1 + 1 - 2 - 2 = -1

http://jsbin.com/iyEhaMa/1/

After all I wonder why this not a build in function in lodash.

Upvotes: 6

stride
stride

Reputation: 1951

What you look for is a way to extend the lodash prototype. It so nicely turns out that you can do it easily with a mixin utility function. Check here the docs: http://lodash.com/docs#mixin

In your example it will look like:

var l = [1,2,3];
var  add = function(a, b){return a+b}


_.mixin({
    add: add 
});


var r =_.chain(l).find(function(a){return a>1}).add(5).value()
console.log(r); ==> 7

and here is live sample on fiddle: http://jsfiddle.net/g2A9C/

Upvotes: 17

Related Questions