queryBuilder.js 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168
  1. const squel = require("squel");
  2. function selectAll(table, whereString) {
  3. let query = squel.select()
  4. .from(table)
  5. .field('*');
  6. if (whereString) {
  7. query = query.where(whereString);
  8. }
  9. return query.toString();
  10. }
  11. function selectOne(table, id) {
  12. return squel.select()
  13. .from(table)
  14. .field('*')
  15. .where("id = ?", id)
  16. .toString();
  17. }
  18. function selectWhere(table, where, operator) {
  19. operator = operator || '=';
  20. let query = squel.select()
  21. .from(table)
  22. .field('*');
  23. for (var key in where) {
  24. query = query.where(key + ' ' + operator + ' ?', where[key]);
  25. }
  26. return query.toString();
  27. }
  28. function deleteWithId(table, id, idKey) {
  29. idKey = idKey !== undefined ? idKey : 'id';
  30. return squel.delete()
  31. .from(table)
  32. .where(idKey + " = ?", id)
  33. .toString();
  34. }
  35. function selectIn(table, ids) {
  36. const where = ids.length === 0 ? '0' :
  37. "id IN (" + ids.join(',') + ")";
  38. return squel.select()
  39. .from(table)
  40. .field('*')
  41. .where(where)
  42. .toString();
  43. }
  44. function selectRelatees(table, relateeKey, relateeId) {
  45. return squel.select()
  46. .from(table)
  47. .field('*')
  48. .where(relateeKey + " = ?", relateeId)
  49. .toString();
  50. }
  51. function selectRelateesIn(table, relateeKey, relateeIds) {
  52. const where = relateeIds.length === 0 ? '0' :
  53. relateeKey + " IN (" + relateeIds.join(',') + ")";
  54. return squel.select()
  55. .from(table)
  56. .field('*')
  57. .where(where)
  58. .toString();
  59. }
  60. // function selectMany(table, ids) {
  61. // const idsString = ids.join(',');
  62. // return squel.select()
  63. // .from(table)
  64. // .field('*')
  65. // .where("id IN ?", idsString)
  66. // .toString();
  67. // }
  68. function getSelectOne(table) {
  69. return id => {
  70. return squel.select()
  71. .from(table)
  72. .field('*')
  73. .where("id = ?", id)
  74. .toString();
  75. };
  76. }
  77. function getInsert(table) {
  78. return attributes => {
  79. attributes = Array.isArray(attributes) ? attributes : [attributes];
  80. return insert(table, attributes);
  81. };
  82. }
  83. function insert(table, rows) {
  84. rows = Array.isArray(rows) ? rows : [rows];
  85. rows.forEach(row => { delete row.id; });
  86. return squel.insert({ replaceSingleQuotes: true, autoQuoteFieldNames: true })
  87. .into(table)
  88. .setFieldsRows(rows)
  89. .toString();
  90. }
  91. function updateOne(table, id, attributes) {
  92. delete attributes.id;
  93. if(attributes.createdAt) {
  94. delete attributes.createdAt;
  95. }
  96. return squel.update({ replaceSingleQuotes: true, autoQuoteFieldNames: true })
  97. .table(table)
  98. .setFields(attributes)
  99. .where('id = ' + id)
  100. .toString();
  101. }
  102. function updateWhere(table, where, attributes) {
  103. delete attributes.id;
  104. if(attributes.createdAt) {
  105. delete attributes.createdAt;
  106. }
  107. let query = squel.update({ replaceSingleQuotes: true, autoQuoteFieldNames: true })
  108. .table(table)
  109. .setFields(attributes);
  110. for (var key in where) {
  111. query = query.where(key + ' = ?', where[key]);
  112. }
  113. return query.toString();
  114. }
  115. function updateIn(table, ids, attributes) {
  116. delete attributes.id;
  117. if(attributes.createdAt) {
  118. delete attributes.createdAt;
  119. }
  120. const where = ids.length === 0 ? '0' :
  121. "id IN (" + ids.join(',') + ")";
  122. return squel.update({ replaceSingleQuotes: true, autoQuoteFieldNames: true })
  123. .table(table)
  124. .setFields(attributes)
  125. .where(where)
  126. .toString();
  127. }
  128. function getUpdateOne(table, id) {
  129. return attributes => updateOne(table, id, attributes);
  130. }
  131. // describe('squel query', () => {
  132. // it('should build a select query', () => {
  133. // const sql =
  134. // sql.should.equal('SELECT * FROM table');
  135. // });
  136. // it('should build an insert query', () => {
  137. // const sql = create('users', { email: 'bh@localhost.local', order: 1, password: '###bloody#hash', createdAt: '2017-02-27' });
  138. // lineLogger(sql);
  139. // });
  140. // });
  141. module.exports = {
  142. selectAll, selectOne, selectIn, selectWhere, selectRelatees, selectRelateesIn, getSelectOne,
  143. insert, getInsert,
  144. updateOne, updateWhere, getUpdateOne, updateIn,
  145. deleteWithId
  146. };