Some extra query range functions

Author: mikebroberts

src/lib/support/querySupport.ts

@@ -8,18 +8,36 @@ import { SkQueryRange } from '../singleEntityOperations'
 // These ranges can be used both for table and GSI queries since the #sk expression attribute name
 // is replaced with either the table or GSI SK attribute name depending on how the query is executed
 
+export function rangeWhereSkEquals(sk: string): SkQueryRange {
+  return queryRange('#sk = :sk', {
+    ':sk': sk
+  })
+}
+
 export function rangeWhereSkGreaterThan(sk: string): SkQueryRange {
   return queryRange('#sk > :sk', {
     ':sk': sk
   })
 }
 
+export function rangeWhereSkGreaterThanOrEquals(sk: string): SkQueryRange {
+  return queryRange('#sk >= :sk', {
+    ':sk': sk
+  })
+}
+
 export function rangeWhereSkLessThan(sk: string): SkQueryRange {
   return queryRange('#sk < :sk', {
     ':sk': sk
   })
 }
 
+export function rangeWhereSkLessThanOrEquals(sk: string): SkQueryRange {
+  return queryRange('#sk <= :sk', {
+    ':sk': sk
+  })
+}
+
 export function rangeWhereSkBetween(from: string, to: string): SkQueryRange {
   return queryRange('#sk BETWEEN :from AND :to', {
     ':from': from,

test/unit/support/querySupport.test.ts

@@ -2,10 +2,22 @@ import { expect, test } from 'vitest'
 import {
   rangeWhereSkBeginsWith,
   rangeWhereSkBetween,
+  rangeWhereSkEquals,
   rangeWhereSkGreaterThan,
-  rangeWhereSkLessThan
+  rangeWhereSkGreaterThanOrEquals,
+  rangeWhereSkLessThan,
+  rangeWhereSkLessThanOrEquals
 } from '../../../src/lib/support/querySupport'
 
+test('skEquals', () => {
+  expect(rangeWhereSkEquals('FIELD#aaa')).toEqual({
+    skConditionExpressionClause: '#sk = :sk',
+    expressionAttributeValues: {
+      ':sk': 'FIELD#aaa'
+    }
+  })
+})
+
 test('skGreaterThan', () => {
   expect(rangeWhereSkGreaterThan('FIELD#aaa')).toEqual({
     skConditionExpressionClause: '#sk > :sk',
@@ -15,6 +27,15 @@ test('skGreaterThan', () => {
   })
 })
 
+test('skGreaterThanOrEquals', () => {
+  expect(rangeWhereSkGreaterThanOrEquals('FIELD#aaa')).toEqual({
+    skConditionExpressionClause: '#sk >= :sk',
+    expressionAttributeValues: {
+      ':sk': 'FIELD#aaa'
+    }
+  })
+})
+
 test('skLessThan', () => {
   expect(rangeWhereSkLessThan('FIELD#aaa')).toEqual({
     skConditionExpressionClause: '#sk < :sk',
@@ -24,6 +45,15 @@ test('skLessThan', () => {
   })
 })
 
+test('skLessThanOrEquals', () => {
+  expect(rangeWhereSkLessThanOrEquals('FIELD#aaa')).toEqual({
+    skConditionExpressionClause: '#sk <= :sk',
+    expressionAttributeValues: {
+      ':sk': 'FIELD#aaa'
+    }
+  })
+})
+
 test('skBetween', () => {
   expect(rangeWhereSkBetween('FIELD#aaa', 'FIELD#zzz')).toEqual({
     skConditionExpressionClause: '#sk BETWEEN :from AND :to',