frodovdr
/
node-red
mirror of https://github.com/node-red/node-red.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Allow multiple keys and values to be passed to `set`

This commit is contained in:
HirokiUchikawa 2018-07-12 19:19:55 +09:00
parent 051c147b41
commit 6b2f5fbb19
2 changed files with 63 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
red/runtime/nodes/context/index.js
View File

@ -268,7 +268,29 @@ function createContext(id,seed) {
}
context = getContextStorage(storage);
}
context.set(scope, key, value, callback);
if (!Array.isArray(key) || !Array.isArray(value) || key.length !== value.length) {
context.set(scope, key, value, callback);
} else {
// If key and value are Array and each length is same, set each key-value pair.
var index = 0;
var cb = function(err, v) {
if (err) {
if (callback) {
callback(err);
}
} else {
index++;
if (index === key.length) {
if (callback) {
callback(null);
}
} else {
context.set(scope, key[index], value[index], cb);
}
}
};
context.set(scope, key[index], value[index], cb);
}
};
obj.keys = function(storage, callback) {
var context;

40
test/red/runtime/nodes/context/index_spec.js
View File

@ -615,6 +615,46 @@ describe('context', function() {
});
}).catch(function(err){ done(err); });
});
it('should store multiple properties if key and value are arrays', function(done) {
Context.init({contextStorage:memoryStorage});
Context.load().then(function(){
var context = Context.get("1","flow");
context.set(["foo1","foo2","foo3"], ["bar1","bar2","bar3"], "memory", function(err){
if (err) {
done(err);
} else {
context.get(["foo1","foo2","foo3"], "memory", function(err,foo1,foo2,foo3){
if (err) {
done(err);
} else {
foo1.should.be.equal("bar1");
foo2.should.be.equal("bar2");
foo3.should.be.equal("bar3");
done();
}
});
}
});
});
});
it('should return an error if an error occurs in storing multiple values', function(done) {
Context.init({contextStorage:contextStorage});
stubSet.onFirstCall().callsArgWith(3, null);
stubSet.onSecondCall().callsArgWith(3, "error2");
stubSet.onThirdCall().callsArgWith(3, null);
Context.load().then(function(){
var context = Context.get("1","flow");
context.set(["foo1","foo2","foo3"], ["bar1","bar2","bar3"], "memory", function(err){
if (err === "error2") {
done();
} else {
done("An error occurred");
}
});
}).catch(function(err){ done(err); });
});
});
describe('delete context',function(){