autocomplete.utils.js 9.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231
  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 hac_jsoncalls(options) {
  17. var _url = typeof options.autocompleteBaseURL != "undefined" ? options.autocompleteBaseURL : HIVE_AUTOCOMPLETE_BASE_URL;
  18. if (_url != ""){
  19. if (options.database != null) {
  20. _url += options.database
  21. }
  22. if (options.table != null) {
  23. _url += "/" + options.table
  24. }
  25. $.ajax({
  26. type: "GET",
  27. url: _url + "?" + Math.random(),
  28. success: options.onDataReceived,
  29. async: options.sync == "undefined"
  30. });
  31. }
  32. }
  33. function hac_hasExpired(timestamp){
  34. var TIME_TO_LIVE_IN_MILLIS = 86400000; // 1 day
  35. return (new Date()).getTime() - timestamp > TIME_TO_LIVE_IN_MILLIS;
  36. }
  37. function hac_getTableAliases(textScanned) {
  38. var _aliases = {};
  39. var _val = textScanned.split("\n").join(" ");
  40. var _from = _val.toUpperCase().indexOf("FROM ");
  41. if (_from > -1) {
  42. var _match = _val.toUpperCase().substring(_from).match(/ ON| LIMIT| WHERE| GROUP| SORT| ORDER BY|;/);
  43. var _to = _val.length;
  44. if (_match) {
  45. _to = _match.index;
  46. }
  47. var _found = _val.substr(_from, _to).replace(/(\r\n|\n|\r)/gm, "").replace(/from/gi, "").replace(/join/gi, ",").split(",");
  48. for (var i = 0; i < _found.length; i++) {
  49. var _tablealias = $.trim(_found[i]).split(" ");
  50. if (_tablealias.length > 1) {
  51. _aliases[_tablealias[1]] = _tablealias[0];
  52. }
  53. }
  54. }
  55. return _aliases;
  56. }
  57. function hac_getTotalStorageUserPrefix() {
  58. var _app = "";
  59. if (typeof HIVE_AUTOCOMPLETE_APP != "undefined") {
  60. _app = HIVE_AUTOCOMPLETE_APP;
  61. }
  62. if (typeof HIVE_AUTOCOMPLETE_USER != "undefined") {
  63. return _app + "_" + HIVE_AUTOCOMPLETE_USER + "_";
  64. }
  65. return (_app != "" ? _app + "_" : "");
  66. }
  67. function hac_getTableColumns(databaseName, tableName, textScanned, callback) {
  68. if (tableName.indexOf("(") > -1) {
  69. tableName = tableName.substr(tableName.indexOf("(") + 1);
  70. }
  71. var _aliases = hac_getTableAliases(textScanned);
  72. if (_aliases[tableName]) {
  73. tableName = _aliases[tableName];
  74. }
  75. if ($.totalStorage(hac_getTotalStorageUserPrefix() + 'columns_' + databaseName + '_' + tableName) != null && $.totalStorage(hac_getTotalStorageUserPrefix() + 'extended_columns_' + databaseName + '_' + tableName) != null) {
  76. callback($.totalStorage(hac_getTotalStorageUserPrefix() + 'columns_' + databaseName + '_' + tableName), $.totalStorage(hac_getTotalStorageUserPrefix() + 'extended_columns_' + databaseName + '_' + tableName));
  77. if ($.totalStorage(hac_getTotalStorageUserPrefix() + 'timestamp_columns_' + databaseName + '_' + tableName) == null || hac_hasExpired($.totalStorage(hac_getTotalStorageUserPrefix() + 'timestamp_columns_' + databaseName + '_' + tableName))){
  78. hac_jsoncalls({
  79. database: databaseName,
  80. table: tableName,
  81. onDataReceived: function (data) {
  82. if (typeof HIVE_AUTOCOMPLETE_GLOBAL_CALLBACK == "function") {
  83. HIVE_AUTOCOMPLETE_GLOBAL_CALLBACK(data);
  84. }
  85. if (data.error) {
  86. hac_errorHandler(data);
  87. }
  88. else {
  89. $.totalStorage(hac_getTotalStorageUserPrefix() + 'columns_' + databaseName + '_' + tableName, (data.columns ? "* " + data.columns.join(" ") : "*"));
  90. $.totalStorage(hac_getTotalStorageUserPrefix() + 'extended_columns_' + databaseName + '_' + tableName, (data.extended_columns ? data.extended_columns : []));
  91. $.totalStorage(hac_getTotalStorageUserPrefix() + 'timestamp_columns_' + databaseName + '_' + tableName, (new Date()).getTime());
  92. }
  93. }
  94. });
  95. }
  96. }
  97. else {
  98. hac_jsoncalls({
  99. database: databaseName,
  100. table: tableName,
  101. onDataReceived: function (data) {
  102. if (typeof HIVE_AUTOCOMPLETE_GLOBAL_CALLBACK == "function") {
  103. HIVE_AUTOCOMPLETE_GLOBAL_CALLBACK(data);
  104. }
  105. if (data.error) {
  106. hac_errorHandler(data);
  107. }
  108. else {
  109. $.totalStorage(hac_getTotalStorageUserPrefix() + 'columns_' + databaseName + '_' + tableName, (data.columns ? "* " + data.columns.join(" ") : "*"));
  110. $.totalStorage(hac_getTotalStorageUserPrefix() + 'extended_columns_' + databaseName + '_' + tableName, (data.extended_columns ? data.extended_columns : []));
  111. $.totalStorage(hac_getTotalStorageUserPrefix() + 'timestamp_columns_' + databaseName + '_' + tableName, (new Date()).getTime());
  112. callback($.totalStorage(hac_getTotalStorageUserPrefix() + 'columns_' + databaseName + '_' + tableName), $.totalStorage(hac_getTotalStorageUserPrefix() + 'extended_columns_' + databaseName + '_' + tableName));
  113. }
  114. }
  115. });
  116. }
  117. }
  118. function hac_tableHasAlias(tableName, textScanned) {
  119. var _aliases = hac_getTableAliases(textScanned);
  120. for (var alias in _aliases) {
  121. if (_aliases[alias] == tableName) {
  122. return true;
  123. }
  124. }
  125. return false;
  126. }
  127. function hac_getTables(databaseName, callback) {
  128. if ($.totalStorage(hac_getTotalStorageUserPrefix() + 'tables_' + databaseName) != null) {
  129. callback($.totalStorage(hac_getTotalStorageUserPrefix() + 'tables_' + databaseName));
  130. if ($.totalStorage(hac_getTotalStorageUserPrefix() + 'timestamp_tables_' + databaseName) == null || hac_hasExpired($.totalStorage(hac_getTotalStorageUserPrefix() + 'timestamp_tables_' + databaseName))){
  131. hac_jsoncalls({
  132. database: databaseName,
  133. onDataReceived: function (data) {
  134. if (typeof HIVE_AUTOCOMPLETE_GLOBAL_CALLBACK == "function") {
  135. HIVE_AUTOCOMPLETE_GLOBAL_CALLBACK(data);
  136. }
  137. if (data.error) {
  138. hac_errorHandler(data);
  139. }
  140. else {
  141. $.totalStorage(hac_getTotalStorageUserPrefix() + 'tables_' + databaseName, data.tables.join(" "));
  142. $.totalStorage(hac_getTotalStorageUserPrefix() + 'timestamp_tables_' + databaseName, (new Date()).getTime());
  143. }
  144. }
  145. });
  146. }
  147. }
  148. else {
  149. hac_jsoncalls({
  150. database: databaseName,
  151. onDataReceived: function (data) {
  152. if (typeof HIVE_AUTOCOMPLETE_GLOBAL_CALLBACK == "function") {
  153. HIVE_AUTOCOMPLETE_GLOBAL_CALLBACK(data);
  154. }
  155. if (data.error) {
  156. hac_errorHandler(data);
  157. }
  158. else {
  159. if (data.tables) {
  160. $.totalStorage(hac_getTotalStorageUserPrefix() + 'tables_' + databaseName, data.tables.join(" "));
  161. $.totalStorage(hac_getTotalStorageUserPrefix() + 'timestamp_tables_' + databaseName, (new Date()).getTime());
  162. callback($.totalStorage(hac_getTotalStorageUserPrefix() + 'tables_' + databaseName));
  163. }
  164. }
  165. }
  166. });
  167. }
  168. }
  169. function hac_getDatabases(callback) {
  170. if ($.totalStorage(hac_getTotalStorageUserPrefix() + 'databases') != null) {
  171. callback($.totalStorage(hac_getTotalStorageUserPrefix() + 'databases'));
  172. if ($.totalStorage(hac_getTotalStorageUserPrefix() + 'timestamp_databases') == null || hac_hasExpired($.totalStorage(hac_getTotalStorageUserPrefix() + 'timestamp_databases'))){
  173. hac_jsoncalls({
  174. onDataReceived: function (data) {
  175. if (typeof HIVE_AUTOCOMPLETE_GLOBAL_CALLBACK == "function") {
  176. HIVE_AUTOCOMPLETE_GLOBAL_CALLBACK(data);
  177. }
  178. if (data.error) {
  179. hac_errorHandler(data);
  180. }
  181. else {
  182. $.totalStorage(hac_getTotalStorageUserPrefix() + 'databases', data.databases);
  183. $.totalStorage(hac_getTotalStorageUserPrefix() + 'timestamp_databases', (new Date()).getTime());
  184. }
  185. }
  186. });
  187. }
  188. }
  189. else {
  190. hac_jsoncalls({
  191. onDataReceived: function (data) {
  192. if (typeof HIVE_AUTOCOMPLETE_GLOBAL_CALLBACK == "function") {
  193. HIVE_AUTOCOMPLETE_GLOBAL_CALLBACK(data);
  194. }
  195. if (data.error) {
  196. hac_errorHandler(data);
  197. }
  198. else {
  199. if (data.databases) {
  200. $.totalStorage(hac_getTotalStorageUserPrefix() + 'databases', data.databases);
  201. $.totalStorage(hac_getTotalStorageUserPrefix() + 'timestamp_databases', (new Date()).getTime());
  202. callback($.totalStorage(hac_getTotalStorageUserPrefix() + 'databases'));
  203. }
  204. }
  205. }
  206. });
  207. }
  208. }
  209. function hac_errorHandler(data) {
  210. $(document).trigger('error.autocomplete');
  211. if (typeof HIVE_AUTOCOMPLETE_FAILS_SILENTLY_ON == "undefined" || data.code == null || HIVE_AUTOCOMPLETE_FAILS_SILENTLY_ON.indexOf(data.code) == -1){
  212. if (typeof HIVE_AUTOCOMPLETE_FAILS_QUIETLY_ON != "undefined" && HIVE_AUTOCOMPLETE_FAILS_QUIETLY_ON.indexOf(data.code) > -1){
  213. $(document).trigger('info', data.error);
  214. }
  215. else {
  216. $(document).trigger('error', data.error);
  217. }
  218. }
  219. }