workflow-editor.ko.js 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637
  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. ko.bindingHandlers.droppable = {
  17. init: function(element, valueAccessor) {
  18. var _dropElement = $(element);
  19. var _options = valueAccessor();
  20. if (_options.enabled){
  21. var _dropOptions = {
  22. hoverClass: 'drop-target-highlight',
  23. drop: _options.onDrop
  24. };
  25. _dropElement.droppable(_dropOptions);
  26. }
  27. }
  28. };
  29. function magicLayout(vm) {
  30. loadLayout(vm, vm.initial.layout);
  31. $(document).trigger("magicLayout");
  32. }
  33. function loadLayout(viewModel, json_layout) {
  34. var _columns = [];
  35. $(json_layout).each(function (cnt, json_col) {
  36. var _rows = [];
  37. $(json_col.rows).each(function (rcnt, json_row) {
  38. var row = new Row([], viewModel);
  39. $(json_row.widgets).each(function (wcnt, widget) {
  40. row.addWidget(new Widget({
  41. size:widget.size,
  42. id: widget.id,
  43. name: widget.name,
  44. widgetType: widget.widgetType,
  45. properties: widget.properties,
  46. offset: widget.offset,
  47. loading: true,
  48. vm: viewModel
  49. }));
  50. });
  51. $(json_row.columns).each(function (ccnt, column) {
  52. var _irows = [];
  53. $(column.rows).each(function (ircnt, json_irow) {
  54. var _irow = new Row([], viewModel);
  55. $(json_irow.widgets).each(function (iwcnt, iwidget) {
  56. _irow.addWidget(new Widget({
  57. size:iwidget.size,
  58. id: iwidget.id,
  59. name: iwidget.name,
  60. widgetType: iwidget.widgetType,
  61. properties: iwidget.properties,
  62. offset: iwidget.offset,
  63. loading: true,
  64. vm: viewModel
  65. }));
  66. });
  67. _irows.push(_irow);
  68. });
  69. row.addColumn(new Column(column.size, _irows));
  70. });
  71. _rows.push(row);
  72. });
  73. var column = new Column(json_col.size, _rows);
  74. _columns = _columns.concat(column);
  75. });
  76. viewModel.columns(_columns);
  77. }
  78. // End dashboard lib
  79. var Node = function (node) {
  80. var self = this;
  81. var type = typeof node.widgetType != "undefined" ? node.widgetType : node.type;
  82. self.id = ko.observable(typeof node.id != "undefined" && node.id != null ? node.id : UUID());
  83. self.name = ko.observable(typeof node.name != "undefined" && node.name != null ? node.name : "");
  84. self.type = ko.observable(typeof type != "undefined" && type != null ? type : "");
  85. self.properties = ko.mapping.fromJS(typeof node.properties != "undefined" && node.properties != null ? node.properties : {});
  86. self.children = ko.mapping.fromJS(typeof node.children != "undefined" && node.children != null ? node.children : []);
  87. self.get_link = function(name) {
  88. var _link = null;
  89. $.each(self.children(), function(index, link) {
  90. if (name in link) {
  91. _link = link;
  92. return false;
  93. }
  94. });
  95. return _link;
  96. }
  97. self.set_link = function(name, node_id) {
  98. var _link = self.get_link(name);
  99. if (_link == null) {
  100. _link = {};
  101. self.children.push(_link);
  102. }
  103. _link[name] = node_id;
  104. }
  105. }
  106. var Workflow = function (vm, workflow) {
  107. var self = this;
  108. self.id = ko.observable(typeof workflow.id != "undefined" && workflow.id != null ? workflow.id : null);
  109. self.uuid = ko.observable(typeof workflow.uuid != "undefined" && workflow.uuid != null ? workflow.uuid : UUID());
  110. self.name = ko.observable(typeof workflow.name != "undefined" && workflow.name != null ? workflow.name : "");
  111. self.properties = ko.mapping.fromJS(typeof workflow.properties != "undefined" && workflow.properties != null ? workflow.properties : {});
  112. self.nodes = ko.observableArray([]);
  113. self.loadNodes = function(workflow) {
  114. var nodes = []
  115. $.each(workflow.nodes, function(index, node) {
  116. var _node = new Node(node);
  117. nodes.push(_node);
  118. });
  119. self.nodes(nodes)
  120. }
  121. self.newNode = function(widget) {
  122. $.ajax({
  123. type: "POST",
  124. url: "/oozie/editor/workflow/new_node/",
  125. data: {
  126. "workflow": ko.mapping.toJSON(workflow),
  127. "node": ko.mapping.toJSON(widget)
  128. },
  129. success: function (data) {
  130. if (data.status == 0) {
  131. viewModel.addActionProperties(data.properties);
  132. viewModel.addActionWorkflows(data.workflows);
  133. }
  134. },
  135. async: false
  136. });
  137. };
  138. self.addNode = function(widget) {
  139. // Todo get parent cell, link nodes... when we have the new layout
  140. $.post("/oozie/editor/workflow/add_node/", {
  141. "workflow": ko.mapping.toJSON(workflow),
  142. "node": ko.mapping.toJSON(widget),
  143. "properties": ko.mapping.toJSON(viewModel.addActionProperties()),
  144. "subworkflow": viewModel.selectedSubWorkflow() ? ko.mapping.toJSON(viewModel.selectedSubWorkflow()) : '{}',
  145. }, function (data) {
  146. if (data.status == 0) {
  147. var _node = ko.mapping.toJS(widget);
  148. _node.properties = data.properties;
  149. _node.name = data.name;
  150. var node = new Node(_node);
  151. // Add to list of nodes
  152. var end = self.nodes.pop();
  153. self.nodes.push(node);
  154. self.nodes.push(end);
  155. // if node != kill node
  156. // Added to the side ?
  157. if (vm.currentlyCreatingFork) {
  158. var parent = self.getNodeById('3f107997-04cc-8733-60a9-a4bb62cebffc');
  159. if (parent.type() != 'fork-widget') {
  160. var fork = new Node(vm.currentlyCreatedFork);
  161. var join = new Node(vm.currentlyCreatedJoin);
  162. // Start node
  163. var afterStartId = ko.mapping.toJS(parent.get_link('to')).to;
  164. var afterStart = self.getNodeById(afterStartId);
  165. fork.children.push({'to': afterStartId});
  166. fork.children.push({'to': node.id()});
  167. parent.get_link('to')['to'] = fork.id();
  168. join.set_link('to', afterStart.get_link('ok')['ok']);
  169. afterStart.set_link('ok', join.id());
  170. node.set_link('ok', join.id());
  171. node.set_link('error', '17c9c895-5a16-7443-bb81-f34b30b21548');
  172. var end = self.nodes.pop();
  173. self.nodes.push(fork);
  174. self.nodes.push(join);
  175. self.nodes.push(end);
  176. // Regular node
  177. // Join node
  178. } else {
  179. // Just add to existing fork
  180. }
  181. } else {
  182. var parentWidget = vm.getWidgetPredecessor(node.id());
  183. var parent = self.getNodeById(parentWidget.id());
  184. if (parentWidget.widgetType() == 'start-widget') {
  185. // Star node link to new node
  186. parent.set_link('to', node.id());
  187. // Link to end
  188. node.set_link('ok', '33430f0f-ebfa-c3ec-f237-3e77efa03d0a');
  189. node.set_link('error', '17c9c895-5a16-7443-bb81-f34b30b21548');
  190. } else if (parentWidget.widgetType() == 'pig-widget') {
  191. // Parent regular node
  192. node.set_link('ok', parent.get_link('ok')['ok']);
  193. node.set_link('error', '17c9c895-5a16-7443-bb81-f34b30b21548');
  194. parent.set_link('ok', node.id());
  195. }
  196. // Parent fork/decision/join...
  197. }
  198. vm.currentlyCreatingFork = false;
  199. } else {
  200. $(document).trigger("error", data.message);
  201. }
  202. }).fail(function (xhr, textStatus, errorThrown) {
  203. $(document).trigger("error", xhr.responseText);
  204. });
  205. };
  206. self.getNodeById = function (node_id) {
  207. var _node = null;
  208. $.each(self.nodes(), function (index, node) {
  209. if (node.id() == node_id) {
  210. _node = node;
  211. return false;
  212. }
  213. });
  214. return _node;
  215. }
  216. }
  217. var WorkflowEditorViewModel = function (layout_json, workflow_json, credentials_json) {
  218. var self = this;
  219. self.isNested = ko.observable(true);
  220. self.isEditing = ko.observable(true);
  221. self.toggleEditing = function () {
  222. self.isEditing(! self.isEditing());
  223. };
  224. self.columns = ko.observable([]);
  225. self.previewColumns = ko.observable("");
  226. self.workflow = new Workflow(self, workflow_json);
  227. self.credentials = ko.mapping.fromJSON(credentials_json);
  228. self.inited = ko.observable(self.columns().length > 0);
  229. self.init = function(callback) {
  230. loadLayout(self, layout_json);
  231. self.workflow.loadNodes(workflow_json);
  232. }
  233. self.depen = ko.observableArray(workflow_json.dependencies);
  234. self.addActionProperties = ko.observableArray([]);
  235. self.addActionWorkflows = ko.observableArray([]);
  236. self.selectedSubWorkflow = ko.observable();
  237. self.currentlyDraggedWidget = null;
  238. self.currentlyCreatingFork = false;
  239. self.currentlyCreatedFork = null;
  240. self.currentlyCreatedJoin = null;
  241. self.isDragging = ko.observable(false);
  242. self.setCurrentDraggedWidget = function (widget) {
  243. self.currentlyDraggedWidget = widget;
  244. }
  245. self.addDraggedWidget = function (row, atBeginning) {
  246. if (self.currentlyDraggedWidget != null) {
  247. var _parentCol = self.getRowParentColumn(row.id());
  248. var _rowIdx = 0;
  249. $.each(_parentCol.rows(), function (i, irow) {
  250. if (irow.id() == row.id()) {
  251. _rowIdx = i;
  252. }
  253. });
  254. var _forkRow = _parentCol.addEmptyRow(false, _rowIdx);
  255. var _id = UUID();
  256. var _fork = new Widget({
  257. size: 12,
  258. id: _id,
  259. name: 'fork' + '-' + _id.slice(0, 4),
  260. widgetType: "fork-widget",
  261. properties: {},
  262. offset: 0,
  263. loading: true,
  264. vm: self
  265. });
  266. _forkRow.widgets([_fork]);
  267. var _w = new Widget({
  268. size: self.currentlyDraggedWidget.size(),
  269. id: UUID(),
  270. name: self.currentlyDraggedWidget.name(),
  271. widgetType: self.currentlyDraggedWidget.widgetType(),
  272. properties: self.currentlyDraggedWidget.properties(),
  273. offset: self.currentlyDraggedWidget.offset(),
  274. loading: true,
  275. vm: self
  276. });
  277. var _col = row.addEmptyColumn(atBeginning);
  278. var _row = new Row([_w], self);
  279. _col.addRow(_row);
  280. var _joinRow = _parentCol.addEmptyRow(false, _rowIdx + 2);
  281. var _id = UUID();
  282. var _join = new Widget({
  283. size: 12,
  284. id: _id,
  285. name: "join" + '-' + _id.slice(0, 4),
  286. widgetType: "join-widget",
  287. properties: {},
  288. offset: 0,
  289. loading: true,
  290. vm: self
  291. });
  292. _joinRow.widgets([_join]);
  293. self.currentlyDraggedWidget = null;
  294. self.currentlyCreatingFork = true;
  295. self.currentlyCreatedFork = ko.mapping.toJS(_fork);
  296. self.currentlyCreatedJoin = ko.mapping.toJS(_join);
  297. return _w;
  298. }
  299. }
  300. self.getWidgetById = function (widget_id) {
  301. var _widget = null;
  302. $.each(self.columns(), function (i, col) {
  303. $.each(col.rows(), function (j, row) {
  304. $.each(row.widgets(), function (z, widget) {
  305. if (widget.id() == widget_id){
  306. _widget = widget;
  307. return false;
  308. }
  309. });
  310. });
  311. });
  312. return _widget;
  313. }
  314. self.removeWidget = function (widget_json) {
  315. self.removeWidgetById(widget_json.id());
  316. }
  317. self.removeWidgetById = function (widget_id) {
  318. $.each(self.columns(), function (i, col) {
  319. self.deeplyRemoveWidgetById(widget_id, col, self)
  320. });
  321. }
  322. self.deeplyRemoveWidgetById = function (widget_id, col, parent) {
  323. if (col) {
  324. $.each(col.rows(), function (j, row) {
  325. if (row && row.widgets()){
  326. $.each(row.widgets(), function (z, widget) {
  327. if (widget.id() == widget_id) {
  328. row.widgets.remove(widget);
  329. col.rows.remove(row);
  330. }
  331. });
  332. }
  333. if (row && row.columns()) {
  334. $.each(row.columns(), function (i, icol) {
  335. self.deeplyRemoveWidgetById(widget_id, icol, row);
  336. });
  337. }
  338. });
  339. if (col.rows().length == 0) {
  340. parent.columns.remove(col);
  341. if (parent.columns().length > 1) {
  342. var _size = Math.max(1, Math.floor(12 / (parent.columns().length)));
  343. parent.columns().forEach(function (icol) {
  344. icol.size(_size);
  345. });
  346. }
  347. else {
  348. var _rows = parent.columns()[0].rows();
  349. var _parentRows = self.getRowParentColumn(parent.id()).rows;
  350. var _prevRowIdx = -1;
  351. for (var i = 0; i < _parentRows().length; i++) {
  352. if (_parentRows()[i].id() == parent.id()) {
  353. break;
  354. }
  355. _prevRowIdx = i;
  356. }
  357. if (_prevRowIdx > -1 && _parentRows()[_prevRowIdx].widgets().length > 0 && _parentRows()[_prevRowIdx].widgets()[0].widgetType() == "fork-widget"){
  358. _parentRows.remove(_parentRows()[_prevRowIdx]);
  359. _parentRows.remove(_parentRows()[_prevRowIdx+1]);
  360. }
  361. for (var i=0;i<_rows.length;i++){
  362. if (i==0){
  363. parent.widgets(_rows[i].widgets());
  364. }
  365. else {
  366. _parentRows.push(_rows[i]);
  367. }
  368. }
  369. parent.columns([]);
  370. }
  371. }
  372. }
  373. }
  374. self.getWidgetPredecessor = function (widget_id) {
  375. var _row = self.getWidgetParentRow(widget_id);
  376. var _col = self.getRowParentColumn(_row.id());
  377. var _prevRow = null;
  378. for (var i = 0; i < _col.rows().length; i++) {
  379. if (_col.rows()[i].id() == _row.id()) {
  380. break;
  381. }
  382. _prevRow = _col.rows()[i];
  383. }
  384. if (_prevRow != null) {
  385. return _prevRow.widgets()[0];
  386. }
  387. else {
  388. var _parentRow = self.getColumnParentRow(_col.id());
  389. var _parentColumn = self.getRowParentColumn(_parentRow.id());
  390. var _prevParentRow = null;
  391. for (var i = 0; i < _parentColumn.rows().length; i++) {
  392. if (_parentColumn.rows()[i].id() == _parentRow.id()) {
  393. break;
  394. }
  395. _prevParentRow = _parentColumn.rows()[i];
  396. }
  397. if (_prevParentRow != null) {
  398. return _prevParentRow.widgets()[0];
  399. }
  400. }
  401. return null;
  402. }
  403. self.getWidgetParentRow = function (widget_id) {
  404. var _row = null;
  405. for (var i = 0; i < self.columns().length; i++) {
  406. _row = self.traverseColumnForWidget(widget_id, self.columns()[i]);
  407. if (_row != null) {
  408. break;
  409. }
  410. }
  411. return _row;
  412. }
  413. self.getRowParentColumn = function (row_id) {
  414. var _column = null;
  415. for (var i = 0; i < self.columns().length; i++) {
  416. _column = self.traverseColumnForColumn(row_id, self.columns()[i]);
  417. }
  418. return _column;
  419. }
  420. self.getColumnParentRow = function (col_id) {
  421. var _row = null;
  422. for (var i = 0; i < self.columns().length; i++) {
  423. _row = self.traverseColumnForRow(col_id, self.columns()[i]);
  424. if (_row != null) {
  425. break;
  426. }
  427. }
  428. return _row;
  429. }
  430. self.getRowParentRow = function (row_id) {
  431. var _col = self.getRowParentColumn(row_id);
  432. if (_col != null) {
  433. return self.getColumnParentRow(_col.id());
  434. }
  435. }
  436. self.traverseColumnForColumn = function (row_id, col) {
  437. var _column = null;
  438. if (col) {
  439. for (var j = 0; j < col.rows().length; j++) {
  440. var row = col.rows()[j];
  441. if (row.id() == row_id) {
  442. _column = col;
  443. break;
  444. }
  445. for (var z = 0; z < row.columns().length; z++) {
  446. _column = self.traverseColumnForColumn(row_id, row.columns()[z]);
  447. if (_column != null) {
  448. break;
  449. }
  450. }
  451. }
  452. }
  453. return _column;
  454. }
  455. self.traverseColumnForRow = function (col_id, col) {
  456. var _row = null;
  457. if (col) {
  458. for (var j = 0; j < col.rows().length; j++) {
  459. var row = col.rows()[j];
  460. for (var z = 0; z < row.columns().length; z++) {
  461. var _col = row.columns()[z];
  462. if (_col.id() == col_id) {
  463. _row = row;
  464. }
  465. else {
  466. _row = self.traverseColumnForRow(col_id, _col);
  467. }
  468. if (_row != null) {
  469. break;
  470. }
  471. }
  472. }
  473. }
  474. return _row;
  475. }
  476. self.traverseColumnForWidget = function (widget_id, col) {
  477. var _row = null;
  478. if (col) {
  479. for (var j = 0; j < col.rows().length; j++) {
  480. var row = col.rows()[j];
  481. for (var z = 0; z < row.widgets().length; z++) {
  482. var widget = row.widgets()[z];
  483. if (widget.id() == widget_id) {
  484. _row = row;
  485. break;
  486. }
  487. }
  488. for (var z = 0; z < row.columns().length; z++) {
  489. _row = self.traverseColumnForWidget(widget_id, row.columns()[z]);
  490. if (_row != null) {
  491. break;
  492. }
  493. }
  494. }
  495. }
  496. return _row;
  497. }
  498. self.save = function () {
  499. $.post("/oozie/editor/workflow/save/", {
  500. "layout": ko.mapping.toJSON(self.columns),
  501. "workflow": ko.mapping.toJSON(self.workflow)
  502. }, function (data) {
  503. if (data.status == 0) {
  504. self.workflow.id(data.id);
  505. $(document).trigger("info", data.message);
  506. if (window.location.search.indexOf("workflow") == -1) {
  507. window.location.hash = '#workflow=' + data.id;
  508. }
  509. }
  510. else {
  511. $(document).trigger("error", data.message);
  512. }
  513. }).fail(function (xhr, textStatus, errorThrown) {
  514. $(document).trigger("error", xhr.responseText);
  515. });
  516. };
  517. self.gen_xml = function () {
  518. $.post("/oozie/editor/workflow/gen_xml/", {
  519. "layout": ko.mapping.toJSON(self.columns),
  520. "workflow": ko.mapping.toJSON(self.workflow)
  521. }, function (data) {
  522. if (data.status == 0) {
  523. alert(data.xml);
  524. }
  525. else {
  526. $(document).trigger("error", data.message);
  527. }
  528. }).fail(function (xhr, textStatus, errorThrown) {
  529. $(document).trigger("error", xhr.responseText);
  530. });
  531. };
  532. self.showSubmitPopup = function () {
  533. // If self.workflow.id() == null, need to save wf for now
  534. $.get("/oozie/editor/workflow/submit/" + self.workflow.id(), {
  535. }, function (data) {
  536. $(document).trigger("showSubmitPopup", data);
  537. }).fail(function (xhr, textStatus, errorThrown) {
  538. $(document).trigger("error", xhr.responseText);
  539. });
  540. };
  541. function bareWidgetBuilder(name, type){
  542. return new Widget({
  543. size: 12,
  544. id: UUID(),
  545. name: name,
  546. widgetType: type
  547. });
  548. }
  549. self.draggableHiveAction = ko.observable(bareWidgetBuilder("Hive Script", "hive-widget"));
  550. self.draggablePigAction = ko.observable(bareWidgetBuilder("Pig Script", "pig-widget"));
  551. self.draggableJavaAction = ko.observable(bareWidgetBuilder("Java program", "java-widget"));
  552. self.draggableMapReduceAction = ko.observable(bareWidgetBuilder("MapReduce job", "mapreduce-widget"));
  553. self.draggableSubworkflowAction = ko.observable(bareWidgetBuilder("Sub workflow", "subworkflow-widget"));
  554. self.draggableStopNode = ko.observable(bareWidgetBuilder("Kill", "kill-widget"));
  555. };