Added optionalQueryName argument to setQueryParams().

RickWong · Apr 1, 2015 · 99ad8a1 · 99ad8a1
1 parent ce89dde
commit 99ad8a1
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
  "name": "react-transmit",
  "description": "Relay-inspired library based on Promises instead of GraphQL.",
- "version": "2.3.1",
+ "version": "2.3.2",
  "license": "BSD-3",
  "repository": {
  "type": "git",

diff --git a/src/lib/createContainer.js b/src/lib/createContainer.js
@@ -75,16 +75,31 @@ module.exports = function (Component, options) {
  this.setQueryParams({});
  }
  },
- setQueryParams: function (nextParams) {
+ setQueryParams: function (nextParams, optionalQueryName) {
  var _this = this;
 
  setTimeout(function () {
  var state = _this.state || {};
  var props = _this.props || {};
+ var promise;
 
  assign(_this.currentParams, nextParams);
 
- Container.getAllQueries(_this.currentParams).then(function (queryResults) {
+ if (optionalQueryName) {
+ promise = Container.getQuery(
+ optionalQueryName, _this.currentParams
+ ).then(function (queryResult) {
+ var queryResults = {};
+ queryResults[optionalQueryName] = queryResult;
+
+ return queryResults;
+ });
+ }
+ else {
+ promise = Container.getAllQueries(_this.currentParams);
+ }
+
+ promise.then(function (queryResults) {
  try {
  _this.setState(queryResults);
  }