workflow-editor.ko.js 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656
  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.removeNode = function(node_id) { alert(node_id);
  207. var node = self.getNodeById(node_id);
  208. var parentWidget = vm.getWidgetPredecessor(node_id);
  209. var parent = self.getNodeById(parentWidget.id());
  210. // if parent start
  211. if (parentWidget.widgetType() == 'start-widget') {
  212. parent.set_link('to', node.get_link('ok')['ok']);
  213. }
  214. self.nodes.remove(node);
  215. // if parent normal node
  216. // if parent is fork
  217. };
  218. self.getNodeById = function (node_id) {
  219. var _node = null;
  220. $.each(self.nodes(), function (index, node) {
  221. if (node.id() == node_id) {
  222. _node = node;
  223. return false;
  224. }
  225. });
  226. return _node;
  227. };
  228. }
  229. var WorkflowEditorViewModel = function (layout_json, workflow_json, credentials_json) {
  230. var self = this;
  231. self.isNested = ko.observable(true);
  232. self.isEditing = ko.observable(true);
  233. self.toggleEditing = function () {
  234. self.isEditing(! self.isEditing());
  235. };
  236. self.columns = ko.observable([]);
  237. self.previewColumns = ko.observable("");
  238. self.workflow = new Workflow(self, workflow_json);
  239. self.credentials = ko.mapping.fromJSON(credentials_json);
  240. self.inited = ko.observable(self.columns().length > 0);
  241. self.init = function(callback) {
  242. loadLayout(self, layout_json);
  243. self.workflow.loadNodes(workflow_json);
  244. }
  245. self.depen = ko.observableArray(workflow_json.dependencies);
  246. self.addActionProperties = ko.observableArray([]);
  247. self.addActionWorkflows = ko.observableArray([]);
  248. self.selectedSubWorkflow = ko.observable();
  249. self.currentlyDraggedWidget = null;
  250. self.currentlyCreatingFork = false;
  251. self.currentlyCreatedFork = null;
  252. self.currentlyCreatedJoin = null;
  253. self.isDragging = ko.observable(false);
  254. self.setCurrentDraggedWidget = function (widget) {
  255. self.currentlyDraggedWidget = widget;
  256. }
  257. self.addDraggedWidget = function (row, atBeginning) {
  258. if (self.currentlyDraggedWidget != null) {
  259. var _parentCol = self.getRowParentColumn(row.id());
  260. var _rowIdx = 0;
  261. $.each(_parentCol.rows(), function (i, irow) {
  262. if (irow.id() == row.id()) {
  263. _rowIdx = i;
  264. }
  265. });
  266. var _forkRow = _parentCol.addEmptyRow(false, _rowIdx);
  267. var _id = UUID();
  268. var _fork = new Widget({
  269. size: 12,
  270. id: _id,
  271. name: 'fork' + '-' + _id.slice(0, 4),
  272. widgetType: "fork-widget",
  273. properties: {},
  274. offset: 0,
  275. loading: true,
  276. vm: self
  277. });
  278. _forkRow.widgets([_fork]);
  279. var _w = new Widget({
  280. size: self.currentlyDraggedWidget.size(),
  281. id: UUID(),
  282. name: self.currentlyDraggedWidget.name(),
  283. widgetType: self.currentlyDraggedWidget.widgetType(),
  284. properties: self.currentlyDraggedWidget.properties(),
  285. offset: self.currentlyDraggedWidget.offset(),
  286. loading: true,
  287. vm: self
  288. });
  289. var _col = row.addEmptyColumn(atBeginning);
  290. var _row = new Row([_w], self);
  291. _col.addRow(_row);
  292. var _joinRow = _parentCol.addEmptyRow(false, _rowIdx + 2);
  293. var _id = UUID();
  294. var _join = new Widget({
  295. size: 12,
  296. id: _id,
  297. name: "join" + '-' + _id.slice(0, 4),
  298. widgetType: "join-widget",
  299. properties: {},
  300. offset: 0,
  301. loading: true,
  302. vm: self
  303. });
  304. _joinRow.widgets([_join]);
  305. self.currentlyDraggedWidget = null;
  306. self.currentlyCreatingFork = true;
  307. self.currentlyCreatedFork = ko.mapping.toJS(_fork);
  308. self.currentlyCreatedJoin = ko.mapping.toJS(_join);
  309. return _w;
  310. }
  311. }
  312. self.getWidgetById = function (widget_id) {
  313. var _widget = null;
  314. $.each(self.columns(), function (i, col) {
  315. $.each(col.rows(), function (j, row) {
  316. $.each(row.widgets(), function (z, widget) {
  317. if (widget.id() == widget_id){
  318. _widget = widget;
  319. return false;
  320. }
  321. });
  322. });
  323. });
  324. return _widget;
  325. }
  326. self.removeWidget = function (widget_json) {
  327. self.workflow.removeNode(widget_json.id());
  328. self.removeWidgetById(widget_json.id());
  329. }
  330. self.removeWidgetById = function (widget_id) {
  331. $.each(self.columns(), function (i, col) {
  332. self.deeplyRemoveWidgetById(widget_id, col, self)
  333. });
  334. }
  335. self.deeplyRemoveWidgetById = function (widget_id, col, parent) {
  336. if (col) {
  337. $.each(col.rows(), function (j, row) {
  338. if (row && row.widgets()){
  339. $.each(row.widgets(), function (z, widget) {
  340. if (widget.id() == widget_id) {
  341. row.widgets.remove(widget);
  342. col.rows.remove(row);
  343. }
  344. });
  345. }
  346. if (row && row.columns()) {
  347. $.each(row.columns(), function (i, icol) {
  348. self.deeplyRemoveWidgetById(widget_id, icol, row);
  349. });
  350. }
  351. });
  352. if (col.rows().length == 0) {
  353. parent.columns.remove(col);
  354. if (parent.columns().length > 1) {
  355. var _size = Math.max(1, Math.floor(12 / (parent.columns().length)));
  356. parent.columns().forEach(function (icol) {
  357. icol.size(_size);
  358. });
  359. }
  360. else {
  361. var _rows = parent.columns()[0].rows();
  362. var _parentRows = self.getRowParentColumn(parent.id()).rows;
  363. var _prevRowIdx = -1;
  364. for (var i = 0; i < _parentRows().length; i++) {
  365. if (_parentRows()[i].id() == parent.id()) {
  366. break;
  367. }
  368. _prevRowIdx = i;
  369. }
  370. if (_prevRowIdx > -1 && _parentRows()[_prevRowIdx].widgets().length > 0 && _parentRows()[_prevRowIdx].widgets()[0].widgetType() == "fork-widget"){
  371. _parentRows.remove(_parentRows()[_prevRowIdx]);
  372. _parentRows.remove(_parentRows()[_prevRowIdx+1]);
  373. }
  374. for (var i=0;i<_rows.length;i++){
  375. if (i==0){
  376. parent.widgets(_rows[i].widgets());
  377. }
  378. else {
  379. _parentRows.push(_rows[i]);
  380. }
  381. }
  382. parent.columns([]);
  383. }
  384. }
  385. }
  386. }
  387. self.getWidgetPredecessor = function (widget_id) {
  388. var _row = self.getWidgetParentRow(widget_id);
  389. var _col = self.getRowParentColumn(_row.id());
  390. var _prevRow = null;
  391. for (var i = 0; i < _col.rows().length; i++) {
  392. if (_col.rows()[i].id() == _row.id()) {
  393. break;
  394. }
  395. _prevRow = _col.rows()[i];
  396. }
  397. if (_prevRow != null) {
  398. return _prevRow.widgets()[0];
  399. }
  400. else {
  401. var _parentRow = self.getColumnParentRow(_col.id());
  402. var _parentColumn = self.getRowParentColumn(_parentRow.id());
  403. var _prevParentRow = null;
  404. for (var i = 0; i < _parentColumn.rows().length; i++) {
  405. if (_parentColumn.rows()[i].id() == _parentRow.id()) {
  406. break;
  407. }
  408. _prevParentRow = _parentColumn.rows()[i];
  409. }
  410. if (_prevParentRow != null) {
  411. return _prevParentRow.widgets()[0];
  412. }
  413. }
  414. return null;
  415. }
  416. self.getWidgetParentRow = function (widget_id) {
  417. var _row = null;
  418. for (var i = 0; i < self.columns().length; i++) {
  419. _row = self.traverseColumnForWidget(widget_id, self.columns()[i]);
  420. if (_row != null) {
  421. break;
  422. }
  423. }
  424. return _row;
  425. }
  426. self.getRowParentColumn = function (row_id) {
  427. var _column = null;
  428. for (var i = 0; i < self.columns().length; i++) {
  429. _column = self.traverseColumnForColumn(row_id, self.columns()[i]);
  430. }
  431. return _column;
  432. }
  433. self.getColumnParentRow = function (col_id) {
  434. var _row = null;
  435. for (var i = 0; i < self.columns().length; i++) {
  436. _row = self.traverseColumnForRow(col_id, self.columns()[i]);
  437. if (_row != null) {
  438. break;
  439. }
  440. }
  441. return _row;
  442. }
  443. self.getRowParentRow = function (row_id) {
  444. var _col = self.getRowParentColumn(row_id);
  445. if (_col != null) {
  446. return self.getColumnParentRow(_col.id());
  447. }
  448. }
  449. self.traverseColumnForColumn = function (row_id, col) {
  450. var _column = null;
  451. if (col) {
  452. for (var j = 0; j < col.rows().length; j++) {
  453. var row = col.rows()[j];
  454. if (row.id() == row_id) {
  455. _column = col;
  456. break;
  457. }
  458. for (var z = 0; z < row.columns().length; z++) {
  459. _column = self.traverseColumnForColumn(row_id, row.columns()[z]);
  460. if (_column != null) {
  461. break;
  462. }
  463. }
  464. }
  465. }
  466. return _column;
  467. }
  468. self.traverseColumnForRow = function (col_id, col) {
  469. var _row = null;
  470. if (col) {
  471. for (var j = 0; j < col.rows().length; j++) {
  472. var row = col.rows()[j];
  473. for (var z = 0; z < row.columns().length; z++) {
  474. var _col = row.columns()[z];
  475. if (_col.id() == col_id) {
  476. _row = row;
  477. }
  478. else {
  479. _row = self.traverseColumnForRow(col_id, _col);
  480. }
  481. if (_row != null) {
  482. break;
  483. }
  484. }
  485. }
  486. }
  487. return _row;
  488. }
  489. self.traverseColumnForWidget = function (widget_id, col) {
  490. var _row = null;
  491. if (col) {
  492. for (var j = 0; j < col.rows().length; j++) {
  493. var row = col.rows()[j];
  494. for (var z = 0; z < row.widgets().length; z++) {
  495. var widget = row.widgets()[z];
  496. if (widget.id() == widget_id) {
  497. _row = row;
  498. break;
  499. }
  500. }
  501. for (var z = 0; z < row.columns().length; z++) {
  502. _row = self.traverseColumnForWidget(widget_id, row.columns()[z]);
  503. if (_row != null) {
  504. break;
  505. }
  506. }
  507. }
  508. }
  509. return _row;
  510. }
  511. self.save = function () {
  512. $.post("/oozie/editor/workflow/save/", {
  513. "layout": ko.mapping.toJSON(self.columns),
  514. "workflow": ko.mapping.toJSON(self.workflow)
  515. }, function (data) {
  516. if (data.status == 0) {
  517. self.workflow.id(data.id);
  518. $(document).trigger("info", data.message);
  519. if (window.location.search.indexOf("workflow") == -1) {
  520. window.location.hash = '#workflow=' + data.id;
  521. }
  522. }
  523. else {
  524. $(document).trigger("error", data.message);
  525. }
  526. }).fail(function (xhr, textStatus, errorThrown) {
  527. $(document).trigger("error", xhr.responseText);
  528. });
  529. };
  530. self.gen_xml = function () {
  531. $.post("/oozie/editor/workflow/gen_xml/", {
  532. "layout": ko.mapping.toJSON(self.columns),
  533. "workflow": ko.mapping.toJSON(self.workflow)
  534. }, function (data) {
  535. if (data.status == 0) {
  536. alert(data.xml);
  537. }
  538. else {
  539. $(document).trigger("error", data.message);
  540. }
  541. }).fail(function (xhr, textStatus, errorThrown) {
  542. $(document).trigger("error", xhr.responseText);
  543. });
  544. };
  545. self.showSubmitPopup = function () {
  546. // If self.workflow.id() == null, need to save wf for now
  547. $.get("/oozie/editor/workflow/submit/" + self.workflow.id(), {
  548. }, function (data) {
  549. $(document).trigger("showSubmitPopup", data);
  550. }).fail(function (xhr, textStatus, errorThrown) {
  551. $(document).trigger("error", xhr.responseText);
  552. });
  553. };
  554. function bareWidgetBuilder(name, type){
  555. return new Widget({
  556. size: 12,
  557. id: UUID(),
  558. name: name,
  559. widgetType: type
  560. });
  561. }
  562. self.draggableHiveAction = ko.observable(bareWidgetBuilder("Hive Script", "hive-widget"));
  563. self.draggablePigAction = ko.observable(bareWidgetBuilder("Pig Script", "pig-widget"));
  564. self.draggableJavaAction = ko.observable(bareWidgetBuilder("Java program", "java-widget"));
  565. self.draggableMapReduceAction = ko.observable(bareWidgetBuilder("MapReduce job", "mapreduce-widget"));
  566. self.draggableSubworkflowAction = ko.observable(bareWidgetBuilder("Sub workflow", "subworkflow-widget"));
  567. self.draggableStopNode = ko.observable(bareWidgetBuilder("Kill", "kill-widget"));
  568. };