rdbms.vm.js 7.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288
  1. // Licensed to Cloudera, Inc. under one
  2. // or more contributor license agreements. See the NOTICE file
  3. // distributed with this work for additional information
  4. // regarding copyright ownership. Cloudera, Inc. licenses this file
  5. // to you under the Apache License, Version 2.0 (the
  6. // "License"); you may not use this file except in compliance
  7. // with the License. You may obtain a copy of the License at
  8. //
  9. // http://www.apache.org/licenses/LICENSE-2.0
  10. //
  11. // Unless required by applicable law or agreed to in writing, software
  12. // distributed under the License is distributed on an "AS IS" BASIS,
  13. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14. // See the License for the specific language governing permissions and
  15. // limitations under the License.
  16. function RdbmsViewModel() {
  17. var self = this;
  18. self.servers = ko.observableArray();
  19. self.selectedServer = ko.observable(0);
  20. self.databases = ko.observableArray();
  21. self.selectedDatabase = ko.observable(0);
  22. self.query = ko.mapping.fromJS({
  23. 'id': -1,
  24. 'query': '',
  25. 'name': null,
  26. 'description': null,
  27. 'errors': []
  28. });
  29. self.rows = ko.observableArray();
  30. self.columns = ko.observableArray();
  31. self.resultsEmpty = ko.observable(false);
  32. self.isExecuting = ko.observable(false);
  33. self.server = ko.computed({
  34. 'read': function() {
  35. if (self.servers().length > 0) {
  36. return self.servers()[self.selectedServer()];
  37. } else{
  38. return null;
  39. }
  40. },
  41. 'write': function(value) {
  42. var filtered = $.each(self.servers(), function(index, server) {
  43. if (server.name() == value) {
  44. self.selectedServer(index);
  45. }
  46. });
  47. }
  48. });
  49. self.database = ko.computed({
  50. 'read': function() {
  51. if (self.databases()) {
  52. return self.databases()[self.selectedDatabase()];
  53. } else{
  54. return "";
  55. }
  56. },
  57. 'write': function(value) {
  58. self.selectedDatabase(self.databases.indexOf(value));
  59. }
  60. });
  61. self.getFirstDatabase = function() {
  62. if (self.databases()) {
  63. return self.databases()[0];
  64. } else {
  65. return null;
  66. }
  67. };
  68. self.getServerNiceName = function(value) {
  69. return self.servers[value];
  70. };
  71. self.updateResults = function(results) {
  72. var rows = [];
  73. self.columns.removeAll(); // Needed for datatables to refresh properly.
  74. self.columns(results.columns);
  75. self.rows.removeAll();
  76. self.rows(results.rows);
  77. };
  78. self.updateServers = function(servers) {
  79. var newServers = [];
  80. $.each(servers, function(name, nice_name) {
  81. newServers.push({
  82. 'name': ko.observable(name),
  83. 'nice_name': ko.observable(nice_name)
  84. });
  85. });
  86. self.servers(newServers);
  87. var last = $.totalStorage('hueRdbmsLastServer') || ((newServers[0].length > 0) ? newServers[0].name() : null);
  88. if (last) {
  89. self.server(last);
  90. $.each(self.servers(), function(index, server) {
  91. if (server.name() == last) {
  92. self.chosenServer(server);
  93. }
  94. });
  95. }
  96. $(document).trigger("update.chosen");
  97. };
  98. self.updateDatabases = function(databases) {
  99. self.databases(databases);
  100. var key = 'hueRdbmsLastDatabase-' + self.server().name();
  101. var last = $.totalStorage(key) || ((databases.length > 0) ? databases[0] : null);
  102. if (last) {
  103. self.database(last);
  104. self.chosenDatabase(last);
  105. }
  106. $(document).trigger("update.chosen");
  107. };
  108. self.updateQuery = function(design) {
  109. self.query.query(design.query);
  110. self.query.id(design.id);
  111. self.query.name(design.name);
  112. self.query.description(design.desc);
  113. self.database(design.database);
  114. self.server(design.server);
  115. };
  116. self.chosenServer = ko.observable();
  117. self.initialChoose = true;
  118. self.chosenServer.subscribe(function(value){
  119. $.each(self.servers(), function(index, server) {
  120. if (server.name() == value.name()) {
  121. self.selectedServer(index);
  122. }
  123. });
  124. if (self.initialChoose){
  125. self.initialChoose = false;
  126. }
  127. else {
  128. $.totalStorage('hueRdbmsLastServer', self.server().name());
  129. }
  130. self.fetchDatabases();
  131. });
  132. self.chosenDatabase = ko.observable();
  133. var error_fn = function(jqXHR, status, errorThrown) {
  134. self.isExecuting(false);
  135. try {
  136. $(document).trigger('server.error', $.parseJSON(jqXHR.responseText));
  137. } catch(e) {
  138. $(document).trigger('server.unmanageable_error', jqXHR.responseText);
  139. }
  140. };
  141. self.explainQuery = function() {
  142. var data = ko.mapping.toJS(self.query);
  143. data.database = self.database();
  144. data.server = self.server().name();
  145. var request = {
  146. url: '/rdbms/api/explain/',
  147. dataType: 'json',
  148. type: 'POST',
  149. success: function(data) {
  150. self.query.errors.removeAll();
  151. if (data.status === 0) {
  152. $(document).trigger('explain.query', data);
  153. self.updateResults(data.results);
  154. self.query.id(data.design);
  155. self.resultsEmpty(data.results.rows.length === 0);
  156. $(document).trigger('explained.query', data);
  157. } else {
  158. self.query.errors.push(data.message);
  159. }
  160. },
  161. error: error_fn,
  162. data: data
  163. };
  164. $.ajax(request);
  165. };
  166. self.fetchQuery = function(id) {
  167. var _id = id || self.query.id();
  168. if (_id && _id != -1) {
  169. var request = {
  170. url: '/rdbms/api/query/' + _id + '/get',
  171. dataType: 'json',
  172. type: 'GET',
  173. success: function(data) {
  174. self.updateQuery(data.design);
  175. },
  176. error: error_fn
  177. };
  178. $.ajax(request);
  179. }
  180. };
  181. self.saveQuery = function() {
  182. var self = this;
  183. if (self.query.query() && self.query.name()) {
  184. var data = ko.mapping.toJS(self.query);
  185. data['desc'] = data['description'];
  186. data['server'] = self.server().name();
  187. data['database'] = self.database();
  188. var url = '/rdbms/api/query/';
  189. if (self.query.id() && self.query.id() != -1) {
  190. url += self.query.id() + '/';
  191. }
  192. var request = {
  193. url: url,
  194. dataType: 'json',
  195. type: 'POST',
  196. success: function(data) {
  197. $(document).trigger('saved.query', data);
  198. },
  199. error: function() {
  200. $(document).trigger('error.query');
  201. },
  202. data: data
  203. };
  204. $.ajax(request);
  205. }
  206. };
  207. self.executeQuery = function() {
  208. $(document).trigger('start_execution.query');
  209. self.isExecuting(true);
  210. var data = ko.mapping.toJS(self.query);
  211. data.database = self.database();
  212. data.server = self.server().name();
  213. var request = {
  214. url: '/rdbms/api/execute/',
  215. dataType: 'json',
  216. type: 'POST',
  217. success: function(data) {
  218. self.query.errors.removeAll();
  219. if (data.status === 0) {
  220. $(document).trigger('execute.query', data);
  221. self.updateResults(data.results);
  222. self.query.id(data.design);
  223. self.resultsEmpty(data.results.rows.length === 0);
  224. self.isExecuting(false);
  225. $(document).trigger('executed.query', data);
  226. } else {
  227. self.isExecuting(false);
  228. self.query.errors.push(data.message);
  229. }
  230. },
  231. error: error_fn,
  232. data: data
  233. };
  234. $.ajax(request);
  235. };
  236. self.fetchServers = function() {
  237. var request = {
  238. url: '/rdbms/api/servers/',
  239. dataType: 'json',
  240. type: 'GET',
  241. success: function(data) {
  242. self.updateServers(data.servers);
  243. self.fetchDatabases();
  244. },
  245. error: error_fn
  246. };
  247. $.ajax(request);
  248. };
  249. self.fetchDatabases = function() {
  250. if (self.server()) {
  251. var request = {
  252. url: '/rdbms/api/servers/' + self.server().name() + '/databases/',
  253. dataType: 'json',
  254. type: 'GET',
  255. success: function(data) {
  256. self.updateDatabases(data.databases);
  257. },
  258. error: error_fn
  259. };
  260. $.ajax(request);
  261. }
  262. };
  263. }