command.js 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277
  1. import { QueryCommand, QUERY_COMMANDS_LITERAL } from './commands/query';
  2. import { LogicCommand, LOGIC_COMMANDS_LITERAL } from './commands/logic';
  3. import { UpdateCommand, UPDATE_COMMANDS_LITERAL } from './commands/update';
  4. import { isArray, isObject, isString } from './utils/type';
  5. import Aggregation from './aggregate';
  6. export const Command = {
  7. eq(val) {
  8. return new QueryCommand(QUERY_COMMANDS_LITERAL.EQ, [val]);
  9. },
  10. neq(val) {
  11. return new QueryCommand(QUERY_COMMANDS_LITERAL.NEQ, [val]);
  12. },
  13. lt(val) {
  14. return new QueryCommand(QUERY_COMMANDS_LITERAL.LT, [val]);
  15. },
  16. lte(val) {
  17. return new QueryCommand(QUERY_COMMANDS_LITERAL.LTE, [val]);
  18. },
  19. gt(val) {
  20. return new QueryCommand(QUERY_COMMANDS_LITERAL.GT, [val]);
  21. },
  22. gte(val) {
  23. return new QueryCommand(QUERY_COMMANDS_LITERAL.GTE, [val]);
  24. },
  25. in(val) {
  26. return new QueryCommand(QUERY_COMMANDS_LITERAL.IN, val);
  27. },
  28. nin(val) {
  29. return new QueryCommand(QUERY_COMMANDS_LITERAL.NIN, val);
  30. },
  31. all(val) {
  32. return new QueryCommand(QUERY_COMMANDS_LITERAL.ALL, val);
  33. },
  34. elemMatch(val) {
  35. return new QueryCommand(QUERY_COMMANDS_LITERAL.ELEM_MATCH, [val]);
  36. },
  37. exists(val) {
  38. return new QueryCommand(QUERY_COMMANDS_LITERAL.EXISTS, [val]);
  39. },
  40. size(val) {
  41. return new QueryCommand(QUERY_COMMANDS_LITERAL.SIZE, [val]);
  42. },
  43. mod(val) {
  44. return new QueryCommand(QUERY_COMMANDS_LITERAL.MOD, [val]);
  45. },
  46. geoNear(val) {
  47. return new QueryCommand(QUERY_COMMANDS_LITERAL.GEO_NEAR, [val]);
  48. },
  49. geoWithin(val) {
  50. return new QueryCommand(QUERY_COMMANDS_LITERAL.GEO_WITHIN, [val]);
  51. },
  52. geoIntersects(val) {
  53. return new QueryCommand(QUERY_COMMANDS_LITERAL.GEO_INTERSECTS, [val]);
  54. },
  55. and(...__expressions__) {
  56. const expressions = isArray(arguments[0]) ? arguments[0] : Array.from(arguments);
  57. return new LogicCommand(LOGIC_COMMANDS_LITERAL.AND, expressions);
  58. },
  59. nor(...__expressions__) {
  60. const expressions = isArray(arguments[0]) ? arguments[0] : Array.from(arguments);
  61. return new LogicCommand(LOGIC_COMMANDS_LITERAL.NOR, expressions);
  62. },
  63. or(...__expressions__) {
  64. const expressions = isArray(arguments[0]) ? arguments[0] : Array.from(arguments);
  65. return new LogicCommand(LOGIC_COMMANDS_LITERAL.OR, expressions);
  66. },
  67. not(...__expressions__) {
  68. const expressions = isArray(arguments[0]) ? arguments[0] : Array.from(arguments);
  69. return new LogicCommand(LOGIC_COMMANDS_LITERAL.NOT, expressions);
  70. },
  71. set(val) {
  72. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.SET, [val]);
  73. },
  74. remove() {
  75. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.REMOVE, []);
  76. },
  77. inc(val) {
  78. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.INC, [val]);
  79. },
  80. mul(val) {
  81. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.MUL, [val]);
  82. },
  83. push(...args) {
  84. let values;
  85. if (isObject(args[0]) && args[0].hasOwnProperty('each')) {
  86. values = {};
  87. const options = args[0];
  88. if (options.each !== undefined) {
  89. values['$each'] = options.each;
  90. }
  91. if (options.position !== undefined) {
  92. values['$position'] = options.position;
  93. }
  94. if (options.sort !== undefined) {
  95. values['$sort'] = options.sort;
  96. }
  97. if (options.slice !== undefined) {
  98. values['$slice'] = options.slice;
  99. }
  100. }
  101. else if (isArray(args[0])) {
  102. values = args[0];
  103. }
  104. else {
  105. values = Array.from(args);
  106. }
  107. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.PUSH, values);
  108. },
  109. pull(values) {
  110. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.PULL, values);
  111. },
  112. pullAll(values) {
  113. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.PULL_ALL, values);
  114. },
  115. pop() {
  116. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.POP, []);
  117. },
  118. shift() {
  119. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.SHIFT, []);
  120. },
  121. unshift(...__values__) {
  122. const values = isArray(arguments[0]) ? arguments[0] : Array.from(arguments);
  123. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.UNSHIFT, values);
  124. },
  125. addToSet(values) {
  126. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.ADD_TO_SET, values);
  127. },
  128. rename(values) {
  129. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.RENAME, [values]);
  130. },
  131. bit(values) {
  132. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.BIT, [values]);
  133. },
  134. max(values) {
  135. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.MAX, [values]);
  136. },
  137. min(values) {
  138. return new UpdateCommand(UPDATE_COMMANDS_LITERAL.MIN, [values]);
  139. },
  140. expr(values) {
  141. return {
  142. $expr: values
  143. };
  144. },
  145. jsonSchema(schema) {
  146. return {
  147. $jsonSchema: schema
  148. };
  149. },
  150. text(values) {
  151. if (isString(values)) {
  152. return {
  153. $search: values.search
  154. };
  155. }
  156. else {
  157. return {
  158. $search: values.search,
  159. $language: values.language,
  160. $caseSensitive: values.caseSensitive,
  161. $diacriticSensitive: values.diacriticSensitive
  162. };
  163. }
  164. },
  165. aggregate: {
  166. pipeline() {
  167. return new Aggregation();
  168. },
  169. abs: param => new AggregationOperator('abs', param),
  170. add: param => new AggregationOperator('add', param),
  171. ceil: param => new AggregationOperator('ceil', param),
  172. divide: param => new AggregationOperator('divide', param),
  173. exp: param => new AggregationOperator('exp', param),
  174. floor: param => new AggregationOperator('floor', param),
  175. ln: param => new AggregationOperator('ln', param),
  176. log: param => new AggregationOperator('log', param),
  177. log10: param => new AggregationOperator('log10', param),
  178. mod: param => new AggregationOperator('mod', param),
  179. multiply: param => new AggregationOperator('multiply', param),
  180. pow: param => new AggregationOperator('pow', param),
  181. sqrt: param => new AggregationOperator('sqrt', param),
  182. subtract: param => new AggregationOperator('subtract', param),
  183. trunc: param => new AggregationOperator('trunc', param),
  184. arrayElemAt: param => new AggregationOperator('arrayElemAt', param),
  185. arrayToObject: param => new AggregationOperator('arrayToObject', param),
  186. concatArrays: param => new AggregationOperator('concatArrays', param),
  187. filter: param => new AggregationOperator('filter', param),
  188. in: param => new AggregationOperator('in', param),
  189. indexOfArray: param => new AggregationOperator('indexOfArray', param),
  190. isArray: param => new AggregationOperator('isArray', param),
  191. map: param => new AggregationOperator('map', param),
  192. range: param => new AggregationOperator('range', param),
  193. reduce: param => new AggregationOperator('reduce', param),
  194. reverseArray: param => new AggregationOperator('reverseArray', param),
  195. size: param => new AggregationOperator('size', param),
  196. slice: param => new AggregationOperator('slice', param),
  197. zip: param => new AggregationOperator('zip', param),
  198. and: param => new AggregationOperator('and', param),
  199. not: param => new AggregationOperator('not', param),
  200. or: param => new AggregationOperator('or', param),
  201. cmp: param => new AggregationOperator('cmp', param),
  202. eq: param => new AggregationOperator('eq', param),
  203. gt: param => new AggregationOperator('gt', param),
  204. gte: param => new AggregationOperator('gte', param),
  205. lt: param => new AggregationOperator('lt', param),
  206. lte: param => new AggregationOperator('lte', param),
  207. neq: param => new AggregationOperator('ne', param),
  208. cond: param => new AggregationOperator('cond', param),
  209. ifNull: param => new AggregationOperator('ifNull', param),
  210. switch: param => new AggregationOperator('switch', param),
  211. dateFromParts: param => new AggregationOperator('dateFromParts', param),
  212. dateFromString: param => new AggregationOperator('dateFromString', param),
  213. dayOfMonth: param => new AggregationOperator('dayOfMonth', param),
  214. dayOfWeek: param => new AggregationOperator('dayOfWeek', param),
  215. dayOfYear: param => new AggregationOperator('dayOfYear', param),
  216. isoDayOfWeek: param => new AggregationOperator('isoDayOfWeek', param),
  217. isoWeek: param => new AggregationOperator('isoWeek', param),
  218. isoWeekYear: param => new AggregationOperator('isoWeekYear', param),
  219. millisecond: param => new AggregationOperator('millisecond', param),
  220. minute: param => new AggregationOperator('minute', param),
  221. month: param => new AggregationOperator('month', param),
  222. second: param => new AggregationOperator('second', param),
  223. hour: param => new AggregationOperator('hour', param),
  224. week: param => new AggregationOperator('week', param),
  225. year: param => new AggregationOperator('year', param),
  226. literal: param => new AggregationOperator('literal', param),
  227. mergeObjects: param => new AggregationOperator('mergeObjects', param),
  228. objectToArray: param => new AggregationOperator('objectToArray', param),
  229. allElementsTrue: param => new AggregationOperator('allElementsTrue', param),
  230. anyElementTrue: param => new AggregationOperator('anyElementTrue', param),
  231. setDifference: param => new AggregationOperator('setDifference', param),
  232. setEquals: param => new AggregationOperator('setEquals', param),
  233. setIntersection: param => new AggregationOperator('setIntersection', param),
  234. setIsSubset: param => new AggregationOperator('setIsSubset', param),
  235. setUnion: param => new AggregationOperator('setUnion', param),
  236. concat: param => new AggregationOperator('concat', param),
  237. dateToString: param => new AggregationOperator('dateToString', param),
  238. indexOfBytes: param => new AggregationOperator('indexOfBytes', param),
  239. indexOfCP: param => new AggregationOperator('indexOfCP', param),
  240. split: param => new AggregationOperator('split', param),
  241. strLenBytes: param => new AggregationOperator('strLenBytes', param),
  242. strLenCP: param => new AggregationOperator('strLenCP', param),
  243. strcasecmp: param => new AggregationOperator('strcasecmp', param),
  244. substr: param => new AggregationOperator('substr', param),
  245. substrBytes: param => new AggregationOperator('substrBytes', param),
  246. substrCP: param => new AggregationOperator('substrCP', param),
  247. toLower: param => new AggregationOperator('toLower', param),
  248. toUpper: param => new AggregationOperator('toUpper', param),
  249. meta: param => new AggregationOperator('meta', param),
  250. addToSet: param => new AggregationOperator('addToSet', param),
  251. avg: param => new AggregationOperator('avg', param),
  252. first: param => new AggregationOperator('first', param),
  253. last: param => new AggregationOperator('last', param),
  254. max: param => new AggregationOperator('max', param),
  255. min: param => new AggregationOperator('min', param),
  256. push: param => new AggregationOperator('push', param),
  257. stdDevPop: param => new AggregationOperator('stdDevPop', param),
  258. stdDevSamp: param => new AggregationOperator('stdDevSamp', param),
  259. sum: param => new AggregationOperator('sum', param),
  260. let: param => new AggregationOperator('let', param)
  261. },
  262. project: {
  263. slice: param => new ProjectionOperator('slice', param),
  264. elemMatch: param => new ProjectionOperator('elemMatch', param)
  265. }
  266. };
  267. export class AggregationOperator {
  268. constructor(name, param) {
  269. this['$' + name] = param;
  270. }
  271. }
  272. export class ProjectionOperator {
  273. constructor(name, param) {
  274. this['$' + name] = param;
  275. }
  276. }
  277. export default Command;