intermediary patch

2026-06-11 23:07:11 +01:00 · 2025-08-22 18:22:27 +03:00
64 changed files with 2332 additions and 4799 deletions
@@ -7,18 +7,15 @@ jobs:
  build:
    runs-on: ubuntu-latest

-    outputs:
-      artifact: shared-libraries
-
    steps:
      - name: Checkout code
-        uses: actions/checkout@v6
+        uses: actions/checkout@v3

      - name: Cross-Compile with xgo
-        uses: crazy-max/ghaction-xgo@de82f877ff4552f03b66c146f608233849e9c3dc # v4.0.0
+        uses: crazy-max/ghaction-xgo@e22d3c8b089adba750d5a74738b8e95d96f0c991 # v3.1.0
        with:
          xgo_version: latest
-          go_version: 1.26.3
+          go_version: 1.24.0
          dest: dist
          pkg: sharedlibrary
          prefix: cosmium
@@ -28,62 +25,7 @@ jobs:
          buildvcs: true

      - name: Upload artifact
-        uses: actions/upload-artifact@v7
+        uses: actions/upload-artifact@v4
        with:
          name: shared-libraries
          path: dist/*
-
-  test:
-    needs: build
-    runs-on: ${{ matrix.os }}
-    strategy:
-      fail-fast: false
-      matrix:
-        include:
-          - os: ubuntu-latest
-            lib_ext: amd64.so
-          - os: windows-latest
-            lib_ext: amd64.dll
-          - os: macos-latest
-            lib_ext: arm64.dylib
-
-    steps:
-      - uses: actions/checkout@v6
-
-      - name: Download shared libraries
-        uses: actions/download-artifact@v8
-        with:
-          name: shared-libraries
-          path: libs
-
-      - name: Install MinGW (GCC) on Windows
-        if: runner.os == 'Windows'
-        run: choco install mingw --no-progress
-
-      - name: Build test loader (Windows)
-        if: runner.os == 'Windows'
-        run: |
-          mkdir build
-          gcc -Wall -o build/test_loader.exe sharedlibrary/tests/*.c
-
-      - name: Build test loader (Unix)
-        if: runner.os != 'Windows'
-        run: |
-          mkdir build
-          gcc -Wall -ldl -o build/test_loader sharedlibrary/tests/*.c
-
-      - name: Run test (Unix)
-        if: runner.os != 'Windows'
-        run: |
-          LIB=$(ls libs/*${{ matrix.lib_ext }} | head -n 1)
-          echo "Testing library: $LIB"
-          chmod +x build/test_loader
-          ./build/test_loader "$LIB"
-
-      - name: Run test (Windows)
-        if: runner.os == 'Windows'
-        shell: pwsh
-        run: |
-          $lib = Get-ChildItem "libs/*${{ matrix.lib_ext }}" | Select-Object -First 1
-          Write-Host "Testing library: $($lib.FullName)"
-          .\build\test_loader.exe $lib.FullName
@@ -3,7 +3,7 @@ name: goreleaser
 on:
  push:
    tags:
-      - "*"
+      - '*'

 permissions:
  contents: write
@@ -14,20 +14,20 @@ jobs:
    runs-on: ubuntu-latest
    steps:
      - name: Checkout
-        uses: actions/checkout@v6
+        uses: actions/checkout@v4
        with:
          fetch-depth: 0

      - name: Set up Go
        uses: actions/setup-go@v5
        with:
-          go-version: 1.26.3
+          go-version: 1.24.0

      - name: Cross-Compile with xgo
-        uses: crazy-max/ghaction-xgo@de82f877ff4552f03b66c146f608233849e9c3dc # v4.0.0
+        uses: crazy-max/ghaction-xgo@e22d3c8b089adba750d5a74738b8e95d96f0c991 # v3.1.0
        with:
          xgo_version: latest
-          go_version: 1.26.3
+          go_version: 1.24.0
          dest: sharedlibrary_dist
          pkg: sharedlibrary
          prefix: cosmium
@@ -9,7 +9,7 @@ SERVER_LOCATION=./cmd/server
 SHARED_LIB_LOCATION=./sharedlibrary
 SHARED_LIB_OPT=-buildmode=c-shared
 XGO_TARGETS=linux/amd64,linux/arm64,windows/amd64,windows/arm64,darwin/amd64,darwin/arm64
-GOVERSION=1.26.3
+GOVERSION=1.24.0

 DIST_DIR=dist

@@ -105,10 +105,10 @@ All mentioned arguments can also be set using environment variables:

 Cosmium supports multiple storage backends for saving, loading, and managing data at runtime.

-| Backend  | Storage Location         | Write Behavior           | Memory Usage         |
-|----------|--------------------------|--------------------------|----------------------|
-| `json` (default) | JSON file on disk 📄 | On application exit ⏳ | 🛑 More than Badger |
-| `badger`  | BadgerDB database on disk ⚡ | Immediately on write 🚀 | ✅ Less than JSON  |
+| Backend  | Storage Location         | Write Behavior           | Memory Usage         | Supports Initial JSON Load |
+|----------|--------------------------|--------------------------|----------------------|----------------------------|
+| `json` (default) | JSON file on disk 📄 | On application exit ⏳ | 🛑 More than Badger | ✅ Yes                      |
+| `badger`  | BadgerDB database on disk ⚡ | Immediately on write 🚀 | ✅ Less than JSON  | ❌ No                       |


 The `badger` backend is generally recommended as it uses less memory and writes data to disk immediately. However, if you need to load initial data from a JSON file, use the `json` backend.
@@ -23,7 +23,6 @@ const (
 func ParseFlags() ServerConfig {
 	host := flag.String("Host", "localhost", "Hostname")
 	port := flag.Int("Port", 8081, "Listen port")
-	rntbdPort := flag.Int("RntbdPort", 10000, "RNTBD listen port")
 	explorerPath := flag.String("ExplorerDir", "", "Path to cosmos-explorer files")
 	tlsCertificatePath := flag.String("Cert", "", "Hostname")
 	tlsCertificateKey := flag.String("CertKey", "", "Hostname")
@@ -36,7 +35,6 @@ func ParseFlags() ServerConfig {
 	flag.Var(logLevel, "LogLevel", fmt.Sprintf("Sets the logging level %s", logLevel.AllowedValuesList()))
 	dataStore := NewEnumValue("json", []string{DataStoreJson, DataStoreBadger})
 	flag.Var(dataStore, "DataStore", fmt.Sprintf("Sets the data store %s", dataStore.AllowedValuesList()))
-	enableRntbd := flag.Bool("ExperimentalEnableRntbd", false, "EXPERIMENTAL: Enable RNTBD (CosmosDB Direct Connection Mode)")

 	flag.Parse()
 	setFlagsFromEnvironment()
@@ -44,7 +42,6 @@ func ParseFlags() ServerConfig {
 	config := ServerConfig{}
 	config.Host = *host
 	config.Port = *port
-	config.RntbdPort = *rntbdPort
 	config.ExplorerPath = *explorerPath
 	config.TLS_CertificatePath = *tlsCertificatePath
 	config.TLS_CertificateKey = *tlsCertificateKey
@@ -55,7 +52,6 @@ func ParseFlags() ServerConfig {
 	config.AccountKey = *accountKey
 	config.LogLevel = logLevel.value
 	config.DataStore = dataStore.value
-	config.EnableRntbd = *enableRntbd

 	config.PopulateCalculatedFields()

@@ -66,7 +62,6 @@ func (c *ServerConfig) PopulateCalculatedFields() {
 	c.DatabaseAccount = c.Host
 	c.DatabaseDomain = c.Host
 	c.DatabaseEndpoint = fmt.Sprintf("https://%s:%d/", c.Host, c.Port)
-	c.RntbdEndpoint = fmt.Sprintf("rntbd://%s:%d/", c.Host, c.RntbdPort)
 	c.ExplorerBaseUrlLocation = ExplorerBaseUrlLocation

 	switch c.LogLevel {
@@ -99,6 +94,11 @@ func (c *ServerConfig) PopulateCalculatedFields() {
 			os.Exit(1)
 		}
 	}
+
+	if c.DataStore == DataStoreBadger && c.InitialDataFilePath != "" {
+		logger.ErrorLn("InitialData option is currently not supported with Badger data store")
+		os.Exit(1)
+	}
 }

 func (c *ServerConfig) ApplyDefaultsToEmptyFields() {
@@ -4,12 +4,10 @@ type ServerConfig struct {
 	DatabaseAccount  string `json:"databaseAccount"`
 	DatabaseDomain   string `json:"databaseDomain"`
 	DatabaseEndpoint string `json:"databaseEndpoint"`
-	RntbdEndpoint    string `json:"rntbdEndpoint"`
 	AccountKey       string `json:"accountKey"`

 	ExplorerPath            string `json:"explorerPath"`
 	Port                    int    `json:"port"`
-	RntbdPort               int    `json:"rntbdPort"`
 	Host                    string `json:"host"`
 	TLS_CertificatePath     string `json:"tlsCertificatePath"`
 	TLS_CertificateKey      string `json:"tlsCertificateKey"`
@@ -19,7 +17,6 @@ type ServerConfig struct {
 	DisableTls              bool   `json:"disableTls"`
 	LogLevel                string `json:"logLevel"`
 	ExplorerBaseUrlLocation string `json:"explorerBaseUrlLocation"`
-	EnableRntbd             bool   `json:"enableRntbd"`

 	DataStore string `json:"dataStore"`
 }
@@ -5,7 +5,6 @@ import (
 	"net/http"

 	"github.com/gin-gonic/gin"
-	"github.com/pikami/cosmium/api/headers"
 	"github.com/pikami/cosmium/internal/constants"
 	"github.com/pikami/cosmium/internal/datastore"
 )
@@ -17,7 +16,7 @@ func (h *Handlers) GetAllCollections(c *gin.Context) {
 	if status == datastore.StatusOk {
 		database, _ := h.dataStore.GetDatabase(databaseId)

-		c.Header(headers.ItemCount, fmt.Sprintf("%d", len(collections)))
+		c.Header("x-ms-item-count", fmt.Sprintf("%d", len(collections)))
 		c.IndentedJSON(http.StatusOK, gin.H{
 			"_rid":                database.ResourceID,
 			"DocumentCollections": collections,
@@ -5,7 +5,6 @@ import (
 	"net/http"

 	"github.com/gin-gonic/gin"
-	"github.com/pikami/cosmium/api/headers"
 	"github.com/pikami/cosmium/internal/constants"
 	"github.com/pikami/cosmium/internal/datastore"
 )
@@ -13,7 +12,7 @@ import (
 func (h *Handlers) GetAllDatabases(c *gin.Context) {
 	databases, status := h.dataStore.GetAllDatabases()
 	if status == datastore.StatusOk {
-		c.Header(headers.ItemCount, fmt.Sprintf("%d", len(databases)))
+		c.Header("x-ms-item-count", fmt.Sprintf("%d", len(databases)))
 		c.IndentedJSON(http.StatusOK, gin.H{
 			"_rid":      "",
 			"Databases": databases,
@@ -9,9 +9,7 @@ import (
 	jsonpatch "github.com/cosmiumdev/json-patch/v5"
 	"github.com/gin-gonic/gin"
 	apimodels "github.com/pikami/cosmium/api/api_models"
-	"github.com/pikami/cosmium/api/headers"
 	"github.com/pikami/cosmium/internal/constants"
-	continuationtoken "github.com/pikami/cosmium/internal/continuation_token"
 	"github.com/pikami/cosmium/internal/converters"
 	"github.com/pikami/cosmium/internal/datastore"
 	"github.com/pikami/cosmium/internal/logger"
@@ -28,7 +26,7 @@ func (h *Handlers) GetAllDocuments(c *gin.Context) {
 	if status == datastore.StatusOk {
 		collection, _ := h.dataStore.GetCollection(databaseId, collectionId)

-		c.Header(headers.ItemCount, fmt.Sprintf("%d", len(documents)))
+		c.Header("x-ms-item-count", fmt.Sprintf("%d", len(documents)))
 		c.IndentedJSON(http.StatusOK, gin.H{
 			"_rid":      collection.ID,
 			"Documents": documents,
@@ -47,9 +45,6 @@ func (h *Handlers) GetDocument(c *gin.Context) {

 	document, status := h.dataStore.GetDocument(databaseId, collectionId, documentId)
 	if status == datastore.StatusOk {
-		if etag, ok := document["_etag"].(string); ok {
-			c.Header(headers.ETag, etag)
-		}
 		c.IndentedJSON(http.StatusOK, document)
 		return
 	}
@@ -93,25 +88,7 @@ func (h *Handlers) ReplaceDocument(c *gin.Context) {
 		return
 	}

-	existingDocument, status := h.dataStore.GetDocument(databaseId, collectionId, documentId)
-	if status == datastore.StatusNotFound {
-		c.IndentedJSON(http.StatusNotFound, constants.NotFoundResponse)
-		return
-	}
-	if status != datastore.StatusOk {
-		c.IndentedJSON(http.StatusInternalServerError, constants.UnknownErrorResponse)
-		return
-	}
-
-	if ifMatch := c.GetHeader(headers.IfMatch); ifMatch != "" {
-		if existingDocument["_etag"] != ifMatch {
-			c.Header(headers.ErrorCode, "PreconditionFailed")
-			c.JSON(http.StatusPreconditionFailed, constants.PreconditionFailedResponse)
-			return
-		}
-	}
-
-	status = h.dataStore.DeleteDocument(databaseId, collectionId, documentId)
+	status := h.dataStore.DeleteDocument(databaseId, collectionId, documentId)
 	if status == datastore.StatusNotFound {
 		c.IndentedJSON(http.StatusNotFound, constants.NotFoundResponse)
 		return
@@ -212,7 +189,7 @@ func (h *Handlers) DocumentsPost(c *gin.Context) {
 	collectionId := c.Param("collId")

 	// Handle batch requests
-	isBatchRequest, _ := strconv.ParseBool(c.GetHeader(headers.IsBatchRequest))
+	isBatchRequest, _ := strconv.ParseBool(c.GetHeader("x-ms-cosmos-is-batch-request"))
 	if isBatchRequest {
 		h.handleBatchRequest(c)
 		return
@@ -224,17 +201,8 @@ func (h *Handlers) DocumentsPost(c *gin.Context) {
 		return
 	}

-	// Handle query plan requests
-	isQueryPlanRequest, _ := strconv.ParseBool(c.GetHeader(headers.IsQueryPlanRequest))
-	if isQueryPlanRequest {
-		c.IndentedJSON(http.StatusOK, constants.QueryPlanResponse)
-		return
-	}
-
-	// Handle query requests
-	isQueryRequest, _ := strconv.ParseBool(c.GetHeader(headers.IsQuery))
-	isQueryRequestAltHeader, _ := strconv.ParseBool(c.GetHeader(headers.Query))
-	if isQueryRequest || isQueryRequestAltHeader {
+	query := requestBody["query"]
+	if query != nil {
 		h.handleDocumentQuery(c, requestBody)
 		return
 	}
@@ -244,7 +212,7 @@ func (h *Handlers) DocumentsPost(c *gin.Context) {
 		return
 	}

-	isUpsert, _ := strconv.ParseBool(c.GetHeader(headers.IsUpsert))
+	isUpsert, _ := strconv.ParseBool(c.GetHeader("x-ms-documentdb-is-upsert"))
 	if isUpsert {
 		h.dataStore.DeleteDocument(databaseId, collectionId, requestBody["id"].(string))
 	}
@@ -279,55 +247,30 @@ func (h *Handlers) handleDocumentQuery(c *gin.Context, requestBody map[string]in
 	databaseId := c.Param("databaseId")
 	collectionId := c.Param("collId")

+	if c.GetHeader("x-ms-cosmos-is-query-plan-request") != "" {
+		c.IndentedJSON(http.StatusOK, constants.QueryPlanResponse)
+		return
+	}
+
 	var queryParameters map[string]interface{}
 	if paramsArray, ok := requestBody["parameters"].([]interface{}); ok {
 		queryParameters = parametersToMap(paramsArray)
 	}

-	collection, collectionStatus := h.dataStore.GetCollection(databaseId, collectionId)
-	if collectionStatus == datastore.StatusNotFound {
-		c.IndentedJSON(http.StatusNotFound, constants.NotFoundResponse)
-		return
-	}
-
-	if collectionStatus != datastore.StatusOk {
-		c.IndentedJSON(http.StatusInternalServerError, constants.UnknownErrorResponse)
-		return
-	}
-
-	continuationToken := continuationtoken.GenerateDefault(collection.ResourceID)
-	continuationTokenHeader := c.GetHeader(headers.ContinuationToken)
-	if continuationTokenHeader != "" {
-		continuationToken = continuationtoken.FromString(continuationTokenHeader)
-	}
-
-	pageMaxItemCount, maxItemCountError := strconv.Atoi(c.GetHeader(headers.MaxItemCount))
-	if maxItemCountError != nil {
-		pageMaxItemCount = 1000
-	}
-
 	queryText := requestBody["query"].(string)
-	executeQueryResult, status := h.executeQueryDocuments(
-		databaseId, collectionId, queryText, queryParameters, pageMaxItemCount, continuationToken.Token.TotalResults)
+	docs, status := h.executeQueryDocuments(databaseId, collectionId, queryText, queryParameters)
 	if status != datastore.StatusOk {
 		// TODO: Currently we return everything if the query fails
-		logger.Infof("Query failed: %s", queryText)
 		h.GetAllDocuments(c)
 		return
 	}

-	resultCount := len(executeQueryResult.Rows)
-	if executeQueryResult.HasMorePages {
-		nextContinuationToken := continuationtoken.Generate(
-			collection.ResourceID, continuationToken.Token.PageIndex+1, continuationToken.Token.TotalResults+resultCount)
-		c.Header(headers.ContinuationToken, nextContinuationToken.ToString())
-	}
-
-	c.Header(headers.ItemCount, fmt.Sprintf("%d", resultCount))
+	collection, _ := h.dataStore.GetCollection(databaseId, collectionId)
+	c.Header("x-ms-item-count", fmt.Sprintf("%d", len(docs)))
 	c.IndentedJSON(http.StatusOK, gin.H{
 		"_rid":      collection.ResourceID,
-		"Documents": executeQueryResult.Rows,
-		"_count":    resultCount,
+		"Documents": docs,
+		"_count":    len(docs),
 	})
 }

@@ -429,23 +372,16 @@ func dataStoreStatusToResponseCode(status datastore.DataStoreStatus) int {
 	}
 }

-func (h *Handlers) executeQueryDocuments(
-	databaseId string,
-	collectionId string,
-	query string,
-	queryParameters map[string]interface{},
-	pageMaxItemCount int,
-	pageCursor int,
-) (memoryexecutor.ExecuteQueryResult, datastore.DataStoreStatus) {
+func (h *Handlers) executeQueryDocuments(databaseId string, collectionId string, query string, queryParameters map[string]interface{}) ([]memoryexecutor.RowType, datastore.DataStoreStatus) {
 	parsedQuery, err := nosql.Parse("", []byte(query))
 	if err != nil {
 		logger.Errorf("Failed to parse query: %s\nerr: %v", query, err)
-		return memoryexecutor.ExecuteQueryResult{}, datastore.BadRequest
+		return nil, datastore.BadRequest
 	}

 	allDocumentsIterator, status := h.dataStore.GetDocumentIterator(databaseId, collectionId)
 	if status != datastore.StatusOk {
-		return memoryexecutor.ExecuteQueryResult{}, status
+		return nil, status
 	}
 	defer allDocumentsIterator.Close()

@@ -453,8 +389,8 @@ func (h *Handlers) executeQueryDocuments(

 	if typedQuery, ok := parsedQuery.(parsers.SelectStmt); ok {
 		typedQuery.Parameters = queryParameters
-		return memoryexecutor.ExecuteQuery(typedQuery, rowsIterator, pageCursor, pageMaxItemCount), datastore.StatusOk
+		return memoryexecutor.ExecuteQuery(typedQuery, rowsIterator), datastore.StatusOk
 	}

-	return memoryexecutor.ExecuteQueryResult{}, datastore.BadRequest
+	return nil, datastore.BadRequest
 }
@@ -6,7 +6,6 @@ import (

 	"github.com/gin-gonic/gin"
 	"github.com/pikami/cosmium/api/config"
-	"github.com/pikami/cosmium/api/headers"
 	"github.com/pikami/cosmium/internal/authentication"
 	"github.com/pikami/cosmium/internal/logger"
 )
@@ -23,8 +22,8 @@ func Authentication(config *config.ServerConfig) gin.HandlerFunc {
 		resourceType := urlToResourceType(requestUrl)
 		resourceId := requestToResourceId(c)

-		authHeader := c.Request.Header.Get(headers.Authorization)
-		date := c.Request.Header.Get(headers.XDate)
+		authHeader := c.Request.Header.Get("authorization")
+		date := c.Request.Header.Get("x-ms-date")
 		expectedSignature := authentication.GenerateSignature(
 			c.Request.Method, resourceType, resourceId, date, config.AccountKey)

@@ -86,7 +85,7 @@ func requestToResourceId(c *gin.Context) string {
 		resourceId += "/udfs/" + udfId
 	}

-	isFeed := c.Request.Header.Get(headers.AIM) == "Incremental Feed"
+	isFeed := c.Request.Header.Get("A-Im") == "Incremental Feed"
 	if resourceType == "pkranges" && isFeed {
 		resourceId = collId
 	}
@@ -4,11 +4,10 @@ import (
 	"net/http"

 	"github.com/gin-gonic/gin"
-	"github.com/pikami/cosmium/api/headers"
 )

 func GetOffers(c *gin.Context) {
-	c.Header(headers.ItemCount, "0")
+	c.Header("x-ms-item-count", "0")
 	c.IndentedJSON(http.StatusOK, gin.H{
 		"_rid":   "",
 		"_count": 0,
@@ -5,7 +5,6 @@ import (
 	"net/http"

 	"github.com/gin-gonic/gin"
-	"github.com/pikami/cosmium/api/headers"
 	"github.com/pikami/cosmium/internal/constants"
 	"github.com/pikami/cosmium/internal/datastore"
 	"github.com/pikami/cosmium/internal/resourceid"
@@ -15,18 +14,18 @@ func (h *Handlers) GetPartitionKeyRanges(c *gin.Context) {
 	databaseId := c.Param("databaseId")
 	collectionId := c.Param("collId")

-	if c.Request.Header.Get(headers.IfNoneMatch) != "" {
+	if c.Request.Header.Get("if-none-match") != "" {
 		c.AbortWithStatus(http.StatusNotModified)
 		return
 	}

 	partitionKeyRanges, status := h.dataStore.GetPartitionKeyRanges(databaseId, collectionId)
 	if status == datastore.StatusOk {
-		c.Header(headers.ETag, "\"420\"")
-		c.Header(headers.LSN, "420")
-		c.Header(headers.CosmosLsn, "420")
-		c.Header(headers.GlobalCommittedLsn, "420")
-		c.Header(headers.ItemCount, fmt.Sprintf("%d", len(partitionKeyRanges)))
+		c.Header("etag", "\"420\"")
+		c.Header("lsn", "420")
+		c.Header("x-ms-cosmos-llsn", "420")
+		c.Header("x-ms-global-committed-lsn", "420")
+		c.Header("x-ms-item-count", fmt.Sprintf("%d", len(partitionKeyRanges)))

 		collectionRid := collectionId
 		collection, _ := h.dataStore.GetCollection(databaseId, collectionId)
@@ -3,7 +3,6 @@ package handlers
 import (
 	"fmt"
 	"net/http"
-	"strings"

 	"github.com/gin-gonic/gin"
 )
@@ -42,46 +41,3 @@ func (h *Handlers) GetServerInfo(c *gin.Context) {
 		"queryEngineConfiguration": "{\"allowNewKeywords\":true,\"maxJoinsPerSqlQuery\":10,\"maxQueryRequestTimeoutFraction\":0.9,\"maxSqlQueryInputLength\":524288,\"maxUdfRefPerSqlQuery\":10,\"queryMaxInMemorySortDocumentCount\":-1000,\"spatialMaxGeometryPointCount\":256,\"sqlAllowNonFiniteNumbers\":false,\"sqlDisableOptimizationFlags\":0,\"enableSpatialIndexing\":true,\"maxInExpressionItemsCount\":2147483647,\"maxLogicalAndPerSqlQuery\":2147483647,\"maxLogicalOrPerSqlQuery\":2147483647,\"maxSpatialQueryCells\":2147483647,\"sqlAllowAggregateFunctions\":true,\"sqlAllowGroupByClause\":true,\"sqlAllowLike\":true,\"sqlAllowSubQuery\":true,\"sqlAllowScalarSubQuery\":true,\"sqlAllowTop\":true}",
 	})
 }
-
-type Address struct {
-	IsPrimary                     bool   `json:"isPrimary"`
-	PhyscialUri                   string `json:"physcialUri"`
-	IsAuxiliary                   bool   `json:"isAuxiliary"`
-	PartitionTargetReplicaSetSize int    `json:"partitionTargetReplicaSetSize"`
-	Protocol                      string `json:"protocol"`
-	PartitionKeyRangeId           string `json:"partitionKeyRangeId"`
-	PartitionIndex                string `json:"partitionIndex"`
-}
-
-func (h *Handlers) GetAddresses(c *gin.Context) {
-	addresses := []Address{}
-
-	if h.config.EnableRntbd {
-		addresses = append(addresses, Address{
-			IsPrimary:                     true,
-			PhyscialUri:                   h.config.RntbdEndpoint,
-			IsAuxiliary:                   false,
-			PartitionTargetReplicaSetSize: 1,
-			Protocol:                      "rntbd",
-			PartitionKeyRangeId:           "0",
-			PartitionIndex:                "0@0",
-		})
-	}
-
-	if !strings.Contains(c.Request.RequestURI, "protocol%20eq%20rntbd") {
-		addresses = append(addresses, Address{
-			IsPrimary:                     true,
-			PhyscialUri:                   h.config.DatabaseEndpoint,
-			IsAuxiliary:                   false,
-			PartitionTargetReplicaSetSize: 1,
-			Protocol:                      "https",
-			PartitionKeyRangeId:           "0",
-			PartitionIndex:                "0@0",
-		})
-	}
-
-	c.IndentedJSON(http.StatusOK, gin.H{
-		"Addresss": addresses,
-		"_count":   len(addresses),
-	})
-}
@@ -5,7 +5,6 @@ import (
 	"net/http"

 	"github.com/gin-gonic/gin"
-	"github.com/pikami/cosmium/api/headers"
 	"github.com/pikami/cosmium/internal/constants"
 	"github.com/pikami/cosmium/internal/datastore"
 )
@@ -17,7 +16,7 @@ func (h *Handlers) GetAllStoredProcedures(c *gin.Context) {
 	sps, status := h.dataStore.GetAllStoredProcedures(databaseId, collectionId)

 	if status == datastore.StatusOk {
-		c.Header(headers.ItemCount, fmt.Sprintf("%d", len(sps)))
+		c.Header("x-ms-item-count", fmt.Sprintf("%d", len(sps)))
 		c.IndentedJSON(http.StatusOK, gin.H{"_rid": "", "StoredProcedures": sps, "_count": len(sps)})
 		return
 	}
@@ -5,7 +5,6 @@ import (
 	"net/http"

 	"github.com/gin-gonic/gin"
-	"github.com/pikami/cosmium/api/headers"
 	"github.com/pikami/cosmium/internal/constants"
 	"github.com/pikami/cosmium/internal/datastore"
 )
@@ -17,7 +16,7 @@ func (h *Handlers) GetAllTriggers(c *gin.Context) {
 	triggers, status := h.dataStore.GetAllTriggers(databaseId, collectionId)

 	if status == datastore.StatusOk {
-		c.Header(headers.ItemCount, fmt.Sprintf("%d", len(triggers)))
+		c.Header("x-ms-item-count", fmt.Sprintf("%d", len(triggers)))
 		c.IndentedJSON(http.StatusOK, gin.H{"_rid": "", "Triggers": triggers, "_count": len(triggers)})
 		return
 	}
@@ -5,7 +5,6 @@ import (
 	"net/http"

 	"github.com/gin-gonic/gin"
-	"github.com/pikami/cosmium/api/headers"
 	"github.com/pikami/cosmium/internal/constants"
 	"github.com/pikami/cosmium/internal/datastore"
 )
@@ -17,7 +16,7 @@ func (h *Handlers) GetAllUserDefinedFunctions(c *gin.Context) {
 	udfs, status := h.dataStore.GetAllUserDefinedFunctions(databaseId, collectionId)

 	if status == datastore.StatusOk {
-		c.Header(headers.ItemCount, fmt.Sprintf("%d", len(udfs)))
+		c.Header("x-ms-item-count", fmt.Sprintf("%d", len(udfs)))
 		c.IndentedJSON(http.StatusOK, gin.H{"_rid": "", "UserDefinedFunctions": udfs, "_count": len(udfs)})
 		return
 	}
@@ -1,30 +0,0 @@
-package headers
-
-const (
-	AIM                = "A-Im"
-	Authorization      = "authorization"
-	CosmosLsn          = "x-ms-cosmos-llsn"
-	ErrorCode          = "x-ms-error-code"
-	ETag               = "etag"
-	GlobalCommittedLsn = "x-ms-global-committed-lsn"
-	IfMatch            = "if-match"
-	IfNoneMatch        = "if-none-match"
-	IsBatchRequest     = "x-ms-cosmos-is-batch-request"
-	IsQueryPlanRequest = "x-ms-cosmos-is-query-plan-request"
-	IsUpsert           = "x-ms-documentdb-is-upsert"
-	ItemCount          = "x-ms-item-count"
-	LSN                = "lsn"
-	XDate              = "x-ms-date"
-	MaxItemCount       = "x-ms-max-item-count"
-	ContinuationToken  = "x-ms-continuation"
-
-	// Kinda retarded, but what can I do ¯\_(ツ)_/¯
-	IsQuery = "x-ms-documentdb-isquery" // Sent from python sdk and web explorer
-	Query   = "x-ms-documentdb-query"   // Sent from Go sdk
-
-	// I kinda don't use these, but I've seen them in the wild xd
-	SupportedCapabilities     = "x-ms-cosmos-sdk-supportedcapabilities"
-	ClientRetryAttemptCount   = "x-ms-client-retry-attempt-count"
-	RemainingTimeInMsOnClient = "x-ms-remaining-time-in-ms-on-client"
-	ConsistencyLevel          = "x-ms-consistency-level"
-)
@@ -31,7 +31,6 @@ func (s *ApiServer) CreateRouter(dataStore datastore.DataStore) {

 	router := gin.Default(func(e *gin.Engine) {
 		e.RedirectTrailingSlash = false
-		e.RemoveExtraSlash = true
 	})

 	if s.config.LogLevel == "debug" {
@@ -80,7 +79,6 @@ func (s *ApiServer) CreateRouter(dataStore datastore.DataStore) {

 	router.GET("/offers", handlers.GetOffers)
 	router.GET("/", routeHandlers.GetServerInfo)
-	router.GET("//addresses", routeHandlers.GetAddresses)

 	router.GET("/cosmium/export", routeHandlers.CosmiumExport)

@@ -121,26 +121,5 @@ func Test_Collections(t *testing.T) {
 				panic(err)
 			}
 		})
-
-		t.Run("Should delete collection with exactly matching name", func(t *testing.T) {
-			ts.DataStore.CreateCollection(testDatabaseName, datastore.Collection{
-				ID: testCollectionName + "extra",
-			})
-			ts.DataStore.CreateCollection(testDatabaseName, datastore.Collection{
-				ID: testCollectionName,
-			})
-
-			collectionResponse, err := databaseClient.NewContainer(testCollectionName)
-			assert.Nil(t, err)
-
-			readResponse, err := collectionResponse.Delete(context.TODO(), &azcosmos.DeleteContainerOptions{})
-			assert.Nil(t, err)
-			assert.Equal(t, readResponse.RawResponse.StatusCode, http.StatusNoContent)
-
-			collections, status := ts.DataStore.GetAllCollections(testDatabaseName)
-			assert.Equal(t, status, datastore.StatusOk)
-			assert.Len(t, collections, 1)
-			assert.Equal(t, collections[0].ID, testCollectionName+"extra")
-		})
 	})
 }
@@ -109,26 +109,5 @@ func Test_Databases(t *testing.T) {
 				panic(err)
 			}
 		})
-
-		t.Run("Should delete database with exactly matching name", func(t *testing.T) {
-			ts.DataStore.CreateDatabase(datastore.Database{
-				ID: testDatabaseName + "extra",
-			})
-			ts.DataStore.CreateDatabase(datastore.Database{
-				ID: testDatabaseName,
-			})
-
-			databaseResponse, err := client.NewDatabase(testDatabaseName)
-			assert.Nil(t, err)
-
-			readResponse, err := databaseResponse.Delete(context.TODO(), &azcosmos.DeleteDatabaseOptions{})
-			assert.Nil(t, err)
-			assert.Equal(t, readResponse.RawResponse.StatusCode, http.StatusNoContent)
-
-			dbs, status := ts.DataStore.GetAllDatabases()
-			assert.Equal(t, status, datastore.StatusOk)
-			assert.Len(t, dbs, 1)
-			assert.Equal(t, dbs[0].ID, testDatabaseName+"extra")
-		})
 	})
 }
@@ -1,47 +0,0 @@
-package tests_test
-
-import (
-	"testing"
-
-	"github.com/Azure/azure-sdk-for-go/sdk/data/azcosmos"
-)
-
-func Test_Documents_ArrayContains(t *testing.T) {
-	presets := []testPreset{PresetJsonStore, PresetBadgerStore}
-
-	runTestsWithPresets(t, "Test_Documents_ArrayContains", presets, func(t *testing.T, ts *TestServer, client *azcosmos.Client) {
-		collectionClient := documents_InitializeDb(t, ts)
-
-		t.Run("Should execute ARRAY_CONTAINS() without partial match argument", func(t *testing.T) {
-			testCosmosQuery(t, collectionClient,
-				`SELECT VALUE ARRAY_CONTAINS(["apple", "banana", "cherry"], "banana") FROM c ORDER BY c.id`,
-				nil,
-				[]interface{}{true, true},
-			)
-		})
-
-		t.Run("Should execute ARRAY_CONTAINS() returning false for missing item", func(t *testing.T) {
-			testCosmosQuery(t, collectionClient,
-				`SELECT VALUE ARRAY_CONTAINS(["apple", "banana", "cherry"], "grape") FROM c ORDER BY c.id`,
-				nil,
-				[]interface{}{false, false},
-			)
-		})
-
-		t.Run("Should execute ARRAY_CONTAINS() with object full match", func(t *testing.T) {
-			testCosmosQuery(t, collectionClient,
-				`SELECT VALUE ARRAY_CONTAINS([{"name": "apple", "color": "red"}], {"name": "apple"}) FROM c ORDER BY c.id`,
-				nil,
-				[]interface{}{false, false},
-			)
-		})
-
-		t.Run("Should execute ARRAY_CONTAINS() with object partial match", func(t *testing.T) {
-			testCosmosQuery(t, collectionClient,
-				`SELECT VALUE ARRAY_CONTAINS([{"name": "apple", "color": "red"}], {"name": "apple"}, true) FROM c ORDER BY c.id`,
-				nil,
-				[]interface{}{true, true},
-			)
-		})
-	})
-}
@@ -1,73 +0,0 @@
-package tests_test
-
-import (
-	"fmt"
-	"testing"
-
-	"github.com/Azure/azure-sdk-for-go/sdk/data/azcosmos"
-	"github.com/pikami/cosmium/api/config"
-	"github.com/pikami/cosmium/internal/datastore"
-	"github.com/stretchr/testify/assert"
-)
-
-func documents_InitializeSingleDocumentDb(t *testing.T, ts *TestServer) *azcosmos.ContainerClient {
-	ts.DataStore.CreateDatabase(datastore.Database{ID: testDatabaseName})
-	ts.DataStore.CreateCollection(testDatabaseName, datastore.Collection{
-		ID: testCollectionName,
-		PartitionKey: struct {
-			Paths   []string "json:\"paths\""
-			Kind    string   "json:\"kind\""
-			Version int      "json:\"Version\""
-		}{
-			Paths: []string{"/pk"},
-		},
-	})
-	ts.DataStore.CreateDocument(testDatabaseName, testCollectionName, map[string]interface{}{"id": "regexmatch-test", "pk": "regexmatch-test"})
-
-	client, err := azcosmos.NewClientFromConnectionString(
-		fmt.Sprintf("AccountEndpoint=%s;AccountKey=%s", ts.URL, config.DefaultAccountKey),
-		&azcosmos.ClientOptions{},
-	)
-	assert.Nil(t, err)
-
-	collectionClient, err := client.NewContainer(testDatabaseName, testCollectionName)
-	assert.Nil(t, err)
-
-	return collectionClient
-}
-
-func Test_Documents_RegexMatch(t *testing.T) {
-	presets := []testPreset{PresetJsonStore, PresetBadgerStore}
-
-	runTestsWithPresets(t, "Test_Documents_RegexMatch", presets, func(t *testing.T, ts *TestServer, client *azcosmos.Client) {
-		collectionClient := documents_InitializeSingleDocumentDb(t, ts)
-
-		t.Run("Should execute REGEXMATCH()", func(t *testing.T) {
-			testCosmosQuery(t, collectionClient,
-				`SELECT VALUE {
-					noModifiers: REGEXMATCH("abcd", "ABC"),
-					caseInsensitive: REGEXMATCH("abcd", "ABC", "i"),
-					wildcardCharacter: REGEXMATCH("abcd", "ab.", ""),
-					ignoreWhiteSpace: REGEXMATCH("abcd", "ab c", "x"),
-					caseInsensitiveAndIgnoreWhiteSpace: REGEXMATCH("abcd", "aB c", "ix"),
-					containNumberBetweenZeroAndNine: REGEXMATCH("03a", "[0-9]"),
-					containPrefix: REGEXMATCH("salt3824908", "salt{1}"),
-					containsFiveLetterWordStartingWithS: REGEXMATCH("shame", "s....", "i")
-				}`,
-				nil,
-				[]interface{}{
-					map[string]interface{}{
-						"noModifiers":                         false,
-						"caseInsensitive":                     true,
-						"wildcardCharacter":                   true,
-						"ignoreWhiteSpace":                    true,
-						"caseInsensitiveAndIgnoreWhiteSpace":  true,
-						"containNumberBetweenZeroAndNine":     true,
-						"containPrefix":                       true,
-						"containsFiveLetterWordStartingWithS": true,
-					},
-				},
-			)
-		})
-	})
-}
@@ -5,7 +5,6 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
-	"io"
 	"net/http"
 	"reflect"
 	"sync"
@@ -15,7 +14,6 @@ import (
 	"github.com/Azure/azure-sdk-for-go/sdk/azcore"
 	"github.com/Azure/azure-sdk-for-go/sdk/data/azcosmos"
 	"github.com/pikami/cosmium/api/config"
-	continuationtoken "github.com/pikami/cosmium/internal/continuation_token"
 	"github.com/pikami/cosmium/internal/datastore"
 	"github.com/stretchr/testify/assert"
 )
@@ -380,80 +378,6 @@ func Test_Documents(t *testing.T) {
 		})
 	})

-	runTestsWithPresets(t, "Test_Documents_ETag_OptimisticConcurrency", presets, func(t *testing.T, ts *TestServer, client *azcosmos.Client) {
-		collectionClient := documents_InitializeDb(t, ts)
-
-		t.Run("Should fail replace with incorrect etag", func(t *testing.T) {
-			context := context.TODO()
-
-			item := map[string]interface{}{"id": "12345", "pk": "123", "isCool": true}
-			bytes, err := json.Marshal(item)
-			assert.Nil(t, err)
-
-			wrongETag := azcore.ETag("\"incorrect-etag\"")
-			_, err = collectionClient.ReplaceItem(
-				context,
-				azcosmos.PartitionKey{},
-				"12345",
-				bytes,
-				&azcosmos.ItemOptions{IfMatchEtag: &wrongETag},
-			)
-			assert.NotNil(t, err)
-
-			var respErr *azcore.ResponseError
-			if errors.As(err, &respErr) {
-				assert.Equal(t, http.StatusPreconditionFailed, respErr.StatusCode)
-				assert.Equal(t, "PreconditionFailed", respErr.RawResponse.Header.Get("x-ms-error-code"))
-
-				responseBody, readErr := io.ReadAll(respErr.RawResponse.Body)
-				assert.Nil(t, readErr)
-				assert.JSONEq(t,
-					`{"code":"PreconditionFailed","message":"Operation cannot be performed because one of the specified precondition is not met."}`,
-					string(responseBody),
-				)
-			} else {
-				panic(err)
-			}
-
-			document, status := ts.DataStore.GetDocument(testDatabaseName, testCollectionName, "12345")
-			assert.Equal(t, datastore.StatusOk, status)
-			assert.Equal(t, false, document["isCool"])
-		})
-
-		t.Run("Should replace with correct etag", func(t *testing.T) {
-			context := context.TODO()
-
-			readResponse, err := collectionClient.ReadItem(context, azcosmos.PartitionKey{}, "12345", nil)
-			assert.Nil(t, err)
-			assert.NotEmpty(t, readResponse.ETag)
-
-			var item map[string]interface{}
-			err = json.Unmarshal(readResponse.Value, &item)
-			assert.Nil(t, err)
-			assert.Equal(t, string(readResponse.ETag), item["_etag"])
-
-			item["pk"] = "999"
-			item["isCool"] = true
-			bytes, err := json.Marshal(item)
-			assert.Nil(t, err)
-
-			etag := readResponse.ETag
-			_, err = collectionClient.ReplaceItem(
-				context,
-				azcosmos.PartitionKey{},
-				"12345",
-				bytes,
-				&azcosmos.ItemOptions{IfMatchEtag: &etag},
-			)
-			assert.Nil(t, err)
-
-			document, status := ts.DataStore.GetDocument(testDatabaseName, testCollectionName, "12345")
-			assert.Equal(t, datastore.StatusOk, status)
-			assert.Equal(t, "999", document["pk"])
-			assert.Equal(t, true, document["isCool"])
-		})
-	})
-
 	runTestsWithPresets(t, "Test_Documents_TransactionalBatch", presets, func(t *testing.T, ts *TestServer, client *azcosmos.Client) {
 		collectionClient := documents_InitializeDb(t, ts)

@@ -501,7 +425,7 @@ func Test_Documents(t *testing.T) {
 			assert.Equal(t, int32(http.StatusNoContent), operationResponse.StatusCode)

 			_, status := ts.DataStore.GetDocument(testDatabaseName, testCollectionName, "12345")
-			assert.Equal(t, datastore.StatusNotFound, status)
+			assert.Equal(t, datastore.StatusNotFound, int(status))
 		})

 		t.Run("Should execute REPLACE transactional batch", func(t *testing.T) {
@@ -588,46 +512,4 @@ func Test_Documents(t *testing.T) {
 			assert.Equal(t, "67890", itemResponseBody["id"])
 		})
 	})
-
-	runTestsWithPresets(t, "Test_Documents_With_Continuation_Token", presets, func(t *testing.T, ts *TestServer, client *azcosmos.Client) {
-		collectionClient := documents_InitializeDb(t, ts)
-
-		t.Run("Should query document with continuation token", func(t *testing.T) {
-			context := context.TODO()
-			pager := collectionClient.NewQueryItemsPager(
-				"SELECT c.id, c[\"pk\"] FROM c ORDER BY c.id",
-				azcosmos.PartitionKey{},
-				&azcosmos.QueryOptions{
-					PageSizeHint: 1,
-				})
-
-			assert.True(t, pager.More())
-
-			firstResponse, err := pager.NextPage(context)
-			assert.Nil(t, err)
-			assert.Equal(t, 1, len(firstResponse.Items))
-			var firstItem map[string]interface{}
-			err = json.Unmarshal(firstResponse.Items[0], &firstItem)
-			assert.Nil(t, err)
-			assert.Equal(t, "12345", firstItem["id"])
-			assert.Equal(t, "123", firstItem["pk"])
-
-			firstContinuationToken := continuationtoken.FromString(*firstResponse.ContinuationToken)
-			assert.Equal(t, 1, firstContinuationToken.Token.PageIndex)
-			assert.Equal(t, 1, firstContinuationToken.Token.TotalResults)
-
-			assert.True(t, pager.More())
-			secondResponse, err := pager.NextPage(context)
-			assert.Nil(t, err)
-			assert.Equal(t, 1, len(secondResponse.Items))
-			var secondItem map[string]interface{}
-			err = json.Unmarshal(secondResponse.Items[0], &secondItem)
-			assert.Nil(t, err)
-			assert.Equal(t, "67890", secondItem["id"])
-			assert.Equal(t, "456", secondItem["pk"])
-			assert.Nil(t, secondResponse.ContinuationToken)
-
-			assert.False(t, pager.More())
-		})
-	})
 }
@@ -8,7 +8,6 @@ import (
 	"time"

 	"github.com/pikami/cosmium/api/config"
-	"github.com/pikami/cosmium/api/headers"
 	"github.com/pikami/cosmium/internal/authentication"
 	"github.com/stretchr/testify/assert"
 )
@@ -27,8 +26,8 @@ func Test_Documents_Read_Trailing_Slash(t *testing.T) {
 		signature := authentication.GenerateSignature("GET", "docs", path, date, config.DefaultAccountKey)
 		httpClient := &http.Client{}
 		req, _ := http.NewRequest("GET", testUrl, nil)
-		req.Header.Add(headers.XDate, date)
-		req.Header.Add(headers.Authorization, "sig="+url.QueryEscape(signature))
+		req.Header.Add("x-ms-date", date)
+		req.Header.Add("authorization", "sig="+url.QueryEscape(signature))
 		res, err := httpClient.Do(req)

 		assert.Nil(t, err)
@@ -1,63 +0,0 @@
-package main
-
-import (
-	"encoding/hex"
-	"encoding/json"
-	"flag"
-	"fmt"
-
-	"github.com/pikami/cosmium/internal/rntbd"
-)
-
-func main() {
-	input := flag.String("input", "", "Input hex string")
-	isResponse := flag.Bool("response", false, "Is response")
-	flag.Parse()
-
-	data, err := hex.DecodeString(*input)
-	if err != nil {
-		fmt.Printf("Error decoding hex string: %v\n", err)
-		return
-	}
-
-	frame, err := rntbd.ParseFrame(data, *isResponse)
-	if err != nil {
-		fmt.Printf("Error parsing frame: %v\n", err)
-		return
-	}
-
-	fmt.Printf("Activity ID: %s\n", hex.EncodeToString(frame.ActivityId))
-	fmt.Printf("Resource Type: %s\n", frame.ResourceType.String())
-	fmt.Printf("Operation Type: %s\n", frame.OperationType.String())
-
-	if len(frame.RequestHeaders) > 0 {
-		fmt.Printf("=== Request Headers ===\n")
-		for header, value := range frame.RequestHeaders {
-			fmt.Printf("%s: %v\n", header.String(), value)
-		}
-	}
-
-	if len(frame.ResponseHeaders) > 0 {
-		fmt.Printf("=== Response Headers ===\n")
-		for header, value := range frame.ResponseHeaders {
-			fmt.Printf("%s: %v\n", header.String(), value)
-		}
-	}
-
-	if len(frame.ContextHeaders) > 0 {
-		fmt.Printf("=== Context Headers ===\n")
-		for header, value := range frame.ContextHeaders {
-			fmt.Printf("%s: %v\n", header.String(), value)
-		}
-	}
-
-	if len(frame.Payload) > 0 {
-		var jsonObj any
-		err := json.Unmarshal(frame.Payload, &jsonObj)
-		if err != nil {
-			fmt.Printf("Payload: %s\n", hex.EncodeToString(frame.Payload))
-		} else {
-			fmt.Printf("Payload: %+v\n", jsonObj)
-		}
-	}
-}
@@ -11,7 +11,6 @@ import (
 	badgerdatastore "github.com/pikami/cosmium/internal/datastore/badger_datastore"
 	jsondatastore "github.com/pikami/cosmium/internal/datastore/json_datastore"
 	"github.com/pikami/cosmium/internal/logger"
-	"github.com/pikami/cosmium/internal/rntbd"
 )

 func main() {
@@ -21,7 +20,6 @@ func main() {
 	switch configuration.DataStore {
 	case config.DataStoreBadger:
 		dataStore = badgerdatastore.NewBadgerDataStore(badgerdatastore.BadgerDataStoreOptions{
-			InitialDataFilePath: configuration.InitialDataFilePath,
 			PersistDataFilePath: configuration.PersistDataFilePath,
 		})
 		logger.InfoLn("Using Badger data store")
@@ -39,15 +37,6 @@ func main() {
 		panic(err)
 	}

-	if configuration.EnableRntbd {
-		rntbdServer := rntbd.NewRntbdServer(configuration.RntbdPort, server)
-		err = rntbdServer.Start()
-		if err != nil {
-			panic(err)
-		}
-		defer rntbdServer.Stop()
-	}
-
 	waitForExit(server, dataStore)
 }

@@ -60,5 +49,6 @@ func waitForExit(server *api.ApiServer, dataStore datastore.DataStore) {

 	// Stop the server
 	server.Stop()
+
 	dataStore.Close()
 }
@@ -1,64 +1,59 @@
 module github.com/pikami/cosmium

-go 1.26.3
+go 1.24.0

 require (
-	github.com/Azure/azure-sdk-for-go/sdk/azcore v1.22.0
-	github.com/Azure/azure-sdk-for-go/sdk/data/azcosmos v1.4.2
+	github.com/Azure/azure-sdk-for-go/sdk/azcore v1.18.2
+	github.com/Azure/azure-sdk-for-go/sdk/data/azcosmos v1.4.0
 	github.com/cosmiumdev/json-patch/v5 v5.9.11
-	github.com/dgraph-io/badger/v4 v4.9.2
-	github.com/gin-gonic/gin v1.12.0
+	github.com/dgraph-io/badger/v4 v4.8.0
+	github.com/gin-gonic/gin v1.10.1
 	github.com/google/uuid v1.6.0
-	github.com/stretchr/testify v1.11.1
+	github.com/stretchr/testify v1.10.0
 	github.com/vmihailenco/msgpack/v5 v5.4.1
-	golang.org/x/exp v0.0.0-20260603202125-055de637280b
+	golang.org/x/exp v0.0.0-20250819193227-8b4c13bb791b
 )

 require (
 	github.com/Azure/azure-sdk-for-go v68.0.0+incompatible // indirect
-	github.com/Azure/azure-sdk-for-go/sdk/internal v1.12.0 // indirect
-	github.com/bytedance/gopkg v0.1.4 // indirect
-	github.com/bytedance/sonic v1.15.2 // indirect
-	github.com/bytedance/sonic/loader v0.5.1 // indirect
+	github.com/Azure/azure-sdk-for-go/sdk/internal v1.11.2 // indirect
+	github.com/bytedance/sonic v1.14.0 // indirect
+	github.com/bytedance/sonic/loader v0.3.0 // indirect
 	github.com/cespare/xxhash/v2 v2.3.0 // indirect
-	github.com/cloudwego/base64x v0.1.7 // indirect
+	github.com/cloudwego/base64x v0.1.6 // indirect
 	github.com/davecgh/go-spew v1.1.1 // indirect
-	github.com/dgraph-io/ristretto/v2 v2.4.0 // indirect
+	github.com/dgraph-io/ristretto/v2 v2.3.0 // indirect
 	github.com/dustin/go-humanize v1.0.1 // indirect
-	github.com/gabriel-vasile/mimetype v1.4.13 // indirect
-	github.com/gin-contrib/sse v1.1.1 // indirect
+	github.com/gabriel-vasile/mimetype v1.4.9 // indirect
+	github.com/gin-contrib/sse v1.1.0 // indirect
 	github.com/go-logr/logr v1.4.3 // indirect
 	github.com/go-logr/stdr v1.2.2 // indirect
 	github.com/go-playground/locales v0.14.1 // indirect
 	github.com/go-playground/universal-translator v0.18.1 // indirect
-	github.com/go-playground/validator/v10 v10.30.3 // indirect
-	github.com/goccy/go-json v0.10.6 // indirect
-	github.com/goccy/go-yaml v1.19.2 // indirect
-	github.com/google/flatbuffers v25.12.19+incompatible // indirect
+	github.com/go-playground/validator/v10 v10.27.0 // indirect
+	github.com/goccy/go-json v0.10.5 // indirect
+	github.com/google/flatbuffers v25.2.10+incompatible // indirect
 	github.com/json-iterator/go v1.1.12 // indirect
-	github.com/klauspost/compress v1.18.6 // indirect
+	github.com/klauspost/compress v1.18.0 // indirect
 	github.com/klauspost/cpuid/v2 v2.3.0 // indirect
 	github.com/leodido/go-urn v1.4.0 // indirect
-	github.com/mattn/go-isatty v0.0.22 // indirect
+	github.com/mattn/go-isatty v0.0.20 // indirect
 	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
 	github.com/modern-go/reflect2 v1.0.2 // indirect
-	github.com/pelletier/go-toml/v2 v2.3.1 // indirect
+	github.com/pelletier/go-toml/v2 v2.2.4 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
-	github.com/quic-go/qpack v0.6.0 // indirect
-	github.com/quic-go/quic-go v0.60.0 // indirect
 	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
-	github.com/ugorji/go/codec v1.3.1 // indirect
+	github.com/ugorji/go/codec v1.3.0 // indirect
 	github.com/vmihailenco/tagparser/v2 v2.0.0 // indirect
-	go.mongodb.org/mongo-driver/v2 v2.6.0 // indirect
-	go.opentelemetry.io/auto/sdk v1.2.1 // indirect
-	go.opentelemetry.io/otel v1.44.0 // indirect
-	go.opentelemetry.io/otel/metric v1.44.0 // indirect
-	go.opentelemetry.io/otel/trace v1.44.0 // indirect
-	golang.org/x/arch v0.28.0 // indirect
-	golang.org/x/crypto v0.53.0 // indirect
-	golang.org/x/net v0.56.0 // indirect
-	golang.org/x/sys v0.46.0 // indirect
-	golang.org/x/text v0.38.0 // indirect
-	google.golang.org/protobuf v1.36.11 // indirect
+	go.opentelemetry.io/auto/sdk v1.1.0 // indirect
+	go.opentelemetry.io/otel v1.37.0 // indirect
+	go.opentelemetry.io/otel/metric v1.37.0 // indirect
+	go.opentelemetry.io/otel/trace v1.37.0 // indirect
+	golang.org/x/arch v0.20.0 // indirect
+	golang.org/x/crypto v0.41.0 // indirect
+	golang.org/x/net v0.43.0 // indirect
+	golang.org/x/sys v0.35.0 // indirect
+	golang.org/x/text v0.28.0 // indirect
+	google.golang.org/protobuf v1.36.7 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )
@@ -1,44 +1,42 @@
 github.com/Azure/azure-sdk-for-go v68.0.0+incompatible h1:fcYLmCpyNYRnvJbPerq7U0hS+6+I79yEDJBqVNcqUzU=
 github.com/Azure/azure-sdk-for-go v68.0.0+incompatible/go.mod h1:9XXNKU+eRnpl9moKnB4QOLf1HestfXbmab5FXxiDBjc=
-github.com/Azure/azure-sdk-for-go/sdk/azcore v1.22.0 h1:aokoqcHvaGjiM3VpjKDfMMnF/8epJ+Q1HLJ7CudztqE=
-github.com/Azure/azure-sdk-for-go/sdk/azcore v1.22.0/go.mod h1:/WYEx9pcM9Y+Dd/APJaNlSvVSvzl54rrMdZT5+Oi2LM=
-github.com/Azure/azure-sdk-for-go/sdk/azidentity v1.13.1 h1:Hk5QBxZQC1jb2Fwj6mpzme37xbCDdNTxU7O9eb5+LB4=
-github.com/Azure/azure-sdk-for-go/sdk/azidentity v1.13.1/go.mod h1:IYus9qsFobWIc2YVwe/WPjcnyCkPKtnHAqUYeebc8z0=
-github.com/Azure/azure-sdk-for-go/sdk/data/azcosmos v1.4.2 h1:zqxnp53f5Jn5PFU5Av4mvyWEbZ7whg72AoOCEzlXFKc=
-github.com/Azure/azure-sdk-for-go/sdk/data/azcosmos v1.4.2/go.mod h1:Krtog/7tz27z75TwM5cIS8bxEH4dcBUezcq+kGVeZEo=
-github.com/Azure/azure-sdk-for-go/sdk/internal v1.12.0 h1:fhqpLE3UEXi9lPaBRpQ6XuRW0nU7hgg4zlmZZa+a9q4=
-github.com/Azure/azure-sdk-for-go/sdk/internal v1.12.0/go.mod h1:7dCRMLwisfRH3dBupKeNCioWYUZ4SS09Z14H+7i8ZoY=
-github.com/AzureAD/microsoft-authentication-library-for-go v1.6.0 h1:XRzhVemXdgvJqCH0sFfrBUTnUJSBrBf7++ypk+twtRs=
-github.com/AzureAD/microsoft-authentication-library-for-go v1.6.0/go.mod h1:HKpQxkWaGLJ+D/5H8QRpyQXA1eKjxkFlOMwck5+33Jk=
-github.com/bytedance/gopkg v0.1.4 h1:oZnQwnX82KAIWb7033bEwtxvTqXcYMxDBaQxo5JJHWM=
-github.com/bytedance/gopkg v0.1.4/go.mod h1:v1zWfPm21Fb+OsyXN2VAHdL6TBb2L88anLQgdyje6R4=
-github.com/bytedance/sonic v1.15.2 h1:90H+rcF/FwLXwfB1cudOLq/je83n683Utf4Cbp0xHCo=
-github.com/bytedance/sonic v1.15.2/go.mod h1:mT2NbXunuaEbnZ+mRIX/vYqKISmgEuHFDI4UzmKx2SA=
-github.com/bytedance/sonic/loader v0.5.1 h1:Ygpfa9zwRCCKSlrp5bBP/b/Xzc3VxsAW+5NIYXrOOpI=
-github.com/bytedance/sonic/loader v0.5.1/go.mod h1:AR4NYCk5DdzZizZ5djGqQ92eEhCCcdf5x77udYiSJRo=
+github.com/Azure/azure-sdk-for-go/sdk/azcore v1.18.2 h1:Hr5FTipp7SL07o2FvoVOX9HRiRH3CR3Mj8pxqCcdD5A=
+github.com/Azure/azure-sdk-for-go/sdk/azcore v1.18.2/go.mod h1:QyVsSSN64v5TGltphKLQ2sQxe4OBQg0J1eKRcVBnfgE=
+github.com/Azure/azure-sdk-for-go/sdk/azidentity v1.10.1 h1:B+blDbyVIG3WaikNxPnhPiJ1MThR03b3vKGtER95TP4=
+github.com/Azure/azure-sdk-for-go/sdk/azidentity v1.10.1/go.mod h1:JdM5psgjfBf5fo2uWOZhflPWyDBZ/O/CNAH9CtsuZE4=
+github.com/Azure/azure-sdk-for-go/sdk/data/azcosmos v1.4.0 h1:TSaH6Lj0m8bDr4vX1+LC1KLQTnLzZb3tOxrx/PLqw+c=
+github.com/Azure/azure-sdk-for-go/sdk/data/azcosmos v1.4.0/go.mod h1:Krtog/7tz27z75TwM5cIS8bxEH4dcBUezcq+kGVeZEo=
+github.com/Azure/azure-sdk-for-go/sdk/internal v1.11.2 h1:9iefClla7iYpfYWdzPCRDozdmndjTm8DXdpCzPajMgA=
+github.com/Azure/azure-sdk-for-go/sdk/internal v1.11.2/go.mod h1:XtLgD3ZD34DAaVIIAyG3objl5DynM3CQ/vMcbBNJZGI=
+github.com/AzureAD/microsoft-authentication-library-for-go v1.4.2 h1:oygO0locgZJe7PpYPXT5A29ZkwJaPqcva7BVeemZOZs=
+github.com/AzureAD/microsoft-authentication-library-for-go v1.4.2/go.mod h1:wP83P5OoQ5p6ip3ScPr0BAq0BvuPAvacpEuSzyouqAI=
+github.com/bytedance/sonic v1.14.0 h1:/OfKt8HFw0kh2rj8N0F6C/qPGRESq0BbaNZgcNXXzQQ=
+github.com/bytedance/sonic v1.14.0/go.mod h1:WoEbx8WTcFJfzCe0hbmyTGrfjt8PzNEBdxlNUO24NhA=
+github.com/bytedance/sonic/loader v0.3.0 h1:dskwH8edlzNMctoruo8FPTJDF3vLtDT0sXZwvZJyqeA=
+github.com/bytedance/sonic/loader v0.3.0/go.mod h1:N8A3vUdtUebEY2/VQC0MyhYeKUFosQU6FxH2JmUe6VI=
 github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
 github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
-github.com/cloudwego/base64x v0.1.7 h1:NppS+Fgzg5ovhn4NkUXaDT3x9jldgH5ToMCqzBSi2zI=
-github.com/cloudwego/base64x v0.1.7/go.mod h1:Cu1PV9zfrSf7ET2tIbWbbEy7jO7HHJ13q4X2SQ8aWYg=
+github.com/cloudwego/base64x v0.1.6 h1:t11wG9AECkCDk5fMSoxmufanudBtJ+/HemLstXDLI2M=
+github.com/cloudwego/base64x v0.1.6/go.mod h1:OFcloc187FXDaYHvrNIjxSe8ncn0OOM8gEHfghB2IPU=
 github.com/cosmiumdev/json-patch/v5 v5.9.11 h1:WD2Wqaz/vO987z2FFdqgkj15HgYZ/Y5TpqE3I4T/iOQ=
 github.com/cosmiumdev/json-patch/v5 v5.9.11/go.mod h1:YPZmckmv4ZY+oxKIOjgq3sIudHVB6VEMcicCS9LtVLM=
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
-github.com/dgraph-io/badger/v4 v4.9.2 h1:Wb5qw8gElqwV1a8msHTeQKova9b1V10heFKMIiPd80E=
-github.com/dgraph-io/badger/v4 v4.9.2/go.mod h1:nJjaJTUOSsQEBhsq209FmwCvMJzEA3e74RjZw6V2pQI=
-github.com/dgraph-io/ristretto/v2 v2.4.0 h1:I/w09yLjhdcVD2QV192UJcq8dPBaAJb9pOuMyNy0XlU=
-github.com/dgraph-io/ristretto/v2 v2.4.0/go.mod h1:0KsrXtXvnv0EqnzyowllbVJB8yBonswa2lTCK2gGo9E=
+github.com/dgraph-io/badger/v4 v4.8.0 h1:JYph1ChBijCw8SLeybvPINizbDKWZ5n/GYbz2yhN/bs=
+github.com/dgraph-io/badger/v4 v4.8.0/go.mod h1:U6on6e8k/RTbUWxqKR0MvugJuVmkxSNc79ap4917h4w=
+github.com/dgraph-io/ristretto/v2 v2.3.0 h1:qTQ38m7oIyd4GAed/QkUZyPFNMnvVWyazGXRwvOt5zk=
+github.com/dgraph-io/ristretto/v2 v2.3.0/go.mod h1:gpoRV3VzrEY1a9dWAYV6T1U7YzfgttXdd/ZzL1s9OZM=
 github.com/dgryski/go-farm v0.0.0-20240924180020-3414d57e47da h1:aIftn67I1fkbMa512G+w+Pxci9hJPB8oMnkcP3iZF38=
 github.com/dgryski/go-farm v0.0.0-20240924180020-3414d57e47da/go.mod h1:SqUrOPUnsFjfmXRMNPybcSiG0BgUW2AuFH8PAnS2iTw=
 github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
 github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
-github.com/gabriel-vasile/mimetype v1.4.13 h1:46nXokslUBsAJE/wMsp5gtO500a4F3Nkz9Ufpk2AcUM=
-github.com/gabriel-vasile/mimetype v1.4.13/go.mod h1:d+9Oxyo1wTzWdyVUPMmXFvp4F9tea18J8ufA774AB3s=
-github.com/gin-contrib/sse v1.1.1 h1:uGYpNwTacv5R68bSGMapo62iLTRa9l5zxGCps4hK6ko=
-github.com/gin-contrib/sse v1.1.1/go.mod h1:QXzuVkA0YO7o/gun03UI1Q+FTI8ZV/n5t03kIQAI89s=
-github.com/gin-gonic/gin v1.12.0 h1:b3YAbrZtnf8N//yjKeU2+MQsh2mY5htkZidOM7O0wG8=
-github.com/gin-gonic/gin v1.12.0/go.mod h1:VxccKfsSllpKshkBWgVgRniFFAzFb9csfngsqANjnLc=
+github.com/gabriel-vasile/mimetype v1.4.9 h1:5k+WDwEsD9eTLL8Tz3L0VnmVh9QxGjRmjBvAG7U/oYY=
+github.com/gabriel-vasile/mimetype v1.4.9/go.mod h1:WnSQhFKJuBlRyLiKohA/2DtIlPFAbguNaG7QCHcyGok=
+github.com/gin-contrib/sse v1.1.0 h1:n0w2GMuUpWDVp7qSpvze6fAu9iRxJY4Hmj6AmBOU05w=
+github.com/gin-contrib/sse v1.1.0/go.mod h1:hxRZ5gVpWMT7Z0B0gSNYqqsSCNIJMjzvm6fqCz9vjwM=
+github.com/gin-gonic/gin v1.10.1 h1:T0ujvqyCSqRopADpgPgiTT63DUQVSfojyME59Ei63pQ=
+github.com/gin-gonic/gin v1.10.1/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y=
 github.com/go-logr/logr v1.2.2/go.mod h1:jdQByPbusPIv2/zmleS9BjJVeZ6kBagPoEUsqbVz/1A=
 github.com/go-logr/logr v1.4.3 h1:CjnDlHq8ikf6E492q6eKboGOC0T8CDaOvkHCIg8idEI=
 github.com/go-logr/logr v1.4.3/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY=
@@ -50,16 +48,14 @@ github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/o
 github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
 github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
 github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
-github.com/go-playground/validator/v10 v10.30.3 h1:4MU6YkEwx7GbcPJOZxrtbu+QfF3pJLJuaYTeAH0DYy8=
-github.com/go-playground/validator/v10 v10.30.3/go.mod h1:4Axh7oCNGcoGkqLoE4YWt6n20mcEIsPRlB7vPk3lpyc=
-github.com/goccy/go-json v0.10.6 h1:p8HrPJzOakx/mn/bQtjgNjdTcN+/S6FcG2CTtQOrHVU=
-github.com/goccy/go-json v0.10.6/go.mod h1:oq7eo15ShAhp70Anwd5lgX2pLfOS3QCiwU/PULtXL6M=
-github.com/goccy/go-yaml v1.19.2 h1:PmFC1S6h8ljIz6gMRBopkjP1TVT7xuwrButHID66PoM=
-github.com/goccy/go-yaml v1.19.2/go.mod h1:XBurs7gK8ATbW4ZPGKgcbrY1Br56PdM69F7LkFRi1kA=
-github.com/golang-jwt/jwt/v5 v5.3.0 h1:pv4AsKCKKZuqlgs5sUmn4x8UlGa0kEVt/puTpKx9vvo=
-github.com/golang-jwt/jwt/v5 v5.3.0/go.mod h1:fxCRLWMO43lRc8nhHWY6LGqRcf+1gQWArsqaEUEa5bE=
-github.com/google/flatbuffers v25.12.19+incompatible h1:haMV2JRRJCe1998HeW/p0X9UaMTK6SDo0ffLn2+DbLs=
-github.com/google/flatbuffers v25.12.19+incompatible/go.mod h1:1AeVuKshWv4vARoZatz6mlQ0JxURH0Kv5+zNeJKJCa8=
+github.com/go-playground/validator/v10 v10.27.0 h1:w8+XrWVMhGkxOaaowyKH35gFydVHOvC0/uWoy2Fzwn4=
+github.com/go-playground/validator/v10 v10.27.0/go.mod h1:I5QpIEbmr8On7W0TktmJAumgzX4CA1XNl4ZmDuVHKKo=
+github.com/goccy/go-json v0.10.5 h1:Fq85nIqj+gXn/S5ahsiTlK3TmC85qgirsdTP/+DeaC4=
+github.com/goccy/go-json v0.10.5/go.mod h1:oq7eo15ShAhp70Anwd5lgX2pLfOS3QCiwU/PULtXL6M=
+github.com/golang-jwt/jwt/v5 v5.2.2 h1:Rl4B7itRWVtYIHFrSNd7vhTiz9UpLdi6gZhZ3wEeDy8=
+github.com/golang-jwt/jwt/v5 v5.2.2/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk=
+github.com/google/flatbuffers v25.2.10+incompatible h1:F3vclr7C3HpB1k9mxCGRMXq6FdUalZ6H/pNX4FP1v0Q=
+github.com/google/flatbuffers v25.2.10+incompatible/go.mod h1:1AeVuKshWv4vARoZatz6mlQ0JxURH0Kv5+zNeJKJCa8=
 github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
 github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
 github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
@@ -67,8 +63,8 @@ github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
 github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
 github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
-github.com/klauspost/compress v1.18.6 h1:2jupLlAwFm95+YDR+NwD2MEfFO9d4z4Prjl1XXDjuao=
-github.com/klauspost/compress v1.18.6/go.mod h1:cwPg85FWrGar70rWktvGQj8/hthj3wpl0PGDogxkrSQ=
+github.com/klauspost/compress v1.18.0 h1:c/Cqfb0r+Yi+JtIEq73FWXVkRonBlf0CRNYc8Zttxdo=
+github.com/klauspost/compress v1.18.0/go.mod h1:2Pp+KzxcywXVXMr50+X0Q/Lsb43OQHYWRCY2AiWywWQ=
 github.com/klauspost/cpuid/v2 v2.3.0 h1:S4CRMLnYUhGeDFDqkGriYKdfoFlDnMtqTiI/sFzhA9Y=
 github.com/klauspost/cpuid/v2 v2.3.0/go.mod h1:hqwkgyIinND0mEev00jJYCxPNVRVXFQeu1XKlok6oO0=
 github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
@@ -79,72 +75,61 @@ github.com/kylelemons/godebug v1.1.0 h1:RPNrshWIDI6G2gRW9EHilWtl7Z6Sb1BR0xunSBf0
 github.com/kylelemons/godebug v1.1.0/go.mod h1:9/0rRGxNHcop5bhtWyNeEfOS8JIWk580+fNqagV/RAw=
 github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ=
 github.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI=
-github.com/mattn/go-isatty v0.0.22 h1:j8l17JJ9i6VGPUFUYoTUKPSgKe/83EYU2zBC7YNKMw4=
-github.com/mattn/go-isatty v0.0.22/go.mod h1:ZXfXG4SQHsB/w3ZeOYbR0PrPwLy+n6xiMrJlRFqopa4=
+github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
+github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
 github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
 github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
 github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
 github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
 github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
-github.com/pelletier/go-toml/v2 v2.3.1 h1:MYEvvGnQjeNkRF1qUuGolNtNExTDwct51yp7olPtrEc=
-github.com/pelletier/go-toml/v2 v2.3.1/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY=
+github.com/pelletier/go-toml/v2 v2.2.4 h1:mye9XuhQ6gvn5h28+VilKrrPoQVanw5PMw/TB0t5Ec4=
+github.com/pelletier/go-toml/v2 v2.2.4/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY=
 github.com/pkg/browser v0.0.0-20240102092130-5ac0b6a4141c h1:+mdjkGKdHQG3305AYmdv1U2eRNDiU2ErMBj1gwrq8eQ=
 github.com/pkg/browser v0.0.0-20240102092130-5ac0b6a4141c/go.mod h1:7rwL4CYBLnjLxUqIJNnCWiEdr3bn6IUYi15bNlnbCCU=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
-github.com/quic-go/go-ossfuzz-seeds v0.1.0 h1:APacT+iIaNF6fd8AGEiN3bT/Jtkd2jz4v4TzM7MFjy0=
-github.com/quic-go/go-ossfuzz-seeds v0.1.0/go.mod h1:3IOHRbJIc+L6YKMwfDtJAM9Vj9k0YY4muhuyUYk5tbk=
-github.com/quic-go/qpack v0.6.0 h1:g7W+BMYynC1LbYLSqRt8PBg5Tgwxn214ZZR34VIOjz8=
-github.com/quic-go/qpack v0.6.0/go.mod h1:lUpLKChi8njB4ty2bFLX2x4gzDqXwUpaO1DP9qMDZII=
-github.com/quic-go/quic-go v0.60.0 h1:xcQioE8OM66UQLeUMHltK1CCcOu3JbVB4JAQdDQSB+0=
-github.com/quic-go/quic-go v0.60.0/go.mod h1:wpKpjmPpftl30sL6pFh7REVpjbcCVy4zt2vDyK1TuJk=
-github.com/rogpeppe/go-internal v1.14.1 h1:UQB4HGPB6osV0SQTLymcB4TgvyWu6ZyliaW0tI/otEQ=
-github.com/rogpeppe/go-internal v1.14.1/go.mod h1:MaRKkUm5W0goXpeCfT7UZI6fk/L7L7so1lCWt35ZSgc=
+github.com/rogpeppe/go-internal v1.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR38lUII=
+github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
 github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
-github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
 github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
 github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
-github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
+github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
+github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
 github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
-github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U=
-github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
 github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
 github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
-github.com/ugorji/go/codec v1.3.1 h1:waO7eEiFDwidsBN6agj1vJQ4AG7lh2yqXyOXqhgQuyY=
-github.com/ugorji/go/codec v1.3.1/go.mod h1:pRBVtBSKl77K30Bv8R2P+cLSGaTtex6fsA2Wjqmfxj4=
+github.com/ugorji/go/codec v1.3.0 h1:Qd2W2sQawAfG8XSvzwhBeoGq71zXOC/Q1E9y/wUcsUA=
+github.com/ugorji/go/codec v1.3.0/go.mod h1:pRBVtBSKl77K30Bv8R2P+cLSGaTtex6fsA2Wjqmfxj4=
 github.com/vmihailenco/msgpack/v5 v5.4.1 h1:cQriyiUvjTwOHg8QZaPihLWeRAAVoCpE00IUPn0Bjt8=
 github.com/vmihailenco/msgpack/v5 v5.4.1/go.mod h1:GaZTsDaehaPpQVyxrf5mtQlH+pc21PIudVV/E3rRQok=
 github.com/vmihailenco/tagparser/v2 v2.0.0 h1:y09buUbR+b5aycVFQs/g70pqKVZNBmxwAhO7/IwNM9g=
 github.com/vmihailenco/tagparser/v2 v2.0.0/go.mod h1:Wri+At7QHww0WTrCBeu4J6bNtoV6mEfg5OIWRZA9qds=
-go.mongodb.org/mongo-driver/v2 v2.6.0 h1:b9sJOYrkmt4l8bY43ZenFBcPlhYIjaOfYHLtbB/5qi8=
-go.mongodb.org/mongo-driver/v2 v2.6.0/go.mod h1:yOI9kBsufol30iFsl1slpdq1I0eHPzybRWdyYUs8K/0=
-go.opentelemetry.io/auto/sdk v1.2.1 h1:jXsnJ4Lmnqd11kwkBV2LgLoFMZKizbCi5fNZ/ipaZ64=
-go.opentelemetry.io/auto/sdk v1.2.1/go.mod h1:KRTj+aOaElaLi+wW1kO/DZRXwkF4C5xPbEe3ZiIhN7Y=
-go.opentelemetry.io/otel v1.44.0 h1:JjwHmHpA4iZ3wBxluu2fbbE7j4kqlE8jXyAyPXH7HqU=
-go.opentelemetry.io/otel v1.44.0/go.mod h1:BMgjTHL9WPRlRjL2oZCBTL4whCGtXch2H4BhOPIAyYc=
-go.opentelemetry.io/otel/metric v1.44.0 h1:1w0gILTcHdr3YI+ixLyjemwrVnsMURbTZFrSYCdDdmc=
-go.opentelemetry.io/otel/metric v1.44.0/go.mod h1:8O7hanEPBNgEMmybD3s2VBKcgWOCsA6tzHBPODAiquo=
-go.opentelemetry.io/otel/trace v1.44.0 h1:jxF5CsGYCe74MCRx2X4g7WsY/VBKRqqpNvXlX/6gtIk=
-go.opentelemetry.io/otel/trace v1.44.0/go.mod h1:oLl1jrMQAVo6v3GAggN+1VH9VIz9iUSvW53sW1Q8PIE=
-go.uber.org/mock v0.6.0 h1:hyF9dfmbgIX5EfOdasqLsWD6xqpNZlXblLB/Dbnwv3Y=
-go.uber.org/mock v0.6.0/go.mod h1:KiVJ4BqZJaMj4svdfmHM0AUx4NJYO8ZNpPnZn1Z+BBU=
-golang.org/x/arch v0.28.0 h1:wVwVdqsTuUbJvhYVCspQYwZXHNYeLSoZnmHD+ggddpQ=
-golang.org/x/arch v0.28.0/go.mod h1:0X+GdSIP+kL5wPmpK7sdkEVTt2XoYP0cSjQSbZBwOi8=
-golang.org/x/crypto v0.53.0 h1:QZ4Muo8THX6CizN2vPPd5fBGHyogrdK9fG4wLPFUsto=
-golang.org/x/crypto v0.53.0/go.mod h1:DNLU434OwVakk9PzuwV8w62mAJpRJL3vsgcfp4Qnsio=
-golang.org/x/exp v0.0.0-20260603202125-055de637280b h1:v1uXiEBHo8QA0LiGCo7UgHMzHT4Kdfpl2zmtH5vaP1Q=
-golang.org/x/exp v0.0.0-20260603202125-055de637280b/go.mod h1:d2fgXJLVs4dYDHUk5lwMIfzRzSrWCfGZb0ZqeLa/Vcw=
-golang.org/x/net v0.56.0 h1:Rw8j/hFzGvJUZwNBXnAtf5sVDVt+65SK2C7IxCxZt5o=
-golang.org/x/net v0.56.0/go.mod h1:D3Ku6r+V6JROoZK144D2XfMHFcMq/0zSfLelVTCFKec=
-golang.org/x/sys v0.46.0 h1:noSf2Fq6F8DBgS+LysIkx7rIExoNHJsxOAtPp4rthXw=
-golang.org/x/sys v0.46.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw=
-golang.org/x/text v0.38.0 h1:sXmwo9DwP3OK9EZ7PqAdaooSGozfl/3a6/xJcbzPRhE=
-golang.org/x/text v0.38.0/go.mod h1:YXZt3QhHUKYT53r2lLKFIVi6Ao1jdzrTR/KQ09qyxF4=
-google.golang.org/protobuf v1.36.11 h1:fV6ZwhNocDyBLK0dj+fg8ektcVegBBuEolpbTQyBNVE=
-google.golang.org/protobuf v1.36.11/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
+go.opentelemetry.io/auto/sdk v1.1.0 h1:cH53jehLUN6UFLY71z+NDOiNJqDdPRaXzTel0sJySYA=
+go.opentelemetry.io/auto/sdk v1.1.0/go.mod h1:3wSPjt5PWp2RhlCcmmOial7AvC4DQqZb7a7wCow3W8A=
+go.opentelemetry.io/otel v1.37.0 h1:9zhNfelUvx0KBfu/gb+ZgeAfAgtWrfHJZcAqFC228wQ=
+go.opentelemetry.io/otel v1.37.0/go.mod h1:ehE/umFRLnuLa/vSccNq9oS1ErUlkkK71gMcN34UG8I=
+go.opentelemetry.io/otel/metric v1.37.0 h1:mvwbQS5m0tbmqML4NqK+e3aDiO02vsf/WgbsdpcPoZE=
+go.opentelemetry.io/otel/metric v1.37.0/go.mod h1:04wGrZurHYKOc+RKeye86GwKiTb9FKm1WHtO+4EVr2E=
+go.opentelemetry.io/otel/trace v1.37.0 h1:HLdcFNbRQBE2imdSEgm/kwqmQj1Or1l/7bW6mxVK7z4=
+go.opentelemetry.io/otel/trace v1.37.0/go.mod h1:TlgrlQ+PtQO5XFerSPUYG0JSgGyryXewPGyayAWSBS0=
+golang.org/x/arch v0.20.0 h1:dx1zTU0MAE98U+TQ8BLl7XsJbgze2WnNKF/8tGp/Q6c=
+golang.org/x/arch v0.20.0/go.mod h1:bdwinDaKcfZUGpH09BB7ZmOfhalA8lQdzl62l8gGWsk=
+golang.org/x/crypto v0.41.0 h1:WKYxWedPGCTVVl5+WHSSrOBT0O8lx32+zxmHxijgXp4=
+golang.org/x/crypto v0.41.0/go.mod h1:pO5AFd7FA68rFak7rOAGVuygIISepHftHnr8dr6+sUc=
+golang.org/x/exp v0.0.0-20250819193227-8b4c13bb791b h1:DXr+pvt3nC887026GRP39Ej11UATqWDmWuS99x26cD0=
+golang.org/x/exp v0.0.0-20250819193227-8b4c13bb791b/go.mod h1:4QTo5u+SEIbbKW1RacMZq1YEfOBqeXa19JeshGi+zc4=
+golang.org/x/net v0.43.0 h1:lat02VYK2j4aLzMzecihNvTlJNQUq316m2Mr9rnM6YE=
+golang.org/x/net v0.43.0/go.mod h1:vhO1fvI4dGsIjh73sWfUVjj3N7CA9WkKJNQm2svM6Jg=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.35.0 h1:vz1N37gP5bs89s7He8XuIYXpyY0+QlsKmzipCbUtyxI=
+golang.org/x/sys v0.35.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
+golang.org/x/text v0.28.0 h1:rhazDwis8INMIwQ4tpjLDzUhx6RlXqZNPEM0huQojng=
+golang.org/x/text v0.28.0/go.mod h1:U8nCwOR8jO/marOQ0QbDiOngZVEBB7MAiitBuMjXiNU=
+google.golang.org/protobuf v1.36.7 h1:IgrO7UwFQGJdRNXH/sQux4R1Dj1WAKcLElzeeRaXV2A=
+google.golang.org/protobuf v1.36.7/go.mod h1:jduwjTPXsFjZGTmRluh+L6NjiWu7pchiJ2/5YcXBHnY=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
 gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
@@ -35,7 +35,3 @@ var UnknownErrorResponse = gin.H{"message": "Unknown error"}
 var NotFoundResponse = gin.H{"message": "NotFound"}
 var ConflictResponse = gin.H{"message": "Conflict"}
 var BadRequestResponse = gin.H{"message": "BadRequest"}
-var PreconditionFailedResponse = gin.H{
-	"code":    "PreconditionFailed",
-	"message": "Operation cannot be performed because one of the specified precondition is not met.",
-}
@@ -1,145 +0,0 @@
-package continuationtoken
-
-import (
-	"encoding/json"
-	"fmt"
-	"strconv"
-	"strings"
-
-	"github.com/pikami/cosmium/internal/logger"
-)
-
-type ContinuationTokenExternal struct {
-	Token string `json:"token"`
-	Range struct {
-		Min string `json:"min"`
-		Max string `json:"max"`
-	} `json:"range"`
-}
-
-type ContinuationToken struct {
-	Token struct {
-		ResourceId   string // RID
-		PageIndex    int    // RT
-		TotalResults int    // TRC
-		ISV          int    // ISV
-		IEO          int    // IEO
-		QCF          int    // QCF
-		LR           int    // LR
-	}
-	Range struct {
-		Min string
-		Max string
-	}
-}
-
-func Generate(resourceid string, pageIndex int, totalResults int) ContinuationToken {
-	ct := ContinuationToken{}
-	ct.Token.ResourceId = resourceid
-	ct.Token.PageIndex = pageIndex
-	ct.Token.TotalResults = totalResults
-	ct.Token.ISV = 2
-	ct.Token.IEO = 65567
-	ct.Token.QCF = 8
-	ct.Token.LR = 1
-	ct.Range.Min = ""
-	ct.Range.Max = "FF"
-
-	return ct
-}
-
-func GenerateDefault(resourceid string) ContinuationToken {
-	return Generate(resourceid, 0, 0)
-}
-
-func (ct *ContinuationToken) ToString() string {
-	token := fmt.Sprintf(
-		"-RID:~%s#RT:%d#TRC:%d#ISV:%d#IEO:%d#QCF:%d#LR:%d",
-		ct.Token.ResourceId,
-		ct.Token.PageIndex,
-		ct.Token.TotalResults,
-		ct.Token.ISV,
-		ct.Token.IEO,
-		ct.Token.QCF,
-		ct.Token.LR,
-	)
-
-	ect := ContinuationTokenExternal{}
-	ect.Token = token
-	ect.Range.Min = ct.Range.Min
-	ect.Range.Max = ct.Range.Max
-
-	json, err := json.Marshal(ect)
-	if err != nil {
-		logger.Error(err, "failed to marshal continuation token")
-		return ""
-	}
-
-	return string(json)
-}
-
-func FromString(token string) ContinuationToken {
-	ect := ContinuationTokenExternal{}
-	err := json.Unmarshal([]byte(token), &ect)
-	if err != nil {
-		logger.Error(err, "failed to unmarshal continuation token")
-		return ContinuationToken{}
-	}
-
-	ct, err := parseContinuationToken(ect.Token, ect.Range.Min, ect.Range.Max)
-	if err != nil {
-		logger.Error(err, "failed to parse continuation token")
-		return ContinuationToken{}
-	}
-
-	return *ct
-}
-
-func parseContinuationToken(token string, minRange string, maxRange string) (*ContinuationToken, error) {
-	const prefix = "-RID:~"
-	if !strings.HasPrefix(token, prefix) {
-		return nil, fmt.Errorf("invalid token prefix")
-	}
-
-	parts := strings.Split(token[len(prefix):], "#")
-	if len(parts) != 7 {
-		return nil, fmt.Errorf("invalid token format: expected 7 fields, got %d", len(parts))
-	}
-
-	ct := &ContinuationToken{}
-
-	ct.Token.ResourceId = parts[0]
-
-	parseIntField := func(part, key string) (int, error) {
-		if !strings.HasPrefix(part, key+":") {
-			return 0, fmt.Errorf("expected %s field", key)
-		}
-		return strconv.Atoi(strings.TrimPrefix(part, key+":"))
-	}
-
-	var err error
-
-	if ct.Token.PageIndex, err = parseIntField(parts[1], "RT"); err != nil {
-		return nil, err
-	}
-	if ct.Token.TotalResults, err = parseIntField(parts[2], "TRC"); err != nil {
-		return nil, err
-	}
-	if ct.Token.ISV, err = parseIntField(parts[3], "ISV"); err != nil {
-		return nil, err
-	}
-	if ct.Token.IEO, err = parseIntField(parts[4], "IEO"); err != nil {
-		return nil, err
-	}
-	if ct.Token.QCF, err = parseIntField(parts[5], "QCF"); err != nil {
-		return nil, err
-	}
-	if ct.Token.LR, err = parseIntField(parts[6], "LR"); err != nil {
-		return nil, err
-	}
-
-	ct.Range.Min = minRange
-	ct.Range.Max = maxRange
-
-	return ct, nil
-}
@@ -1,35 +0,0 @@
-package continuationtoken
-
-import (
-	"testing"
-
-	"github.com/stretchr/testify/assert"
-)
-
-func Test_Generate(t *testing.T) {
-	token := Generate("test-resource-id", 1, 100)
-
-	assert.Equal(t, "test-resource-id", token.Token.ResourceId)
-	assert.Equal(t, 1, token.Token.PageIndex)
-	assert.Equal(t, 100, token.Token.TotalResults)
-}
-
-func Test_FromString(t *testing.T) {
-	token := FromString("{\"token\":\"-RID:~test-resource-id#RT:1#TRC:100#ISV:2#IEO:65567#QCF:8#LR:1\",\"range\":{\"min\":\"\",\"max\":\"FF\"}}")
-
-	assert.Equal(t, "test-resource-id", token.Token.ResourceId)
-	assert.Equal(t, 1, token.Token.PageIndex)
-	assert.Equal(t, 100, token.Token.TotalResults)
-}
-
-func Test_ToString(t *testing.T) {
-	token := Generate("test-resource-id", 1, 100)
-	assert.Equal(t, "{\"token\":\"-RID:~test-resource-id#RT:1#TRC:100#ISV:2#IEO:65567#QCF:8#LR:1\",\"range\":{\"min\":\"\",\"max\":\"FF\"}}", token.ToString())
-}
-
-func Test_GenerateDefault(t *testing.T) {
-	token := GenerateDefault("test-resource-id")
-	assert.Equal(t, "test-resource-id", token.Token.ResourceId)
-	assert.Equal(t, 0, token.Token.PageIndex)
-	assert.Equal(t, 0, token.Token.TotalResults)
-}
@@ -1,25 +1,18 @@
 package badgerdatastore

 import (
-	"encoding/json"
-	"log"
-	"os"
 	"time"

 	"github.com/dgraph-io/badger/v4"
-	"github.com/pikami/cosmium/internal/datastore"
 	"github.com/pikami/cosmium/internal/logger"
 )

 type BadgerDataStore struct {
-	db        *badger.DB
-	gcTicker  *time.Ticker
-	gcDone    chan struct{}
-	gcStopped chan struct{}
+	db       *badger.DB
+	gcTicker *time.Ticker
 }

 type BadgerDataStoreOptions struct {
-	InitialDataFilePath string
 	PersistDataFilePath string
 }

@@ -38,14 +31,10 @@ func NewBadgerDataStore(options BadgerDataStoreOptions) *BadgerDataStore {
 	gcTicker := time.NewTicker(5 * time.Minute)

 	ds := &BadgerDataStore{
-		db:        db,
-		gcTicker:  gcTicker,
-		gcDone:    make(chan struct{}),
-		gcStopped: make(chan struct{}),
+		db:       db,
+		gcTicker: gcTicker,
 	}

-	ds.initializeDataStore(options.InitialDataFilePath)
-
 	go ds.runGarbageCollector()

 	return ds
@@ -54,8 +43,7 @@ func NewBadgerDataStore(options BadgerDataStoreOptions) *BadgerDataStore {
 func (r *BadgerDataStore) Close() {
 	if r.gcTicker != nil {
 		r.gcTicker.Stop()
-		close(r.gcDone)
-		<-r.gcStopped
+		r.gcTicker = nil
 	}

 	r.db.Close()
@@ -68,69 +56,11 @@ func (r *BadgerDataStore) DumpToJson() (string, error) {
 }

 func (r *BadgerDataStore) runGarbageCollector() {
-	defer close(r.gcStopped)
-
-	for {
-		select {
-		case <-r.gcTicker.C:
-			for {
-				err := r.db.RunValueLogGC(0.7)
-				if err != nil {
-					break
-				}
-			}
-		case <-r.gcDone:
-			return
-		}
-	}
-}
-
-func (r *BadgerDataStore) initializeDataStore(initialDataFilePath string) {
-	if initialDataFilePath == "" {
-		return
-	}
-
-	stat, err := os.Stat(initialDataFilePath)
-	if err != nil {
-		panic(err)
-	}
-
-	if stat.IsDir() {
-		logger.ErrorLn("Argument '-Persist' must be a path to file, not a directory.")
-		os.Exit(1)
-	}
-
-	jsonData, err := os.ReadFile(initialDataFilePath)
-	if err != nil {
-		log.Fatalf("Error reading state JSON file: %v", err)
-		return
-	}
-
-	var state datastore.InitialDataModel
-	if err := json.Unmarshal([]byte(jsonData), &state); err != nil {
-		log.Fatalf("Error parsing state JSON file: %v", err)
-		return
-	}
-
-	for dbName, dbModel := range state.Databases {
-		r.CreateDatabase(dbModel)
-		for colName, colModel := range state.Collections[dbName] {
-			r.CreateCollection(dbName, colModel)
-			for _, docModel := range state.Documents[dbName][colName] {
-				r.CreateDocument(dbName, colName, docModel)
-			}
-
-			for _, triggerModel := range state.Triggers[dbName][colName] {
-				r.CreateTrigger(dbName, colName, triggerModel)
-			}
-
-			for _, spModel := range state.StoredProcedures[dbName][colName] {
-				r.CreateStoredProcedure(dbName, colName, spModel)
-			}
-
-			for _, udfModel := range state.UserDefinedFunctions[dbName][colName] {
-				r.CreateUserDefinedFunction(dbName, colName, udfModel)
-			}
+	for range r.gcTicker.C {
+	again:
+		err := r.db.RunValueLogGC(0.7)
+		if err == nil {
+			goto again
 		}
 	}
 }
@@ -22,8 +22,7 @@ func (r *BadgerDataStore) GetAllCollections(databaseId string) ([]datastore.Coll
 		return nil, datastore.StatusNotFound
 	}

-	prefix := generateKey(resourceid.ResourceTypeCollection, databaseId, "", "") + "/"
-	colls, status := listByPrefix[datastore.Collection](r.db, prefix)
+	colls, status := listByPrefix[datastore.Collection](r.db, generateKey(resourceid.ResourceTypeCollection, databaseId, "", ""))
 	if status == datastore.StatusOk {
 		return colls, datastore.StatusOk
 	}
@@ -50,10 +49,11 @@ func (r *BadgerDataStore) DeleteCollection(databaseId string, collectionId strin
 	defer txn.Discard()

 	prefixes := []string{
-		generateKey(resourceid.ResourceTypeDocument, databaseId, collectionId, "") + "/",
-		generateKey(resourceid.ResourceTypeTrigger, databaseId, collectionId, "") + "/",
-		generateKey(resourceid.ResourceTypeStoredProcedure, databaseId, collectionId, "") + "/",
-		generateKey(resourceid.ResourceTypeUserDefinedFunction, databaseId, collectionId, "") + "/",
+		generateKey(resourceid.ResourceTypeDocument, databaseId, collectionId, ""),
+		generateKey(resourceid.ResourceTypeTrigger, databaseId, collectionId, ""),
+		generateKey(resourceid.ResourceTypeStoredProcedure, databaseId, collectionId, ""),
+		generateKey(resourceid.ResourceTypeUserDefinedFunction, databaseId, collectionId, ""),
+		collectionKey,
 	}
 	for _, prefix := range prefixes {
 		if err := deleteKeysByPrefix(txn, prefix); err != nil {
@@ -61,8 +61,6 @@ func (r *BadgerDataStore) DeleteCollection(databaseId string, collectionId strin
 		}
 	}

-	deleteKey(txn, collectionKey)
-
 	err := txn.Commit()
 	if err != nil {
 		logger.ErrorLn("Error while committing transaction:", err)
@@ -38,11 +38,12 @@ func (r *BadgerDataStore) DeleteDatabase(id string) datastore.DataStoreStatus {
 	defer txn.Discard()

 	prefixes := []string{
-		generateKey(resourceid.ResourceTypeCollection, id, "", "") + "/",
-		generateKey(resourceid.ResourceTypeDocument, id, "", "") + "/",
-		generateKey(resourceid.ResourceTypeTrigger, id, "", "") + "/",
-		generateKey(resourceid.ResourceTypeStoredProcedure, id, "", "") + "/",
-		generateKey(resourceid.ResourceTypeUserDefinedFunction, id, "", "") + "/",
+		generateKey(resourceid.ResourceTypeCollection, id, "", ""),
+		generateKey(resourceid.ResourceTypeDocument, id, "", ""),
+		generateKey(resourceid.ResourceTypeTrigger, id, "", ""),
+		generateKey(resourceid.ResourceTypeStoredProcedure, id, "", ""),
+		generateKey(resourceid.ResourceTypeUserDefinedFunction, id, "", ""),
+		databaseKey,
 	}
 	for _, prefix := range prefixes {
 		if err := deleteKeysByPrefix(txn, prefix); err != nil {
@@ -50,8 +51,6 @@ func (r *BadgerDataStore) DeleteDatabase(id string) datastore.DataStoreStatus {
 		}
 	}

-	deleteKey(txn, databaseKey)
-
 	err := txn.Commit()
 	if err != nil {
 		logger.ErrorLn("Error while committing transaction:", err)
@@ -202,22 +202,3 @@ func deleteKeysByPrefix(txn *badger.Txn, prefix string) error {

 	return nil
 }
-
-func deleteKey(txn *badger.Txn, key string) error {
-	_, err := txn.Get([]byte(key))
-	if err == badger.ErrKeyNotFound {
-		return nil
-	}
-	if err != nil {
-		logger.ErrorLn("Error while checking if key exists:", err)
-		return err
-	}
-
-	err = txn.Delete([]byte(key))
-	if err != nil {
-		logger.ErrorLn("Error while deleting key:", err)
-		return err
-	}
-
-	return nil
-}
@@ -24,8 +24,7 @@ func (r *BadgerDataStore) GetAllDocuments(databaseId string, collectionId string
 		return nil, datastore.StatusNotFound
 	}

-	prefix := generateKey(resourceid.ResourceTypeDocument, databaseId, collectionId, "") + "/"
-	docs, status := listByPrefix[datastore.Document](r.db, prefix)
+	docs, status := listByPrefix[datastore.Document](r.db, generateKey(resourceid.ResourceTypeDocument, databaseId, collectionId, ""))
 	if status == datastore.StatusOk {
 		return docs, datastore.StatusOk
 	}
@@ -46,8 +45,7 @@ func (r *BadgerDataStore) GetDocumentIterator(databaseId string, collectionId st
 		return nil, datastore.StatusNotFound
 	}

-	prefix := generateKey(resourceid.ResourceTypeDocument, databaseId, collectionId, "") + "/"
-	iter := NewBadgerDocumentIterator(txn, prefix)
+	iter := NewBadgerDocumentIterator(txn, generateKey(resourceid.ResourceTypeDocument, databaseId, collectionId, ""))
 	return iter, datastore.StatusOk
 }

@@ -24,8 +24,7 @@ func (r *BadgerDataStore) GetAllStoredProcedures(databaseId string, collectionId
 		return nil, datastore.StatusNotFound
 	}

-	prefix := generateKey(resourceid.ResourceTypeStoredProcedure, databaseId, collectionId, "") + "/"
-	storedProcedures, status := listByPrefix[datastore.StoredProcedure](r.db, prefix)
+	storedProcedures, status := listByPrefix[datastore.StoredProcedure](r.db, generateKey(resourceid.ResourceTypeStoredProcedure, databaseId, collectionId, ""))
 	if status == datastore.StatusOk {
 		return storedProcedures, datastore.StatusOk
 	}
@@ -24,8 +24,7 @@ func (r *BadgerDataStore) GetAllTriggers(databaseId string, collectionId string)
 		return nil, datastore.StatusNotFound
 	}

-	prefix := generateKey(resourceid.ResourceTypeTrigger, databaseId, collectionId, "") + "/"
-	triggers, status := listByPrefix[datastore.Trigger](r.db, prefix)
+	triggers, status := listByPrefix[datastore.Trigger](r.db, generateKey(resourceid.ResourceTypeTrigger, databaseId, collectionId, ""))
 	if status == datastore.StatusOk {
 		return triggers, datastore.StatusOk
 	}
@@ -24,8 +24,7 @@ func (r *BadgerDataStore) GetAllUserDefinedFunctions(databaseId string, collecti
 		return nil, datastore.StatusNotFound
 	}

-	prefix := generateKey(resourceid.ResourceTypeUserDefinedFunction, databaseId, collectionId, "") + "/"
-	udfs, status := listByPrefix[datastore.UserDefinedFunction](r.db, prefix)
+	udfs, status := listByPrefix[datastore.UserDefinedFunction](r.db, generateKey(resourceid.ResourceTypeUserDefinedFunction, databaseId, collectionId, ""))
 	if status == datastore.StatusOk {
 		return udfs, datastore.StatusOk
 	}
@@ -1,21 +0,0 @@
-package datastore
-
-type InitialDataModel struct {
-	// Map databaseId -> Database
-	Databases map[string]Database `json:"databases"`
-
-	// Map databaseId -> collectionId -> Collection
-	Collections map[string]map[string]Collection `json:"collections"`
-
-	// Map databaseId -> collectionId -> documentId -> Documents
-	Documents map[string]map[string]map[string]Document `json:"documents"`
-
-	// Map databaseId -> collectionId -> triggerId -> Trigger
-	Triggers map[string]map[string]map[string]Trigger `json:"triggers"`
-
-	// Map databaseId -> collectionId -> spId -> StoredProcedure
-	StoredProcedures map[string]map[string]map[string]StoredProcedure `json:"sprocs"`
-
-	// Map databaseId -> collectionId -> udfId -> UserDefinedFunction
-	UserDefinedFunctions map[string]map[string]map[string]UserDefinedFunction `json:"udfs"`
-}
@@ -11,12 +11,12 @@ type Database struct {
 type DataStoreStatus int

 const (
-	StatusOk       DataStoreStatus = 1
-	StatusNotFound DataStoreStatus = 2
-	Conflict       DataStoreStatus = 3
-	BadRequest     DataStoreStatus = 4
-	IterEOF        DataStoreStatus = 5
-	Unknown        DataStoreStatus = 6
+	StatusOk       = 1
+	StatusNotFound = 2
+	Conflict       = 3
+	BadRequest     = 4
+	IterEOF        = 5
+	Unknown        = 6
 )

 type TriggerOperation string
@@ -1,138 +0,0 @@
-package rntbd
-
-import (
-	"bytes"
-	"encoding/binary"
-)
-
-type RntbdResponseFrame struct {
-	StatusCode      uint16
-	ResourceType    RntbdResourceType
-	ActivityId      []byte
-	ResponseHeaders []RntbdResponseHeader
-	Payload         []byte
-}
-
-type RntbdResponseHeader struct {
-	HeaderId   uint16
-	TokenType  RntbdTokenType
-	TokenValue any
-}
-
-type RntbdResponseFrameBuilder struct {
-	frame RntbdResponseFrame
-}
-
-func (b *RntbdResponseFrameBuilder) AddHeader(headerId uint16, tokenType RntbdTokenType, tokenValue any) {
-	b.frame.ResponseHeaders = append(b.frame.ResponseHeaders, RntbdResponseHeader{
-		HeaderId:   headerId,
-		TokenType:  tokenType,
-		TokenValue: tokenValue,
-	})
-}
-
-func (b *RntbdResponseFrameBuilder) AddPayload(payload []byte) {
-	b.frame.Payload = payload
-}
-
-func (b *RntbdResponseFrameBuilder) SetStatusCode(statusCode uint16) {
-	b.frame.StatusCode = statusCode
-}
-
-func (b *RntbdResponseFrameBuilder) SetResourceType(resourceType RntbdResourceType) {
-	b.frame.ResourceType = resourceType
-}
-
-func (b *RntbdResponseFrameBuilder) SetActivityId(activityId []byte) {
-	b.frame.ActivityId = activityId
-}
-
-func (b *RntbdResponseFrameBuilder) Build() *RntbdResponseFrame {
-	return &b.frame
-}
-
-func (f *RntbdResponseFrame) ToBytes() []byte {
-	var buffer bytes.Buffer
-
-	binary.Write(&buffer, binary.LittleEndian, f.StatusCode)
-	binary.Write(&buffer, binary.LittleEndian, uint16(f.ResourceType))
-	binary.Write(&buffer, binary.LittleEndian, f.ActivityId)
-
-	for _, header := range f.ResponseHeaders {
-		binary.Write(&buffer, binary.LittleEndian, header.HeaderId)
-		binary.Write(&buffer, binary.LittleEndian, uint8(header.TokenType))
-
-		switch header.TokenType {
-		case RntbdTokenTypeByte:
-			buffer.Write(header.TokenValue.([]byte))
-		case RntbdTokenTypeUShort:
-			binary.Write(&buffer, binary.LittleEndian, header.TokenValue.(uint16))
-		case RntbdTokenTypeULong:
-			binary.Write(&buffer, binary.LittleEndian, header.TokenValue.(uint32))
-		case RntbdTokenTypeLong:
-			binary.Write(&buffer, binary.LittleEndian, header.TokenValue.(int32))
-		case RntbdTokenTypeULongLong:
-			binary.Write(&buffer, binary.LittleEndian, header.TokenValue.(uint64))
-		case RntbdTokenTypeLongLong:
-			binary.Write(&buffer, binary.LittleEndian, header.TokenValue.(int64))
-		case RntbdTokenTypeGuid:
-			buffer.Write(header.TokenValue.([]byte))
-
-		case RntbdTokenTypeSmallString:
-			binary.Write(&buffer, binary.LittleEndian, uint8(len(header.TokenValue.(string))))
-			buffer.WriteString(header.TokenValue.(string))
-		case RntbdTokenTypeString:
-			binary.Write(&buffer, binary.LittleEndian, uint16(len(header.TokenValue.(string))))
-			buffer.WriteString(header.TokenValue.(string))
-		case RntbdTokenTypeULongString:
-			binary.Write(&buffer, binary.LittleEndian, uint32(len(header.TokenValue.(string))))
-			buffer.WriteString(header.TokenValue.(string))
-		case RntbdTokenTypeSmallBytes:
-			binary.Write(&buffer, binary.LittleEndian, uint8(len(header.TokenValue.([]byte))))
-			buffer.Write(header.TokenValue.([]byte))
-		case RntbdTokenTypeBytes:
-			binary.Write(&buffer, binary.LittleEndian, uint16(len(header.TokenValue.([]byte))))
-			buffer.Write(header.TokenValue.([]byte))
-		case RntbdTokenTypeULongBytes:
-			binary.Write(&buffer, binary.LittleEndian, uint32(len(header.TokenValue.([]byte))))
-			buffer.Write(header.TokenValue.([]byte))
-		case RntbdTokenTypeFloat:
-			binary.Write(&buffer, binary.LittleEndian, header.TokenValue.(float32))
-		case RntbdTokenTypeDouble:
-			binary.Write(&buffer, binary.LittleEndian, header.TokenValue.(float64))
-		case RntbdTokenTypeInvalid:
-			panic("invalid token type")
-		default:
-			panic("invalid token type")
-		}
-	}
-
-	payloadSize := uint32(0)
-	if len(f.Payload) > 0 {
-		payloadSize = uint32(len(f.Payload)) + 4
-	}
-
-	frameSize := uint32(buffer.Len()) + 4
-	result := make([]byte, frameSize+payloadSize)
-	binary.LittleEndian.PutUint32(result, frameSize)
-	copy(result[4:], buffer.Bytes())
-
-	if len(f.Payload) > 0 {
-		binary.LittleEndian.PutUint32(result[frameSize:], payloadSize-4)
-		copy(result[frameSize+4:], f.Payload)
-	}
-
-	return result
-}
-
-func buildContextFrame(requestFrame *RntbdFrame) []byte {
-	builder := RntbdResponseFrameBuilder{}
-	builder.SetStatusCode(200)
-	builder.SetResourceType(RntbdResourceTypeConnection)
-	builder.SetActivityId(requestFrame.ActivityId)
-	builder.AddHeader(uint16(RntbdContextHeaderServerAgent), RntbdTokenTypeSmallString, "DocumentDB Server")
-	builder.AddHeader(uint16(RntbdContextHeaderServerVersion), RntbdTokenTypeSmallString, " version=2.14.0.0")
-	builder.AddHeader(uint16(RntbdContextHeaderIdleTimeoutInSeconds), RntbdTokenTypeULong, uint32(120))
-	builder.AddHeader(uint16(RntbdContextHeaderUnauthenticatedTimeoutInSeconds), RntbdTokenTypeULong, uint32(25))
-	return builder.Build().ToBytes()
-}
@@ -1,746 +0,0 @@
-package rntbd
-
-import (
-	"fmt"
-
-	"github.com/pikami/cosmium/api/headers"
-)
-
-type RntbdOperationType uint16
-
-const (
-	RntbdOperationTypeConnection                       RntbdOperationType = 0x0000
-	RntbdOperationTypeCreate                           RntbdOperationType = 0x0001
-	RntbdOperationTypeUpdate                           RntbdOperationType = 0x0002
-	RntbdOperationTypeRead                             RntbdOperationType = 0x0003
-	RntbdOperationTypeReadFeed                         RntbdOperationType = 0x0004
-	RntbdOperationTypeDelete                           RntbdOperationType = 0x0005
-	RntbdOperationTypeReplace                          RntbdOperationType = 0x0006
-	RntbdOperationTypeExecuteJavaScript                RntbdOperationType = 0x0008
-	RntbdOperationTypeSQLQuery                         RntbdOperationType = 0x0009
-	RntbdOperationTypePause                            RntbdOperationType = 0x000A
-	RntbdOperationTypeResume                           RntbdOperationType = 0x000B
-	RntbdOperationTypeStop                             RntbdOperationType = 0x000C
-	RntbdOperationTypeRecycle                          RntbdOperationType = 0x000D
-	RntbdOperationTypeCrash                            RntbdOperationType = 0x000E
-	RntbdOperationTypeQuery                            RntbdOperationType = 0x000F
-	RntbdOperationTypeForceConfigRefresh               RntbdOperationType = 0x0010
-	RntbdOperationTypeHead                             RntbdOperationType = 0x0011
-	RntbdOperationTypeHeadFeed                         RntbdOperationType = 0x0012
-	RntbdOperationTypeUpsert                           RntbdOperationType = 0x0013
-	RntbdOperationTypeRecreate                         RntbdOperationType = 0x0014
-	RntbdOperationTypeThrottle                         RntbdOperationType = 0x0015
-	RntbdOperationTypeGetSplitPoint                    RntbdOperationType = 0x0016
-	RntbdOperationTypePreCreateValidation              RntbdOperationType = 0x0017
-	RntbdOperationTypeBatchApply                       RntbdOperationType = 0x0018
-	RntbdOperationTypeAbortSplit                       RntbdOperationType = 0x0019
-	RntbdOperationTypeCompleteSplit                    RntbdOperationType = 0x001A
-	RntbdOperationTypeOfferUpdateOperation             RntbdOperationType = 0x001B
-	RntbdOperationTypeOfferPreGrowValidation           RntbdOperationType = 0x001C
-	RntbdOperationTypeBatchReportThroughputUtilization RntbdOperationType = 0x001D
-	RntbdOperationTypeCompletePartitionMigration       RntbdOperationType = 0x001E
-	RntbdOperationTypeAbortPartitionMigration          RntbdOperationType = 0x001F
-	RntbdOperationTypePreReplaceValidation             RntbdOperationType = 0x0020
-	RntbdOperationTypeAddComputeGatewayRequestCharges  RntbdOperationType = 0x0021
-	RntbdOperationTypeMigratePartition                 RntbdOperationType = 0x0022
-)
-
-type RntbdResourceType uint16
-
-const (
-	RntbdResourceTypeConnection              RntbdResourceType = 0x0000
-	RntbdResourceTypeDatabase                RntbdResourceType = 0x0001
-	RntbdResourceTypeCollection              RntbdResourceType = 0x0002
-	RntbdResourceTypeDocument                RntbdResourceType = 0x0003
-	RntbdResourceTypeAttachment              RntbdResourceType = 0x0004
-	RntbdResourceTypeUser                    RntbdResourceType = 0x0005
-	RntbdResourceTypePermission              RntbdResourceType = 0x0006
-	RntbdResourceTypeStoredProcedure         RntbdResourceType = 0x0007
-	RntbdResourceTypeConflict                RntbdResourceType = 0x0008
-	RntbdResourceTypeTrigger                 RntbdResourceType = 0x0009
-	RntbdResourceTypeUserDefinedFunction     RntbdResourceType = 0x000A
-	RntbdResourceTypeModule                  RntbdResourceType = 0x000B
-	RntbdResourceTypeReplica                 RntbdResourceType = 0x000C
-	RntbdResourceTypeModuleCommand           RntbdResourceType = 0x000D
-	RntbdResourceTypeRecord                  RntbdResourceType = 0x000E
-	RntbdResourceTypeOffer                   RntbdResourceType = 0x000F
-	RntbdResourceTypePartitionSetInformation RntbdResourceType = 0x0010
-	RntbdResourceTypeXPReplicatorAddress     RntbdResourceType = 0x0011
-	RntbdResourceTypeMasterPartition         RntbdResourceType = 0x0012
-	RntbdResourceTypeServerPartition         RntbdResourceType = 0x0013
-	RntbdResourceTypeDatabaseAccount         RntbdResourceType = 0x0014
-	RntbdResourceTypeTopology                RntbdResourceType = 0x0015
-	RntbdResourceTypePartitionKeyRange       RntbdResourceType = 0x0016
-	RntbdResourceTypeSchema                  RntbdResourceType = 0x0018
-	RntbdResourceTypeBatchApply              RntbdResourceType = 0x0019
-	RntbdResourceTypeRestoreMetadata         RntbdResourceType = 0x001A
-	RntbdResourceTypeComputeGatewayCharges   RntbdResourceType = 0x001B
-	RntbdResourceTypeRidRange                RntbdResourceType = 0x001C
-	RntbdResourceTypeUserDefinedType         RntbdResourceType = 0x001D
-)
-
-type RntbdRequestHeader uint16
-
-const (
-	RntbdRequestHeaderResourceId                                RntbdRequestHeader = 0x0000 // RntbdTokenType.Bytes, required = false
-	RntbdRequestHeaderAuthorizationToken                        RntbdRequestHeader = 0x0001 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderPayloadPresent                            RntbdRequestHeader = 0x0002 // RntbdTokenType.Byte, required = true
-	RntbdRequestHeaderDate                                      RntbdRequestHeader = 0x0003 // RntbdTokenType.SmallString, required = false
-	RntbdRequestHeaderPageSize                                  RntbdRequestHeader = 0x0004 // RntbdTokenType.ULong, required = false
-	RntbdRequestHeaderSessionToken                              RntbdRequestHeader = 0x0005 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderContinuationToken                         RntbdRequestHeader = 0x0006 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderIndexingDirective                         RntbdRequestHeader = 0x0007 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderMatch                                     RntbdRequestHeader = 0x0008 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderPreTriggerInclude                         RntbdRequestHeader = 0x0009 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderPostTriggerInclude                        RntbdRequestHeader = 0x000A // RntbdTokenType.String, required = false
-	RntbdRequestHeaderIsFanout                                  RntbdRequestHeader = 0x000B // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderCollectionPartitionIndex                  RntbdRequestHeader = 0x000C // RntbdTokenType.ULong, required = false
-	RntbdRequestHeaderCollectionServiceIndex                    RntbdRequestHeader = 0x000D // RntbdTokenType.ULong, required = false
-	RntbdRequestHeaderPreTriggerExclude                         RntbdRequestHeader = 0x000E // RntbdTokenType.String, required = false
-	RntbdRequestHeaderPostTriggerExclude                        RntbdRequestHeader = 0x000F // RntbdTokenType.String, required = false
-	RntbdRequestHeaderConsistencyLevel                          RntbdRequestHeader = 0x0010 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderEntityId                                  RntbdRequestHeader = 0x0011 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderResourceSchemaName                        RntbdRequestHeader = 0x0012 // RntbdTokenType.SmallString, required = false
-	RntbdRequestHeaderReplicaPath                               RntbdRequestHeader = 0x0013 // RntbdTokenType.String, required = true
-	RntbdRequestHeaderResourceTokenExpiry                       RntbdRequestHeader = 0x0014 // RntbdTokenType.ULong, required = false
-	RntbdRequestHeaderDatabaseName                              RntbdRequestHeader = 0x0015 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderCollectionName                            RntbdRequestHeader = 0x0016 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderDocumentName                              RntbdRequestHeader = 0x0017 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderAttachmentName                            RntbdRequestHeader = 0x0018 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderUserName                                  RntbdRequestHeader = 0x0019 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderPermissionName                            RntbdRequestHeader = 0x001A // RntbdTokenType.String, required = false
-	RntbdRequestHeaderStoredProcedureName                       RntbdRequestHeader = 0x001B // RntbdTokenType.String, required = false
-	RntbdRequestHeaderUserDefinedFunctionName                   RntbdRequestHeader = 0x001C // RntbdTokenType.String, required = false
-	RntbdRequestHeaderTriggerName                               RntbdRequestHeader = 0x001D // RntbdTokenType.String, required = false
-	RntbdRequestHeaderEnableScanInQuery                         RntbdRequestHeader = 0x001E // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderEmitVerboseTracesInQuery                  RntbdRequestHeader = 0x001F // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderConflictName                              RntbdRequestHeader = 0x0020 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderBindReplicaDirective                      RntbdRequestHeader = 0x0021 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderPrimaryMasterKey                          RntbdRequestHeader = 0x0022 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderSecondaryMasterKey                        RntbdRequestHeader = 0x0023 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderPrimaryReadonlyKey                        RntbdRequestHeader = 0x0024 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderSecondaryReadonlyKey                      RntbdRequestHeader = 0x0025 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderProfileRequest                            RntbdRequestHeader = 0x0026 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderEnableLowPrecisionOrderBy                 RntbdRequestHeader = 0x0027 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderClientVersion                             RntbdRequestHeader = 0x0028 // RntbdTokenType.SmallString, required = false
-	RntbdRequestHeaderCanCharge                                 RntbdRequestHeader = 0x0029 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderCanThrottle                               RntbdRequestHeader = 0x002A // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderPartitionKey                              RntbdRequestHeader = 0x002B // RntbdTokenType.String, required = false
-	RntbdRequestHeaderPartitionKeyRangeId                       RntbdRequestHeader = 0x002C // RntbdTokenType.String, required = false
-	RntbdRequestHeaderNotUsed2D                                 RntbdRequestHeader = 0x002D // RntbdTokenType.Invalid, required = false
-	RntbdRequestHeaderNotUsed2E                                 RntbdRequestHeader = 0x002E // RntbdTokenType.Invalid, required = false
-	RntbdRequestHeaderNotUsed2F                                 RntbdRequestHeader = 0x002F // RntbdTokenType.Invalid, required = false
-	RntbdRequestHeaderMigrateCollectionDirective                RntbdRequestHeader = 0x0031 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderNotUsed32                                 RntbdRequestHeader = 0x0032 // RntbdTokenType.Invalid, required = false
-	RntbdRequestHeaderSupportSpatialLegacyCoordinates           RntbdRequestHeader = 0x0033 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderPartitionCount                            RntbdRequestHeader = 0x0034 // RntbdTokenType.ULong, required = false
-	RntbdRequestHeaderCollectionRid                             RntbdRequestHeader = 0x0035 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderPartitionKeyRangeName                     RntbdRequestHeader = 0x0036 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderSchemaName                                RntbdRequestHeader = 0x003A // RntbdTokenType.String, required = false
-	RntbdRequestHeaderFilterBySchemaRid                         RntbdRequestHeader = 0x003B // RntbdTokenType.String, required = false
-	RntbdRequestHeaderUsePolygonsSmallerThanAHemisphere         RntbdRequestHeader = 0x003C // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderGatewaySignature                          RntbdRequestHeader = 0x003D // RntbdTokenType.String, required = false
-	RntbdRequestHeaderEnableLogging                             RntbdRequestHeader = 0x003E // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderAIM                                       RntbdRequestHeader = 0x003F // RntbdTokenType.String, required = false
-	RntbdRequestHeaderPopulateQuotaInfo                         RntbdRequestHeader = 0x0040 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderDisableRUPerMinuteUsage                   RntbdRequestHeader = 0x0041 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderPopulateQueryMetrics                      RntbdRequestHeader = 0x0042 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderResponseContinuationTokenLimitInKb        RntbdRequestHeader = 0x0043 // RntbdTokenType.ULong, required = false
-	RntbdRequestHeaderPopulatePartitionStatistics               RntbdRequestHeader = 0x0044 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderRemoteStorageType                         RntbdRequestHeader = 0x0045 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderCollectionRemoteStorageSecurityIdentifier RntbdRequestHeader = 0x0046 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderIfModifiedSince                           RntbdRequestHeader = 0x0047 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderPopulateCollectionThroughputInfo          RntbdRequestHeader = 0x0048 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderRemainingTimeInMsOnClientRequest          RntbdRequestHeader = 0x0049 // RntbdTokenType.ULong, required = false
-	RntbdRequestHeaderClientRetryAttemptCount                   RntbdRequestHeader = 0x004A // RntbdTokenType.ULong, required = false
-	RntbdRequestHeaderTargetLsn                                 RntbdRequestHeader = 0x004B // RntbdTokenType.LongLong, required = false
-	RntbdRequestHeaderTargetGlobalCommittedLsn                  RntbdRequestHeader = 0x004C // RntbdTokenType.LongLong, required = false
-	RntbdRequestHeaderTransportRequestID                        RntbdRequestHeader = 0x004D // RntbdTokenType.ULong, required = false
-	RntbdRequestHeaderRestoreMetadaFilter                       RntbdRequestHeader = 0x004E // RntbdTokenType.String, required = false
-	RntbdRequestHeaderRestoreParams                             RntbdRequestHeader = 0x004F // RntbdTokenType.String, required = false
-	RntbdRequestHeaderShareThroughput                           RntbdRequestHeader = 0x0050 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderPartitionResourceFilter                   RntbdRequestHeader = 0x0051 // RntbdTokenType.String, required = false
-	RntbdRequestHeaderIsReadOnlyScript                          RntbdRequestHeader = 0x0052 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderIsAutoScaleRequest                        RntbdRequestHeader = 0x0053 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderForceQueryScan                            RntbdRequestHeader = 0x0054 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderCanOfferReplaceComplete                   RntbdRequestHeader = 0x0056 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderExcludeSystemProperties                   RntbdRequestHeader = 0x0057 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderBinaryId                                  RntbdRequestHeader = 0x0058 // RntbdTokenType.Bytes, required = false
-	RntbdRequestHeaderTimeToLiveInSeconds                       RntbdRequestHeader = 0x0059 // RntbdTokenType.Long, required = false
-	RntbdRequestHeaderEffectivePartitionKey                     RntbdRequestHeader = 0x005A // RntbdTokenType.Bytes, required = false
-	RntbdRequestHeaderBinaryPassthroughRequest                  RntbdRequestHeader = 0x005B // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderUserDefinedTypeName                       RntbdRequestHeader = 0x005C // RntbdTokenType.String, required = false
-	RntbdRequestHeaderEnableDynamicRidRangeAllocation           RntbdRequestHeader = 0x005D // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderEnumerationDirection                      RntbdRequestHeader = 0x005E // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderStartId                                   RntbdRequestHeader = 0x005F // RntbdTokenType.Bytes, required = false
-	RntbdRequestHeaderEndId                                     RntbdRequestHeader = 0x0060 // RntbdTokenType.Bytes, required = false
-	RntbdRequestHeaderFanoutOperationState                      RntbdRequestHeader = 0x0061 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderStartEpk                                  RntbdRequestHeader = 0x0062 // RntbdTokenType.Bytes, required = false
-	RntbdRequestHeaderEndEpk                                    RntbdRequestHeader = 0x0063 // RntbdTokenType.Bytes, required = false
-	RntbdRequestHeaderReadFeedKeyType                           RntbdRequestHeader = 0x0064 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderContentSerializationFormat                RntbdRequestHeader = 0x0065 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderAllowTentativeWrites                      RntbdRequestHeader = 0x0066 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderIsUserRequest                             RntbdRequestHeader = 0x0067 // RntbdTokenType.Byte, required = false
-	RntbdRequestHeaderSharedOfferThroughput                     RntbdRequestHeader = 0x0068 // RntbdTokenType.ULong, required = false
-
-	RntbdRequestHeaderSDKSupportedCapabilities RntbdRequestHeader = 0x00A2 // RntbdTokenType.ULong, required = ?
-)
-
-type RntbdResponseHeaderType uint16
-
-const (
-	RntbdResponseHeaderPayloadPresent               RntbdResponseHeaderType = 0x0000 // RntbdTokenType.Byte, required = true
-	RntbdResponseHeaderLastStateChangeDateTime      RntbdResponseHeaderType = 0x0002 // RntbdTokenType.SmallString, required = false
-	RntbdResponseHeaderContinuationToken            RntbdResponseHeaderType = 0x0003 // RntbdTokenType.String, required = false
-	RntbdResponseHeaderETag                         RntbdResponseHeaderType = 0x0004 // RntbdTokenType.String, required = false
-	RntbdResponseHeaderReadsPerformed               RntbdResponseHeaderType = 0x0007 // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderWritesPerformed              RntbdResponseHeaderType = 0x0008 // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderQueriesPerformed             RntbdResponseHeaderType = 0x0009 // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderIndexTermsGenerated          RntbdResponseHeaderType = 0x000A // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderScriptsExecuted              RntbdResponseHeaderType = 0x000B // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderRetryAfterMilliseconds       RntbdResponseHeaderType = 0x000C // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderIndexingDirective            RntbdResponseHeaderType = 0x000D // RntbdTokenType.Byte, required = false
-	RntbdResponseHeaderStorageMaxResoureQuota       RntbdResponseHeaderType = 0x000E // RntbdTokenType.String, required = false
-	RntbdResponseHeaderStorageResourceQuotaUsage    RntbdResponseHeaderType = 0x000F // RntbdTokenType.String, required = false
-	RntbdResponseHeaderSchemaVersion                RntbdResponseHeaderType = 0x0010 // RntbdTokenType.SmallString, required = false
-	RntbdResponseHeaderCollectionPartitionIndex     RntbdResponseHeaderType = 0x0011 // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderCollectionServiceIndex       RntbdResponseHeaderType = 0x0012 // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderLSN                          RntbdResponseHeaderType = 0x0013 // RntbdTokenType.LongLong, required = false
-	RntbdResponseHeaderItemCount                    RntbdResponseHeaderType = 0x0014 // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderRequestCharge                RntbdResponseHeaderType = 0x0015 // RntbdTokenType.Double, required = false
-	RntbdResponseHeaderOwnerFullName                RntbdResponseHeaderType = 0x0017 // RntbdTokenType.String, required = false
-	RntbdResponseHeaderOwnerId                      RntbdResponseHeaderType = 0x0018 // RntbdTokenType.String, required = false
-	RntbdResponseHeaderDatabaseAccountId            RntbdResponseHeaderType = 0x0019 // RntbdTokenType.String, required = false
-	RntbdResponseHeaderQuorumAckedLSN               RntbdResponseHeaderType = 0x001A // RntbdTokenType.LongLong, required = false
-	RntbdResponseHeaderRequestValidationFailure     RntbdResponseHeaderType = 0x001B // RntbdTokenType.Byte, required = false
-	RntbdResponseHeaderSubStatus                    RntbdResponseHeaderType = 0x001C // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderCollectionUpdateProgress     RntbdResponseHeaderType = 0x001D // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderCurrentWriteQuorum           RntbdResponseHeaderType = 0x001E // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderCurrentReplicaSetSize        RntbdResponseHeaderType = 0x001F // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderCollectionLazyIndexProgress  RntbdResponseHeaderType = 0x0020 // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderPartitionKeyRangeId          RntbdResponseHeaderType = 0x0021 // RntbdTokenType.String, required = false
-	RntbdResponseHeaderLogResults                   RntbdResponseHeaderType = 0x0025 // RntbdTokenType.String, required = false
-	RntbdResponseHeaderXPRole                       RntbdResponseHeaderType = 0x0026 // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderIsRUPerMinuteUsed            RntbdResponseHeaderType = 0x0027 // RntbdTokenType.Byte, required = false
-	RntbdResponseHeaderQueryMetrics                 RntbdResponseHeaderType = 0x0028 // RntbdTokenType.String, required = false
-	RntbdResponseHeaderGlobalCommittedLSN           RntbdResponseHeaderType = 0x0029 // RntbdTokenType.LongLong, required = false
-	RntbdResponseHeaderNumberOfReadRegions          RntbdResponseHeaderType = 0x0030 // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderOfferReplacePending          RntbdResponseHeaderType = 0x0031 // RntbdTokenType.Byte, required = false
-	RntbdResponseHeaderItemLSN                      RntbdResponseHeaderType = 0x0032 // RntbdTokenType.LongLong, required = false
-	RntbdResponseHeaderRestoreState                 RntbdResponseHeaderType = 0x0033 // RntbdTokenType.String, required = false
-	RntbdResponseHeaderCollectionSecurityIdentifier RntbdResponseHeaderType = 0x0034 // RntbdTokenType.String, required = false
-	RntbdResponseHeaderTransportRequestID           RntbdResponseHeaderType = 0x0035 // RntbdTokenType.ULong, required = false
-	RntbdResponseHeaderShareThroughput              RntbdResponseHeaderType = 0x0036 // RntbdTokenType.Byte, required = false
-	RntbdResponseHeaderDisableRntbdChannel          RntbdResponseHeaderType = 0x0038 // RntbdTokenType.Byte, required = false
-	RntbdResponseHeaderServerDateTimeUtc            RntbdResponseHeaderType = 0x0039 // RntbdTokenType.SmallString, required = false
-	RntbdResponseHeaderLocalLSN                     RntbdResponseHeaderType = 0x003A // RntbdTokenType.LongLong, required = false
-	RntbdResponseHeaderQuorumAckedLocalLSN          RntbdResponseHeaderType = 0x003B // RntbdTokenType.LongLong, required = false
-	RntbdResponseHeaderItemLocalLSN                 RntbdResponseHeaderType = 0x003C // RntbdTokenType.LongLong, required = false
-	RntbdResponseHeaderHasTentativeWrites           RntbdResponseHeaderType = 0x003D // RntbdTokenType.Byte, required = false
-	RntbdResponseHeaderSessionToken                 RntbdResponseHeaderType = 0x003E // RntbdTokenType.String, required = false
-)
-
-type RntbdContextHeader uint16
-
-const (
-	RntbdContextHeaderProtocolVersion                 RntbdContextHeader = 0x0000 // RntbdTokenType.ULong, required = false
-	RntbdContextHeaderClientVersion                   RntbdContextHeader = 0x0001 // RntbdTokenType.SmallString, required = false
-	RntbdContextHeaderServerAgent                     RntbdContextHeader = 0x0002 // RntbdTokenType.SmallString, required = true
-	RntbdContextHeaderServerVersion                   RntbdContextHeader = 0x0003 // RntbdTokenType.SmallString, required = true
-	RntbdContextHeaderIdleTimeoutInSeconds            RntbdContextHeader = 0x0004 // RntbdTokenType.ULong, required = false
-	RntbdContextHeaderUnauthenticatedTimeoutInSeconds RntbdContextHeader = 0x0005 // RntbdTokenType.ULong, required = false
-)
-
-type RntbdTokenType uint8
-
-const (
-	RntbdTokenTypeByte        RntbdTokenType = 0x00 // 8bit boolean
-	RntbdTokenTypeUShort      RntbdTokenType = 0x01 // 16bit unsigned integer
-	RntbdTokenTypeULong       RntbdTokenType = 0x02 // 32bit unsigned integer
-	RntbdTokenTypeLong        RntbdTokenType = 0x03 // 32bit signed integer
-	RntbdTokenTypeULongLong   RntbdTokenType = 0x04 // 64bit unsigned integer
-	RntbdTokenTypeLongLong    RntbdTokenType = 0x05 // 64bit signed integer
-	RntbdTokenTypeGuid        RntbdTokenType = 0x06 // 128bit GUID
-	RntbdTokenTypeSmallString RntbdTokenType = 0x07 // 8bit len + string
-	RntbdTokenTypeString      RntbdTokenType = 0x08 // 16bit len + string
-	RntbdTokenTypeULongString RntbdTokenType = 0x09 // 32bit len + string
-	RntbdTokenTypeSmallBytes  RntbdTokenType = 0x0A // 8bit len + bytes
-	RntbdTokenTypeBytes       RntbdTokenType = 0x0B // 16bit len + bytes
-	RntbdTokenTypeULongBytes  RntbdTokenType = 0x0C // 32bit len + bytes
-	RntbdTokenTypeFloat       RntbdTokenType = 0x0D // 32bit float
-	RntbdTokenTypeDouble      RntbdTokenType = 0x0E // 64bit double
-	RntbdTokenTypeInvalid     RntbdTokenType = 0x0F // Invalid token type
-)
-
-func (h RntbdRequestHeader) String() string {
-	switch h {
-	case RntbdRequestHeaderResourceId:
-		return "RntbdRequestHeaderResourceId"
-	case RntbdRequestHeaderAuthorizationToken:
-		return headers.Authorization
-	case RntbdRequestHeaderPayloadPresent:
-		return "RntbdRequestHeaderPayloadPresent"
-	case RntbdRequestHeaderDate:
-		return headers.XDate
-	case RntbdRequestHeaderPageSize:
-		return "RntbdRequestHeaderPageSize"
-	case RntbdRequestHeaderSessionToken:
-		return "RntbdRequestHeaderSessionToken"
-	case RntbdRequestHeaderContinuationToken:
-		return "RntbdRequestHeaderContinuationToken"
-	case RntbdRequestHeaderIndexingDirective:
-		return "RntbdRequestHeaderIndexingDirective"
-	case RntbdRequestHeaderMatch:
-		return "RntbdRequestHeaderMatch"
-	case RntbdRequestHeaderPreTriggerInclude:
-		return "RntbdRequestHeaderPreTriggerInclude"
-	case RntbdRequestHeaderPostTriggerInclude:
-		return "RntbdRequestHeaderPostTriggerInclude"
-	case RntbdRequestHeaderIsFanout:
-		return "RntbdRequestHeaderIsFanout"
-	case RntbdRequestHeaderCollectionPartitionIndex:
-		return "RntbdRequestHeaderCollectionPartitionIndex"
-	case RntbdRequestHeaderCollectionServiceIndex:
-		return "RntbdRequestHeaderCollectionServiceIndex"
-	case RntbdRequestHeaderPreTriggerExclude:
-		return "RntbdRequestHeaderPreTriggerExclude"
-	case RntbdRequestHeaderPostTriggerExclude:
-		return "RntbdRequestHeaderPostTriggerExclude"
-	case RntbdRequestHeaderConsistencyLevel:
-		return headers.ConsistencyLevel
-	case RntbdRequestHeaderEntityId:
-		return "RntbdRequestHeaderEntityId"
-	case RntbdRequestHeaderResourceSchemaName:
-		return "RntbdRequestHeaderResourceSchemaName"
-	case RntbdRequestHeaderReplicaPath:
-		return "RntbdRequestHeaderReplicaPath"
-	case RntbdRequestHeaderResourceTokenExpiry:
-		return "RntbdRequestHeaderResourceTokenExpiry"
-	case RntbdRequestHeaderDatabaseName:
-		return "RntbdRequestHeaderDatabaseName"
-	case RntbdRequestHeaderCollectionName:
-		return "RntbdRequestHeaderCollectionName"
-	case RntbdRequestHeaderDocumentName:
-		return "RntbdRequestHeaderDocumentName"
-	case RntbdRequestHeaderAttachmentName:
-		return "RntbdRequestHeaderAttachmentName"
-	case RntbdRequestHeaderUserName:
-		return "RntbdRequestHeaderUserName"
-	case RntbdRequestHeaderPermissionName:
-		return "RntbdRequestHeaderPermissionName"
-	case RntbdRequestHeaderStoredProcedureName:
-		return "RntbdRequestHeaderStoredProcedureName"
-	case RntbdRequestHeaderUserDefinedFunctionName:
-		return "RntbdRequestHeaderUserDefinedFunctionName"
-	case RntbdRequestHeaderTriggerName:
-		return "RntbdRequestHeaderTriggerName"
-	case RntbdRequestHeaderEnableScanInQuery:
-		return "RntbdRequestHeaderEnableScanInQuery"
-	case RntbdRequestHeaderEmitVerboseTracesInQuery:
-		return "RntbdRequestHeaderEmitVerboseTracesInQuery"
-	case RntbdRequestHeaderConflictName:
-		return "RntbdRequestHeaderConflictName"
-	case RntbdRequestHeaderBindReplicaDirective:
-		return "RntbdRequestHeaderBindReplicaDirective"
-	case RntbdRequestHeaderPrimaryMasterKey:
-		return "RntbdRequestHeaderPrimaryMasterKey"
-	case RntbdRequestHeaderSecondaryMasterKey:
-		return "RntbdRequestHeaderSecondaryMasterKey"
-	case RntbdRequestHeaderPrimaryReadonlyKey:
-		return "RntbdRequestHeaderPrimaryReadonlyKey"
-	case RntbdRequestHeaderSecondaryReadonlyKey:
-		return "RntbdRequestHeaderSecondaryReadonlyKey"
-	case RntbdRequestHeaderProfileRequest:
-		return "RntbdRequestHeaderProfileRequest"
-	case RntbdRequestHeaderEnableLowPrecisionOrderBy:
-		return "RntbdRequestHeaderEnableLowPrecisionOrderBy"
-	case RntbdRequestHeaderClientVersion:
-		return "RntbdRequestHeaderClientVersion"
-	case RntbdRequestHeaderCanCharge:
-		return "RntbdRequestHeaderCanCharge"
-	case RntbdRequestHeaderCanThrottle:
-		return "RntbdRequestHeaderCanThrottle"
-	case RntbdRequestHeaderPartitionKey:
-		return "RntbdRequestHeaderPartitionKey"
-	case RntbdRequestHeaderPartitionKeyRangeId:
-		return "RntbdRequestHeaderPartitionKeyRangeId"
-	case RntbdRequestHeaderNotUsed2D:
-		return "RntbdRequestHeaderNotUsed2D"
-	case RntbdRequestHeaderNotUsed2E:
-		return "RntbdRequestHeaderNotUsed2E"
-	case RntbdRequestHeaderNotUsed2F:
-		return "RntbdRequestHeaderNotUsed2F"
-	case RntbdRequestHeaderMigrateCollectionDirective:
-		return "RntbdRequestHeaderMigrateCollectionDirective"
-	case RntbdRequestHeaderNotUsed32:
-		return "RntbdRequestHeaderNotUsed32"
-	case RntbdRequestHeaderSupportSpatialLegacyCoordinates:
-		return "RntbdRequestHeaderSupportSpatialLegacyCoordinates"
-	case RntbdRequestHeaderPartitionCount:
-		return "RntbdRequestHeaderPartitionCount"
-	case RntbdRequestHeaderCollectionRid:
-		return "RntbdRequestHeaderCollectionRid"
-	case RntbdRequestHeaderPartitionKeyRangeName:
-		return "RntbdRequestHeaderPartitionKeyRangeName"
-	case RntbdRequestHeaderSchemaName:
-		return "RntbdRequestHeaderSchemaName"
-	case RntbdRequestHeaderFilterBySchemaRid:
-		return "RntbdRequestHeaderFilterBySchemaRid"
-	case RntbdRequestHeaderUsePolygonsSmallerThanAHemisphere:
-		return "RntbdRequestHeaderUsePolygonsSmallerThanAHemisphere"
-	case RntbdRequestHeaderGatewaySignature:
-		return "RntbdRequestHeaderGatewaySignature"
-	case RntbdRequestHeaderEnableLogging:
-		return "RntbdRequestHeaderEnableLogging"
-	case RntbdRequestHeaderAIM:
-		return headers.AIM
-	case RntbdRequestHeaderPopulateQuotaInfo:
-		return "RntbdRequestHeaderPopulateQuotaInfo"
-	case RntbdRequestHeaderDisableRUPerMinuteUsage:
-		return "RntbdRequestHeaderDisableRUPerMinuteUsage"
-	case RntbdRequestHeaderPopulateQueryMetrics:
-		return "RntbdRequestHeaderPopulateQueryMetrics"
-	case RntbdRequestHeaderResponseContinuationTokenLimitInKb:
-		return "RntbdRequestHeaderResponseContinuationTokenLimitInKb"
-	case RntbdRequestHeaderPopulatePartitionStatistics:
-		return "RntbdRequestHeaderPopulatePartitionStatistics"
-	case RntbdRequestHeaderRemoteStorageType:
-		return "RntbdRequestHeaderRemoteStorageType"
-	case RntbdRequestHeaderCollectionRemoteStorageSecurityIdentifier:
-		return "RntbdRequestHeaderCollectionRemoteStorageSecurityIdentifier"
-	case RntbdRequestHeaderIfModifiedSince:
-		return "RntbdRequestHeaderIfModifiedSince"
-	case RntbdRequestHeaderPopulateCollectionThroughputInfo:
-		return "RntbdRequestHeaderPopulateCollectionThroughputInfo"
-	case RntbdRequestHeaderRemainingTimeInMsOnClientRequest:
-		return headers.RemainingTimeInMsOnClient
-	case RntbdRequestHeaderClientRetryAttemptCount:
-		return headers.ClientRetryAttemptCount
-	case RntbdRequestHeaderTargetLsn:
-		return "RntbdRequestHeaderTargetLsn"
-	case RntbdRequestHeaderTargetGlobalCommittedLsn:
-		return "RntbdRequestHeaderTargetGlobalCommittedLsn"
-	case RntbdRequestHeaderTransportRequestID:
-		return "RntbdRequestHeaderTransportRequestID"
-	case RntbdRequestHeaderRestoreMetadaFilter:
-		return "RntbdRequestHeaderRestoreMetadaFilter"
-	case RntbdRequestHeaderRestoreParams:
-		return "RntbdRequestHeaderRestoreParams"
-	case RntbdRequestHeaderShareThroughput:
-		return "RntbdRequestHeaderShareThroughput"
-	case RntbdRequestHeaderPartitionResourceFilter:
-		return "RntbdRequestHeaderPartitionResourceFilter"
-	case RntbdRequestHeaderIsReadOnlyScript:
-		return "RntbdRequestHeaderIsReadOnlyScript"
-	case RntbdRequestHeaderIsAutoScaleRequest:
-		return "RntbdRequestHeaderIsAutoScaleRequest"
-	case RntbdRequestHeaderForceQueryScan:
-		return "RntbdRequestHeaderForceQueryScan"
-	case RntbdRequestHeaderCanOfferReplaceComplete:
-		return "RntbdRequestHeaderCanOfferReplaceComplete"
-	case RntbdRequestHeaderExcludeSystemProperties:
-		return "RntbdRequestHeaderExcludeSystemProperties"
-	case RntbdRequestHeaderBinaryId:
-		return "RntbdRequestHeaderBinaryId"
-	case RntbdRequestHeaderTimeToLiveInSeconds:
-		return "RntbdRequestHeaderTimeToLiveInSeconds"
-	case RntbdRequestHeaderEffectivePartitionKey:
-		return "RntbdRequestHeaderEffectivePartitionKey"
-	case RntbdRequestHeaderBinaryPassthroughRequest:
-		return "RntbdRequestHeaderBinaryPassthroughRequest"
-	case RntbdRequestHeaderUserDefinedTypeName:
-		return "RntbdRequestHeaderUserDefinedTypeName"
-	case RntbdRequestHeaderEnableDynamicRidRangeAllocation:
-		return "RntbdRequestHeaderEnableDynamicRidRangeAllocation"
-	case RntbdRequestHeaderEnumerationDirection:
-		return "RntbdRequestHeaderEnumerationDirection"
-	case RntbdRequestHeaderStartId:
-		return "RntbdRequestHeaderStartId"
-	case RntbdRequestHeaderEndId:
-		return "RntbdRequestHeaderEndId"
-	case RntbdRequestHeaderFanoutOperationState:
-		return "RntbdRequestHeaderFanoutOperationState"
-	case RntbdRequestHeaderStartEpk:
-		return "RntbdRequestHeaderStartEpk"
-	case RntbdRequestHeaderEndEpk:
-		return "RntbdRequestHeaderEndEpk"
-	case RntbdRequestHeaderReadFeedKeyType:
-		return "RntbdRequestHeaderReadFeedKeyType"
-	case RntbdRequestHeaderContentSerializationFormat:
-		return "RntbdRequestHeaderContentSerializationFormat"
-	case RntbdRequestHeaderAllowTentativeWrites:
-		return "RntbdRequestHeaderAllowTentativeWrites"
-	case RntbdRequestHeaderIsUserRequest:
-		return "RntbdRequestHeaderIsUserRequest"
-	case RntbdRequestHeaderSharedOfferThroughput:
-		return "RntbdRequestHeaderSharedOfferThroughput"
-	case RntbdRequestHeaderSDKSupportedCapabilities:
-		return headers.SupportedCapabilities
-	}
-
-	return fmt.Sprintf("RntbdRequestHeader(%d)", h)
-}
-
-func (h RntbdContextHeader) String() string {
-	switch h {
-	case RntbdContextHeaderProtocolVersion:
-		return "RntbdContextHeaderProtocolVersion"
-	case RntbdContextHeaderClientVersion:
-		return "RntbdContextHeaderClientVersion"
-	case RntbdContextHeaderServerAgent:
-		return "RntbdContextHeaderServerAgent"
-	case RntbdContextHeaderServerVersion:
-		return "RntbdContextHeaderServerVersion"
-	case RntbdContextHeaderIdleTimeoutInSeconds:
-		return "RntbdContextHeaderIdleTimeoutInSeconds"
-	case RntbdContextHeaderUnauthenticatedTimeoutInSeconds:
-		return "RntbdContextHeaderUnauthenticatedTimeoutInSeconds"
-	}
-
-	return fmt.Sprintf("RntbdContextHeader(%d)", h)
-}
-
-func (h RntbdResponseHeaderType) String() string {
-	switch h {
-	case RntbdResponseHeaderPayloadPresent:
-		return "PayloadPresent"
-	case RntbdResponseHeaderLastStateChangeDateTime:
-		return "LastStateChangeDateTime"
-	case RntbdResponseHeaderContinuationToken:
-		return "ContinuationToken"
-	case RntbdResponseHeaderETag:
-		return "ETag"
-	case RntbdResponseHeaderReadsPerformed:
-		return "ReadsPerformed"
-	case RntbdResponseHeaderWritesPerformed:
-		return "WritesPerformed"
-	case RntbdResponseHeaderQueriesPerformed:
-		return "QueriesPerformed"
-	case RntbdResponseHeaderIndexTermsGenerated:
-		return "IndexTermsGenerated"
-	case RntbdResponseHeaderScriptsExecuted:
-		return "ScriptsExecuted"
-	case RntbdResponseHeaderRetryAfterMilliseconds:
-		return "RetryAfterMilliseconds"
-	case RntbdResponseHeaderIndexingDirective:
-		return "IndexingDirective"
-	case RntbdResponseHeaderStorageMaxResoureQuota:
-		return "StorageMaxResoureQuota"
-	case RntbdResponseHeaderStorageResourceQuotaUsage:
-		return "StorageResourceQuotaUsage"
-	case RntbdResponseHeaderSchemaVersion:
-		return "SchemaVersion"
-	case RntbdResponseHeaderCollectionPartitionIndex:
-		return "CollectionPartitionIndex"
-	case RntbdResponseHeaderCollectionServiceIndex:
-		return "CollectionServiceIndex"
-	case RntbdResponseHeaderLSN:
-		return "LSN"
-	case RntbdResponseHeaderItemCount:
-		return "ItemCount"
-	case RntbdResponseHeaderRequestCharge:
-		return "RequestCharge"
-	case RntbdResponseHeaderOwnerFullName:
-		return "OwnerFullName"
-	case RntbdResponseHeaderOwnerId:
-		return "OwnerId"
-	case RntbdResponseHeaderDatabaseAccountId:
-		return "DatabaseAccountId"
-	case RntbdResponseHeaderQuorumAckedLSN:
-		return "QuorumAckedLSN"
-	case RntbdResponseHeaderRequestValidationFailure:
-		return "RequestValidationFailure"
-	case RntbdResponseHeaderSubStatus:
-		return "SubStatus"
-	case RntbdResponseHeaderCollectionUpdateProgress:
-		return "CollectionUpdateProgress"
-	case RntbdResponseHeaderCurrentWriteQuorum:
-		return "CurrentWriteQuorum"
-	case RntbdResponseHeaderCurrentReplicaSetSize:
-		return "CurrentReplicaSetSize"
-	case RntbdResponseHeaderCollectionLazyIndexProgress:
-		return "CollectionLazyIndexProgress"
-	case RntbdResponseHeaderPartitionKeyRangeId:
-		return "PartitionKeyRangeId"
-	case RntbdResponseHeaderLogResults:
-		return "LogResults"
-	case RntbdResponseHeaderXPRole:
-		return "XPRole"
-	case RntbdResponseHeaderIsRUPerMinuteUsed:
-		return "IsRUPerMinuteUsed"
-	case RntbdResponseHeaderQueryMetrics:
-		return "QueryMetrics"
-	case RntbdResponseHeaderGlobalCommittedLSN:
-		return "GlobalCommittedLSN"
-	case RntbdResponseHeaderNumberOfReadRegions:
-		return "NumberOfReadRegions"
-	case RntbdResponseHeaderOfferReplacePending:
-		return "OfferReplacePending"
-	case RntbdResponseHeaderItemLSN:
-		return "ItemLSN"
-	case RntbdResponseHeaderRestoreState:
-		return "RestoreState"
-	case RntbdResponseHeaderCollectionSecurityIdentifier:
-		return "CollectionSecurityIdentifier"
-	case RntbdResponseHeaderTransportRequestID:
-		return "TransportRequestID"
-	case RntbdResponseHeaderShareThroughput:
-		return "ShareThroughput"
-	case RntbdResponseHeaderDisableRntbdChannel:
-		return "DisableRntbdChannel"
-	case RntbdResponseHeaderServerDateTimeUtc:
-		return "ServerDateTimeUtc"
-	case RntbdResponseHeaderLocalLSN:
-		return "LocalLSN"
-	case RntbdResponseHeaderQuorumAckedLocalLSN:
-		return "QuorumAckedLocalLSN"
-	case RntbdResponseHeaderItemLocalLSN:
-		return "ItemLocalLSN"
-	case RntbdResponseHeaderHasTentativeWrites:
-		return "HasTentativeWrites"
-	case RntbdResponseHeaderSessionToken:
-		return "SessionToken"
-	}
-
-	return fmt.Sprintf("RntbdResponseHeaderType(%d)", h)
-}
-
-func (r RntbdResourceType) String() string {
-	switch r {
-	case RntbdResourceTypeConnection:
-		return "Connection"
-	case RntbdResourceTypeDatabase:
-		return "Database"
-	case RntbdResourceTypeCollection:
-		return "Collection"
-	case RntbdResourceTypeDocument:
-		return "Document"
-	case RntbdResourceTypeAttachment:
-		return "Attachment"
-	case RntbdResourceTypeUser:
-		return "User"
-	case RntbdResourceTypePermission:
-		return "Permission"
-	case RntbdResourceTypeStoredProcedure:
-		return "StoredProcedure"
-	case RntbdResourceTypeConflict:
-		return "Conflict"
-	case RntbdResourceTypeTrigger:
-		return "Trigger"
-	case RntbdResourceTypeUserDefinedFunction:
-		return "UserDefinedFunction"
-	case RntbdResourceTypeModule:
-		return "Module"
-	case RntbdResourceTypeReplica:
-		return "Replica"
-	case RntbdResourceTypeModuleCommand:
-		return "ModuleCommand"
-	case RntbdResourceTypeRecord:
-		return "Record"
-	case RntbdResourceTypeOffer:
-		return "Offer"
-	case RntbdResourceTypePartitionSetInformation:
-		return "PartitionSetInformation"
-	case RntbdResourceTypeXPReplicatorAddress:
-		return "XPReplicatorAddress"
-	case RntbdResourceTypeMasterPartition:
-		return "MasterPartition"
-	case RntbdResourceTypeServerPartition:
-		return "ServerPartition"
-	case RntbdResourceTypeDatabaseAccount:
-		return "DatabaseAccount"
-	case RntbdResourceTypeTopology:
-		return "Topology"
-	case RntbdResourceTypePartitionKeyRange:
-		return "PartitionKeyRange"
-	case RntbdResourceTypeSchema:
-		return "Schema"
-	case RntbdResourceTypeBatchApply:
-		return "BatchApply"
-	case RntbdResourceTypeRestoreMetadata:
-		return "RestoreMetadata"
-	case RntbdResourceTypeComputeGatewayCharges:
-		return "ComputeGatewayCharges"
-	case RntbdResourceTypeRidRange:
-		return "RidRange"
-	case RntbdResourceTypeUserDefinedType:
-		return "UserDefinedType"
-	}
-
-	return fmt.Sprintf("RntbdResourceType(%d)", r)
-}
-
-func (o RntbdOperationType) String() string {
-	switch o {
-	case RntbdOperationTypeConnection:
-		return "Connection"
-	case RntbdOperationTypeCreate:
-		return "Create"
-	case RntbdOperationTypeUpdate:
-		return "Update"
-	case RntbdOperationTypeRead:
-		return "Read"
-	case RntbdOperationTypeReadFeed:
-		return "ReadFeed"
-	case RntbdOperationTypeDelete:
-		return "Delete"
-	case RntbdOperationTypeReplace:
-		return "Replace"
-	case RntbdOperationTypeExecuteJavaScript:
-		return "ExecuteJavaScript"
-	case RntbdOperationTypeSQLQuery:
-		return "SQLQuery"
-	case RntbdOperationTypePause:
-		return "Pause"
-	case RntbdOperationTypeResume:
-		return "Resume"
-	case RntbdOperationTypeStop:
-		return "Stop"
-	case RntbdOperationTypeRecycle:
-		return "Recycle"
-	case RntbdOperationTypeCrash:
-		return "Crash"
-	case RntbdOperationTypeQuery:
-		return "Query"
-	case RntbdOperationTypeForceConfigRefresh:
-		return "ForceConfigRefresh"
-	case RntbdOperationTypeHead:
-		return "Head"
-	case RntbdOperationTypeHeadFeed:
-		return "HeadFeed"
-	case RntbdOperationTypeUpsert:
-		return "Upsert"
-	case RntbdOperationTypeRecreate:
-		return "Recreate"
-	case RntbdOperationTypeThrottle:
-		return "Throttle"
-	case RntbdOperationTypeGetSplitPoint:
-		return "GetSplitPoint"
-	case RntbdOperationTypePreCreateValidation:
-		return "PreCreateValidation"
-	case RntbdOperationTypeBatchApply:
-		return "BatchApply"
-	case RntbdOperationTypeAbortSplit:
-		return "AbortSplit"
-	case RntbdOperationTypeCompleteSplit:
-		return "CompleteSplit"
-	case RntbdOperationTypeOfferUpdateOperation:
-		return "OfferUpdateOperation"
-	case RntbdOperationTypeOfferPreGrowValidation:
-		return "OfferPreGrowValidation"
-	case RntbdOperationTypeBatchReportThroughputUtilization:
-		return "BatchReportThroughputUtilization"
-	case RntbdOperationTypeCompletePartitionMigration:
-		return "CompletePartitionMigration"
-	case RntbdOperationTypeAbortPartitionMigration:
-		return "AbortPartitionMigration"
-	case RntbdOperationTypePreReplaceValidation:
-		return "PreReplaceValidation"
-	case RntbdOperationTypeAddComputeGatewayRequestCharges:
-		return "AddComputeGatewayRequestCharges"
-	case RntbdOperationTypeMigratePartition:
-		return "MigratePartition"
-	}
-
-	return fmt.Sprintf("RntbdOperationType(%d)", o)
-}
@@ -1,128 +0,0 @@
-package rntbd
-
-import (
-	"bytes"
-	"fmt"
-	"io"
-	"net/http"
-	"net/http/httptest"
-	"net/url"
-	"strconv"
-
-	"github.com/pikami/cosmium/api/headers"
-)
-
-func (f *RntbdFrame) ToHttpRequest() *http.Request {
-	req := &http.Request{
-		Method: operationTypeToHttpMethod(f.OperationType),
-		URL:    &url.URL{Path: frameToPath(f)},
-		Body:   io.NopCloser(bytes.NewReader(f.Payload)),
-		Header: http.Header{},
-	}
-
-	switch f.OperationType {
-	case RntbdOperationTypeQuery, RntbdOperationTypeSQLQuery:
-		req.Header.Set(headers.Query, "true")
-	case RntbdOperationTypeUpsert:
-		req.Header.Set(headers.IsUpsert, "true")
-	}
-
-	if ifMatch, ok := f.RequestHeaders[RntbdRequestHeaderMatch]; ok {
-		if ifMatchString, ok := ifMatch.(string); ok {
-			req.Header.Set(headers.IfMatch, ifMatchString)
-		}
-	}
-
-	if continuationToken, ok := f.RequestHeaders[RntbdRequestHeaderContinuationToken]; ok {
-		if continuationTokenString, ok := continuationToken.(string); ok {
-			req.Header.Set(headers.ContinuationToken, continuationTokenString)
-		}
-	}
-
-	if maxItemCount, ok := f.RequestHeaders[RntbdRequestHeaderPageSize]; ok {
-		if maxItemCountString, ok := maxItemCount.(uint64); ok {
-			req.Header.Set(headers.MaxItemCount, fmt.Sprintf("%d", maxItemCountString))
-		}
-	}
-
-	return req
-}
-
-func ToRntbdResponseFrame(responseWriter *httptest.ResponseRecorder) *RntbdResponseFrameBuilder {
-	builder := &RntbdResponseFrameBuilder{}
-	builder.SetStatusCode(uint16(responseWriter.Code))
-
-	if responseWriter.Header().Get(headers.ETag) != "" {
-		builder.AddHeader(uint16(RntbdResponseHeaderETag), RntbdTokenTypeString, responseWriter.Header().Get(headers.ETag))
-	}
-
-	if responseWriter.Header().Get(headers.ContinuationToken) != "" {
-		builder.AddHeader(uint16(RntbdResponseHeaderContinuationToken), RntbdTokenTypeString, responseWriter.Header().Get(headers.ContinuationToken))
-	}
-
-	if responseWriter.Header().Get(headers.ItemCount) != "" {
-		itemCount, err := strconv.ParseUint(responseWriter.Header().Get(headers.ItemCount), 10, 32)
-		if err != nil {
-			panic(err)
-		}
-
-		builder.AddHeader(uint16(RntbdResponseHeaderItemCount), RntbdTokenTypeULong, uint32(itemCount))
-	}
-
-	if responseWriter.Body.Len() > 0 {
-		builder.AddHeader(uint16(RntbdResponseHeaderPayloadPresent), RntbdTokenTypeByte, []byte{1})
-		builder.AddPayload(responseWriter.Body.Bytes())
-	} else {
-		builder.AddHeader(uint16(RntbdResponseHeaderPayloadPresent), RntbdTokenTypeByte, []byte{0})
-	}
-
-	return builder
-}
-
-func operationTypeToHttpMethod(operationType RntbdOperationType) string {
-	switch operationType {
-	case RntbdOperationTypeRead,
-		RntbdOperationTypeReadFeed:
-		return http.MethodGet
-	case RntbdOperationTypeCreate,
-		RntbdOperationTypeUpsert,
-		RntbdOperationTypeQuery,
-		RntbdOperationTypeSQLQuery:
-		return http.MethodPost
-	case RntbdOperationTypeUpdate,
-		RntbdOperationTypeReplace:
-		return http.MethodPut
-	case RntbdOperationTypeDelete:
-		return http.MethodDelete
-	}
-
-	panic(fmt.Sprintf("Unknown operation type: %d", operationType))
-}
-
-func frameToPath(frame *RntbdFrame) string {
-	databaseName, databaseOk := frame.RequestHeaders[RntbdRequestHeaderDatabaseName]
-	collectionName, collectionOk := frame.RequestHeaders[RntbdRequestHeaderCollectionName]
-	documentName, documentOk := frame.RequestHeaders[RntbdRequestHeaderDocumentName]
-
-	urlPath := ""
-
-	if databaseOk {
-		urlPath += fmt.Sprintf("/dbs/%s", databaseName)
-	} else if frame.ResourceType == RntbdResourceTypeDatabase {
-		urlPath += "/dbs"
-	}
-
-	if collectionOk {
-		urlPath += fmt.Sprintf("/colls/%s", collectionName)
-	} else if frame.ResourceType == RntbdResourceTypeCollection {
-		urlPath += "/colls"
-	}
-
-	if documentOk {
-		urlPath += fmt.Sprintf("/docs/%s", documentName)
-	} else if frame.ResourceType == RntbdResourceTypeDocument {
-		urlPath += "/docs"
-	}
-
-	return urlPath
-}
@@ -1,212 +0,0 @@
-package rntbd
-
-import (
-	"bufio"
-	"bytes"
-	"encoding/binary"
-	"encoding/hex"
-	"fmt"
-	"io"
-	"os"
-
-	"github.com/pikami/cosmium/internal/logger"
-)
-
-type RntbdFrame struct {
-	ResourceType    RntbdResourceType
-	OperationType   RntbdOperationType
-	ActivityId      []byte
-	RequestHeaders  map[RntbdRequestHeader]any
-	ResponseHeaders map[RntbdResponseHeaderType]any
-	ContextHeaders  map[RntbdContextHeader]any
-	Payload         []byte
-}
-
-func ReadFrame(reader *bufio.Reader) (*RntbdFrame, error) {
-	sizeBytes := readBytes(reader, 4)
-	size := binary.LittleEndian.Uint32(sizeBytes)
-
-	payload := readBytes(reader, int(size)-4)
-
-	frame, err := parseFrame_Int(payload, false)
-	if err != nil {
-		return nil, err
-	}
-
-	if payloadPresent, ok := frame.RequestHeaders[RntbdRequestHeaderPayloadPresent]; ok && payloadPresent.([]byte)[0] == 1 {
-		payloadSize := binary.LittleEndian.Uint32(readBytes(reader, 4))
-		payload := readBytes(reader, int(payloadSize))
-		frame.Payload = payload
-	}
-
-	if payloadPresent, ok := frame.ResponseHeaders[RntbdResponseHeaderPayloadPresent]; ok && payloadPresent.([]byte)[0] == 1 {
-		payloadSize := binary.LittleEndian.Uint32(readBytes(reader, 4))
-		payload := readBytes(reader, int(payloadSize))
-		frame.Payload = payload
-	}
-
-	return frame, nil
-}
-
-func ParseFrame(data []byte, isResponse bool) (*RntbdFrame, error) {
-	if len(data) < 4 {
-		return nil, fmt.Errorf("data too short")
-	}
-
-	reader := bufio.NewReader(bytes.NewReader(data))
-	sizeBytes := readBytes(reader, 4)
-	size := binary.LittleEndian.Uint32(sizeBytes)
-
-	payload := readBytes(reader, int(size)-4)
-	frame, err := parseFrame_Int(payload, isResponse)
-	if err != nil {
-		return nil, err
-	}
-
-	if payloadPresent, ok := frame.RequestHeaders[RntbdRequestHeaderPayloadPresent]; ok && payloadPresent.([]byte)[0] == 1 {
-		payloadSize := binary.LittleEndian.Uint32(readBytes(reader, 4))
-		payload := readBytes(reader, int(payloadSize))
-		frame.Payload = payload
-	}
-
-	if payloadPresent, ok := frame.ResponseHeaders[RntbdResponseHeaderPayloadPresent]; ok && payloadPresent.([]byte)[0] == 1 {
-		payloadSize := binary.LittleEndian.Uint32(readBytes(reader, 4))
-		payload := readBytes(reader, int(payloadSize))
-		frame.Payload = payload
-	}
-
-	leftOverBytes, err := io.ReadAll(reader)
-	if err != nil {
-		logger.ErrorLn("Error reading leftOverBytes:", err)
-	}
-
-	if len(leftOverBytes) > 0 {
-		logger.ErrorLn("Left over bytes:", hex.EncodeToString(leftOverBytes))
-	}
-
-	return frame, nil
-}
-
-func parseFrame_Int(data []byte, isResponse bool) (*RntbdFrame, error) {
-	payloadReader := bufio.NewReader(bytes.NewReader(data))
-
-	resourceTypeBytes := readBytes(payloadReader, 2)
-	resourceType := binary.LittleEndian.Uint16(resourceTypeBytes)
-
-	operationTypeBytes := readBytes(payloadReader, 2)
-	operationType := RntbdOperationType(binary.LittleEndian.Uint16(operationTypeBytes))
-
-	activityIdBytes := readBytes(payloadReader, 16)
-
-	requestHeaders := make(map[RntbdRequestHeader]any)
-	responseHeaders := make(map[RntbdResponseHeaderType]any)
-	contextHeaders := make(map[RntbdContextHeader]any)
-	for {
-		if _, err := payloadReader.Peek(1); err != nil {
-			break
-		}
-
-		headerIdBytes := readBytes(payloadReader, 2)
-		headerId := binary.LittleEndian.Uint16(headerIdBytes)
-
-		token, err := parseRntbdToken(payloadReader)
-		if err != nil {
-			return nil, err
-		}
-
-		if resourceType == uint16(RntbdResourceTypeConnection) {
-			contextHeaders[RntbdContextHeader(headerId)] = token
-		} else if isResponse {
-			responseHeaders[RntbdResponseHeaderType(headerId)] = token
-		} else {
-			requestHeaders[RntbdRequestHeader(headerId)] = token
-		}
-	}
-
-	return &RntbdFrame{
-		ResourceType:    RntbdResourceType(resourceType),
-		OperationType:   RntbdOperationType(operationType),
-		ActivityId:      activityIdBytes,
-		RequestHeaders:  requestHeaders,
-		ResponseHeaders: responseHeaders,
-		ContextHeaders:  contextHeaders,
-	}, nil
-}
-
-func parseRntbdToken(reader *bufio.Reader) (any, error) {
-	tokenTypeBytes := readBytes(reader, 1)
-	tokenType := RntbdTokenType(tokenTypeBytes[0])
-
-	switch tokenType {
-	case RntbdTokenTypeByte:
-		token := readBytes(reader, 1)
-		return token, nil
-	case RntbdTokenTypeUShort:
-		token := binary.LittleEndian.Uint16(readBytes(reader, 2))
-		return token, nil
-	case RntbdTokenTypeULong:
-		token := binary.LittleEndian.Uint32(readBytes(reader, 4))
-		return token, nil
-	case RntbdTokenTypeLong:
-		token := int32(binary.LittleEndian.Uint32(readBytes(reader, 4)))
-		return token, nil
-	case RntbdTokenTypeULongLong:
-		token := binary.LittleEndian.Uint64(readBytes(reader, 8))
-		return token, nil
-	case RntbdTokenTypeLongLong:
-		token := int64(binary.LittleEndian.Uint64(readBytes(reader, 8)))
-		return token, nil
-	case RntbdTokenTypeGuid:
-		token := readBytes(reader, 16)
-		return token, nil
-	case RntbdTokenTypeSmallString:
-		lengthBytes := readBytes(reader, 1)
-		length := uint8(lengthBytes[0])
-		token := readBytes(reader, int(length))
-		return string(token), nil
-	case RntbdTokenTypeString:
-		length := binary.LittleEndian.Uint16(readBytes(reader, 2))
-		token := readBytes(reader, int(length))
-		return string(token), nil
-	case RntbdTokenTypeULongString:
-		length := binary.LittleEndian.Uint32(readBytes(reader, 4))
-		token := readBytes(reader, int(length))
-		return string(token), nil
-	case RntbdTokenTypeSmallBytes:
-		lengthBytes := readBytes(reader, 1)
-		length := uint8(lengthBytes[0])
-		token := readBytes(reader, int(length))
-		return token, nil
-	case RntbdTokenTypeBytes:
-		length := binary.LittleEndian.Uint16(readBytes(reader, 2))
-		token := readBytes(reader, int(length))
-		return token, nil
-	case RntbdTokenTypeULongBytes:
-		length := binary.LittleEndian.Uint32(readBytes(reader, 4))
-		token := readBytes(reader, int(length))
-		return token, nil
-	case RntbdTokenTypeFloat:
-		// I can't be bothered to implement this, let's just return a byte array
-		token := readBytes(reader, 4)
-		return token, nil
-	case RntbdTokenTypeDouble:
-		// I can't be bothered to implement this, let's just return a byte array
-		token := readBytes(reader, 8)
-		return token, nil
-	case RntbdTokenTypeInvalid:
-		return nil, fmt.Errorf("invalid token type")
-	}
-
-	return nil, fmt.Errorf("invalid token type")
-}
-
-func readBytes(reader *bufio.Reader, n int) []byte {
-	bytes := make([]byte, n)
-	_, err := io.ReadFull(reader, bytes)
-	if err != nil {
-		logger.ErrorLn("Error reading bytes:", err)
-		os.Exit(0)
-	}
-
-	return bytes
-}
@@ -1,120 +0,0 @@
-package rntbd
-
-import (
-	"bufio"
-	"crypto/tls"
-	"fmt"
-	"net"
-	"net/http/httptest"
-
-	"github.com/pikami/cosmium/api"
-	"github.com/pikami/cosmium/internal/logger"
-	tlsprovider "github.com/pikami/cosmium/internal/tls_provider"
-)
-
-type RntbdServer struct {
-	port      int
-	listener  net.Listener
-	apiServer *api.ApiServer
-}
-
-func NewRntbdServer(port int, apiServer *api.ApiServer) *RntbdServer {
-	return &RntbdServer{port: port, apiServer: apiServer}
-}
-
-func (s *RntbdServer) Start() error {
-	tlsConfig := tlsprovider.GetDefaultTlsConfig()
-	listener, err := tls.Listen("tcp", fmt.Sprintf(":%d", s.port), tlsConfig)
-	if err != nil {
-		return fmt.Errorf("failed to listen on port %d: %w", s.port, err)
-	}
-	s.listener = listener
-
-	go func() {
-		for {
-			conn, err := s.listener.Accept()
-			if err != nil {
-				logger.ErrorLn("Failed to accept connection:", err)
-				continue
-			}
-
-			go s.handleConnection(conn)
-		}
-	}()
-
-	return nil
-}
-
-func (s *RntbdServer) Stop() error {
-	return s.listener.Close()
-}
-
-func (s *RntbdServer) handleConnection(conn net.Conn) {
-	defer conn.Close()
-
-	reader := bufio.NewReader(conn)
-	writer := bufio.NewWriter(conn)
-
-	for {
-		_, err := reader.Peek(4)
-		if err != nil {
-			return
-		}
-
-		frame, err := ReadFrame(reader)
-		if err != nil {
-			logger.ErrorLn("Failed to read frame:", err)
-			continue
-		}
-
-		if frame.ResourceType == RntbdResourceTypeConnection {
-			responseFrame := buildContextFrame(frame)
-			_, err := writer.Write(responseFrame)
-			writer.Flush()
-			if err != nil {
-				logger.ErrorLn("Failed to write response frame:", err)
-				continue
-			}
-			continue
-		} else if frame.ResourceType == RntbdResourceTypeDatabase ||
-			frame.ResourceType == RntbdResourceTypeCollection ||
-			frame.ResourceType == RntbdResourceTypeDocument {
-			responseFrameBytes := s.passToApiServer(frame)
-			_, err := writer.Write(responseFrameBytes)
-			writer.Flush()
-			if err != nil {
-				logger.ErrorLn("Failed to write response frame:", err)
-				continue
-			}
-			continue
-		} else {
-			logger.Errorf("Received Unhandled RNTBD request from: %s with resource type: %s\n", conn.RemoteAddr(), frame.ResourceType.String())
-		}
-	}
-}
-
-func (s *RntbdServer) passToApiServer(frame *RntbdFrame) []byte {
-	req := frame.ToHttpRequest()
-	responseWriter := httptest.NewRecorder()
-	s.apiServer.GetRouter().ServeHTTP(responseWriter, req)
-
-	responseFrameBuilder := ToRntbdResponseFrame(responseWriter)
-	responseFrameBuilder.SetActivityId(frame.ActivityId)
-	if transportRequestId, ok := frame.RequestHeaders[RntbdRequestHeaderTransportRequestID]; ok {
-		responseFrameBuilder.AddHeader(uint16(RntbdResponseHeaderTransportRequestID), RntbdTokenTypeULong, transportRequestId)
-	}
-
-	responseFrameBuilder.AddHeader(uint16(RntbdResponseHeaderItemLSN), RntbdTokenTypeLongLong, int64(420))
-	responseFrameBuilder.AddHeader(uint16(RntbdResponseHeaderLocalLSN), RntbdTokenTypeLongLong, int64(420))
-	responseFrameBuilder.AddHeader(uint16(RntbdResponseHeaderGlobalCommittedLSN), RntbdTokenTypeLongLong, int64(420))
-	responseFrameBuilder.AddHeader(uint16(RntbdResponseHeaderItemLocalLSN), RntbdTokenTypeLongLong, int64(420))
-	responseFrameBuilder.AddHeader(uint16(RntbdResponseHeaderLSN), RntbdTokenTypeLongLong, int64(420))
-	responseFrameBuilder.AddHeader(uint16(RntbdResponseHeaderQuorumAckedLSN), RntbdTokenTypeLongLong, int64(420))
-	responseFrameBuilder.AddHeader(uint16(RntbdResponseHeaderQuorumAckedLocalLSN), RntbdTokenTypeLongLong, int64(420))
-	responseFrameBuilder.AddHeader(uint16(RntbdResponseHeaderCurrentReplicaSetSize), RntbdTokenTypeULong, uint32(1))
-
-	responseFrame := responseFrameBuilder.Build()
-	responseFrameBytes := responseFrame.ToBytes()
-
-	return responseFrameBytes
-}
@@ -107,7 +107,6 @@ const (
 	FunctionCallContains     FunctionCallType = "Contains"
 	FunctionCallEndsWith     FunctionCallType = "EndsWith"
 	FunctionCallStartsWith   FunctionCallType = "StartsWith"
-	FunctionCallRegexMatch   FunctionCallType = "RegexMatch"
 	FunctionCallIndexOf      FunctionCallType = "IndexOf"
 	FunctionCallToString     FunctionCallType = "ToString"
 	FunctionCallUpper        FunctionCallType = "Upper"
@@ -143,8 +142,6 @@ const (
 	FunctionCallSetIntersect     FunctionCallType = "SetIntersect"
 	FunctionCallSetUnion         FunctionCallType = "SetUnion"

-	FunctionCallIif FunctionCallType = "Iif"
-
 	FunctionCallMathAbs              FunctionCallType = "MathAbs"
 	FunctionCallMathAcos             FunctionCallType = "MathAcos"
 	FunctionCallMathAsin             FunctionCallType = "MathAsin"
@@ -188,7 +185,9 @@ const (
 	FunctionCallAggregateMin   FunctionCallType = "AggregateMin"
 	FunctionCallAggregateSum   FunctionCallType = "AggregateSum"

-	FunctionCallIn FunctionCallType = "In"
+	FunctionCallIif FunctionCallType = "Iif"
+	FunctionCallIn  FunctionCallType = "In"
+	FunctionCallUDF FunctionCallType = "UDF"
 )

 var AggregateFunctions = []FunctionCallType{
@@ -201,5 +200,6 @@ var AggregateFunctions = []FunctionCallType{

 type FunctionCall struct {
 	Arguments []interface{}
+	UdfName   string
 	Type      FunctionCallType
 }
@@ -112,65 +112,6 @@ func Test_Parse(t *testing.T) {
 		)
 	})

-	t.Run("Should parse NOT IN function", func(t *testing.T) {
-		testQueryParse(
-			t,
-			`SELECT c.id FROM c WHERE c.id NOT IN ("123", "456")`,
-			parsers.SelectStmt{
-				SelectItems: []parsers.SelectItem{
-					{
-						Path: []string{"c", "id"},
-						Type: parsers.SelectItemTypeField,
-					},
-				},
-				Table: parsers.Table{SelectItem: testutils.SelectItem_Path("c")},
-				Filters: parsers.SelectItem{
-					Type:   parsers.SelectItemTypeFunctionCall,
-					Invert: true,
-					Value: parsers.FunctionCall{
-						Type: parsers.FunctionCallIn,
-						Arguments: []interface{}{
-							parsers.SelectItem{
-								Path: []string{"c", "id"},
-								Type: parsers.SelectItemTypeField,
-							},
-							testutils.SelectItem_Constant_String("123"),
-							testutils.SelectItem_Constant_String("456"),
-						},
-					},
-				},
-			},
-		)
-	})
-
-	t.Run("Should parse NOT with parentheses", func(t *testing.T) {
-		testQueryParse(
-			t,
-			`SELECT c.id FROM c WHERE NOT (c.id IN ("123", "456"))`,
-			parsers.SelectStmt{
-				SelectItems: []parsers.SelectItem{
-					{Path: []string{"c", "id"}, Type: parsers.SelectItemTypeField},
-				},
-				Table: parsers.Table{SelectItem: testutils.SelectItem_Path("c")},
-				Filters: parsers.SelectItem{
-					Type:   parsers.SelectItemTypeFunctionCall,
-					Invert: true,
-					Value: parsers.FunctionCall{
-						Type: parsers.FunctionCallIn,
-						Arguments: []interface{}{
-							parsers.SelectItem{
-								Path: []string{"c", "id"},
-								Type: parsers.SelectItemTypeField,
-							},
-							testutils.SelectItem_Constant_String("123"),
-							testutils.SelectItem_Constant_String("456"),
-						},
-					},
-				},
-			},
-		)
-	})
-
 	t.Run("Should parse IN function with function call", func(t *testing.T) {
 		testQueryParse(
 			t,
@@ -245,4 +186,55 @@ func Test_Parse(t *testing.T) {
 			},
 		)
 	})
+
+	t.Run("Should parse SELECT with UDF function", func(t *testing.T) {
+		testQueryParse(
+			t,
+			`SELECT t.name, udf.CalculateTax(t.income, t.category) FROM t`,
+			parsers.SelectStmt{
+				SelectItems: []parsers.SelectItem{
+					testutils.SelectItem_Path("t", "name"),
+					{
+						Type: parsers.SelectItemTypeFunctionCall,
+						Value: parsers.FunctionCall{
+							Type:    parsers.FunctionCallUDF,
+							UdfName: "CalculateTax",
+							Arguments: []interface{}{
+								testutils.SelectItem_Path("t", "income"),
+								testutils.SelectItem_Path("t", "category"),
+							},
+						},
+					},
+				},
+				Table: parsers.Table{SelectItem: testutils.SelectItem_Path("t")},
+			},
+		)
+	})
+
+	t.Run("Should parse WHERE with UDF function", func(t *testing.T) {
+		testQueryParse(
+			t,
+			`SELECT c.id FROM c WHERE udf.IsEligible(c.status) = true`,
+			parsers.SelectStmt{
+				SelectItems: []parsers.SelectItem{
+					testutils.SelectItem_Path("c", "id"),
+				},
+				Table: parsers.Table{SelectItem: testutils.SelectItem_Path("c")},
+				Filters: parsers.ComparisonExpression{
+					Left: parsers.SelectItem{
+						Type: parsers.SelectItemTypeFunctionCall,
+						Value: parsers.FunctionCall{
+							Type:    parsers.FunctionCallUDF,
+							UdfName: "IsEligible",
+							Arguments: []interface{}{
+								testutils.SelectItem_Path("c", "status"),
+							},
+						},
+					},
+					Operation: "=",
+					Right:     testutils.SelectItem_Constant_Bool(true),
+				},
+			},
+		)
+	})
 }
@@ -154,6 +154,14 @@ func createFunctionCall(functionType parsers.FunctionCallType, arguments []inter
    return parsers.FunctionCall{Type: functionType, Arguments: arguments}, nil
 }

+func createUDFCall(functionName interface{}, arguments []interface{}) (parsers.FunctionCall, error) {
+    return parsers.FunctionCall{
+        Type: parsers.FunctionCallUDF, 
+        UdfName: functionName.(string),
+        Arguments: arguments,
+    }, nil
+}
+
 func joinStrings(array []interface{}) string {
    var stringsArray []string
    for _, elem := range array {
@@ -425,15 +433,7 @@ MulDivExpression <- left:SelectItemWithParentheses operations:(ws op:MultiplyOrD
    return makeMathExpression(left, operations)
 }

-SelectItemWithParentheses <- inv:(Not ws)? "(" ws ex:OrExpression ws ")" {
-    if inv != nil {
-        if ex1, ok := ex.(parsers.SelectItem); ok {
-            ex1.Invert = true
-            return ex1, nil
-        }
-    }
-    return ex, nil
-}
+SelectItemWithParentheses <- "(" ws ex:OrExpression ws ")" { return ex, nil }
    / inv:(Not ws)? ex:SelectItem {
    if inv != nil {
        ex1 := ex.(parsers.SelectItem)
@@ -519,7 +519,8 @@ BooleanLiteral <- ("true"i / "false"i) {
    return parsers.Constant{Type: parsers.ConstantTypeBoolean, Value: boolValue}, nil
 }

-FunctionCall <- StringFunctions
+FunctionCall <- UDFFunction
+    / StringFunctions
    / TypeCheckingFunctions
    / ArrayFunctions
    / ConditionalFunctions
@@ -527,6 +528,20 @@ FunctionCall <- StringFunctions
    / AggregateFunctions
    / MathFunctions

+UDFFunction <- "udf"i ws "." ws functionName:Identifier ws "(" ws arguments:UDFArgumentList? ws ")" {
+    if arguments == nil {
+        return createUDFCall(functionName, []interface{}{})
+    }
+    return createUDFCall(functionName, arguments.([]interface{}))
+}
+
+UDFArgumentList <- arg1:SelectItem others:(ws "," ws arg:SelectItem { return arg, nil })* {
+    if others == nil {
+        return []interface{}{arg1}, nil
+    }
+    return append([]interface{}{arg1}, others.([]interface{})...), nil
+}
+
 StringFunctions <- StringEqualsExpression
    / ToStringExpression
    / ConcatExpression
@@ -681,8 +696,6 @@ ThreeArgumentStringFunctionExpression <- function:ThreeArgumentStringFunction ws
        functionType = parsers.FunctionCallEndsWith
    case "STARTSWITH":
        functionType = parsers.FunctionCallStartsWith
-    case "REGEXMATCH":
-        functionType = parsers.FunctionCallRegexMatch
    case "INDEX_OF":
        functionType = parsers.FunctionCallIndexOf
    }
@@ -690,7 +703,7 @@ ThreeArgumentStringFunctionExpression <- function:ThreeArgumentStringFunction ws
    return createFunctionCall(functionType, []interface{}{ex1, ex2, ignoreCase})
 }

-ThreeArgumentStringFunction <- ("CONTAINS"i / "ENDSWITH"i / "STARTSWITH"i / "REGEXMATCH"i / "INDEX_OF"i) {
+ThreeArgumentStringFunction <- ("CONTAINS"i / "ENDSWITH"i / "STARTSWITH"i / "INDEX_OF"i) {
    return string(c.text), nil
 }

@@ -813,33 +826,10 @@ MathNumberBinExpression <- "NumberBin"i ws "(" ws ex1:SelectItem others:(ws ","
 MathPiExpression <- "PI"i ws "(" ws ")" { return createFunctionCall(parsers.FunctionCallMathPi, []interface{}{}) }
 MathRandExpression <- "RAND"i ws "(" ws ")" { return createFunctionCall(parsers.FunctionCallMathRand, []interface{}{}) }

-InFunction <- ex1:SelectProperty ws notIn:("NOT"i ws)? In ws "(" ws ex2:SelectItem others:(ws "," ws ex:SelectItem { return ex, nil })* ws ")" {
-    arguments := append([]interface{}{ex1, ex2}, others.([]interface{})...)
-    functionCall, _ := createFunctionCall(parsers.FunctionCallIn, arguments)
-    
-    if notIn != nil {
-        return parsers.SelectItem{
-            Type: parsers.SelectItemTypeFunctionCall,
-            Value: functionCall,
-            Invert: true,
-        }, nil
-    }
-    
-    return functionCall, nil
-}
-/ "(" ws ex1:SelectItem ws notIn:("NOT"i ws)? In ws "(" ws ex2:SelectItem others:(ws "," ws ex:SelectItem { return ex, nil })* ws ")" ws ")" {
-    arguments := append([]interface{}{ex1, ex2}, others.([]interface{})...)
-    functionCall, _ := createFunctionCall(parsers.FunctionCallIn, arguments)
-
-    if notIn != nil {
-        return parsers.SelectItem{
-            Type: parsers.SelectItemTypeFunctionCall,
-            Value: functionCall,
-            Invert: true,
-        }, nil
-    }
-    
-    return functionCall, nil
+InFunction <- ex1:SelectProperty ws In ws "(" ws ex2:SelectItem others:(ws "," ws ex:SelectItem { return ex, nil })* ws ")" {
+    return createFunctionCall(parsers.FunctionCallIn, append([]interface{}{ex1, ex2}, others.([]interface{})...))
+} / "(" ws ex1:SelectItem ws In ws "(" ws ex2:SelectItem others:(ws "," ws ex:SelectItem { return ex, nil })* ws ")" ws ")" {
+    return createFunctionCall(parsers.FunctionCallIn, append([]interface{}{ex1, ex2}, others.([]interface{})...))
 }

 AvgAggregateExpression   <- "AVG"i "(" ws ex:SelectItem  ws ")" {
@@ -168,32 +168,6 @@ func Test_Execute_StringFunctions(t *testing.T) {
 		)
 	})

-	t.Run("Should parse function REGEXMATCH()", func(t *testing.T) {
-		testQueryParse(
-			t,
-			`SELECT REGEXMATCH(c.id, "aB c", "ix") FROM c`,
-			parsers.SelectStmt{
-				SelectItems: []parsers.SelectItem{
-					{
-						Type: parsers.SelectItemTypeFunctionCall,
-						Value: parsers.FunctionCall{
-							Type: parsers.FunctionCallRegexMatch,
-							Arguments: []interface{}{
-								parsers.SelectItem{
-									Path: []string{"c", "id"},
-									Type: parsers.SelectItemTypeField,
-								},
-								testutils.SelectItem_Constant_String("aB c"),
-								testutils.SelectItem_Constant_String("ix"),
-							},
-						},
-					},
-				},
-				Table: parsers.Table{SelectItem: testutils.SelectItem_Path("c")},
-			},
-		)
-	})
-
 	t.Run("Should parse function INDEX_OF()", func(t *testing.T) {
 		testQueryParse(
 			t,
@@ -25,7 +25,7 @@ func (r rowContext) array_Contains(arguments []interface{}) bool {
 	exprToSearch := r.resolveSelectItem(arguments[1].(parsers.SelectItem))

 	partialSearch := false
-	if len(arguments) > 2 && arguments[2] != nil {
+	if len(arguments) > 2 {
 		boolExpr := r.resolveSelectItem(arguments[2].(parsers.SelectItem))
 		if boolValue, ok := boolExpr.(bool); ok {
 			partialSearch = boolValue
@@ -162,8 +162,6 @@ func (r rowContext) selectItem_SelectItemTypeFunctionCall(functionCall parsers.F
 		return r.strings_EndsWith(functionCall.Arguments)
 	case parsers.FunctionCallStartsWith:
 		return r.strings_StartsWith(functionCall.Arguments)
-	case parsers.FunctionCallRegexMatch:
-		return r.strings_RegexMatch(functionCall.Arguments)
 	case parsers.FunctionCallConcat:
 		return r.strings_Concat(functionCall.Arguments)
 	case parsers.FunctionCallIndexOf:
@@ -233,9 +231,6 @@ func (r rowContext) selectItem_SelectItemTypeFunctionCall(functionCall parsers.F
 	case parsers.FunctionCallSetUnion:
 		return r.set_Union(functionCall.Arguments)

-	case parsers.FunctionCallIif:
-		return r.misc_Iif(functionCall.Arguments)
-
 	case parsers.FunctionCallMathAbs:
 		return r.math_Abs(functionCall.Arguments)
 	case parsers.FunctionCallMathAcos:
@@ -322,6 +317,10 @@ func (r rowContext) selectItem_SelectItemTypeFunctionCall(functionCall parsers.F

 	case parsers.FunctionCallIn:
 		return r.misc_In(functionCall.Arguments)
+	case parsers.FunctionCallIif:
+		return r.misc_Iif(functionCall.Arguments)
+	case parsers.FunctionCallUDF:
+		return r.misc_UDF(functionCall.Arguments)
 	}

 	logger.Errorf("Unknown function call type: %v", functionCall.Type)
@@ -5,46 +5,18 @@ import (
 	"github.com/pikami/cosmium/parsers"
 )

-type ExecuteQueryResult struct {
-	Rows         []RowType
-	HasMorePages bool
-}
-
-func ExecuteQuery(
-	query parsers.SelectStmt,
-	documents rowTypeIterator,
-	offset int,
-	limit int,
-) ExecuteQueryResult {
+func ExecuteQuery(query parsers.SelectStmt, documents rowTypeIterator) []RowType {
 	resultIter := executeQuery(query, &rowTypeToRowContextIterator{documents: documents, query: query})
-
-	result := &ExecuteQueryResult{
-		Rows:         make([]RowType, 0),
-		HasMorePages: false,
-	}
-
-	for i := 0; i < offset; i++ {
-		_, status := resultIter.Next()
-		if status != datastore.StatusOk {
-			break
-		}
-	}
-
-	for i := 0; i < limit; i++ {
+	result := make([]RowType, 0)
+	for {
 		row, status := resultIter.Next()
 		if status != datastore.StatusOk {
 			break
 		}

-		result.Rows = append(result.Rows, row)
+		result = append(result, row)
 	}
-
-	_, status := resultIter.Next()
-	if status == datastore.StatusOk {
-		result.HasMorePages = true
-	}
-
-	return *result
+	return result
 }

 func executeQuery(query parsers.SelectStmt, documents rowIterator) rowTypeIterator {
@@ -29,3 +29,7 @@ func (r rowContext) misc_Iif(arguments []interface{}) interface{} {

 	return r.resolveSelectItem(arguments[2].(parsers.SelectItem))
 }
+
+func (r rowContext) misc_UDF(arguments []interface{}) interface{} {
+	return "TODO"
+}
@@ -38,10 +38,10 @@ func testQueryExecute(
 	expectedData []memoryexecutor.RowType,
 ) {
 	iter := NewTestDocumentIterator(data)
-	result := memoryexecutor.ExecuteQuery(query, iter, 0, 1000)
+	result := memoryexecutor.ExecuteQuery(query, iter)

-	if !reflect.DeepEqual(result.Rows, expectedData) {
-		t.Errorf("execution result does not match expected data.\nExpected: %+v\nGot: %+v", expectedData, result.Rows)
+	if !reflect.DeepEqual(result, expectedData) {
+		t.Errorf("execution result does not match expected data.\nExpected: %+v\nGot: %+v", expectedData, result)
 	}
 }

@@ -149,41 +149,6 @@ func Test_Execute(t *testing.T) {
 		)
 	})

-	t.Run("Should execute NOT IN function", func(t *testing.T) {
-		testQueryExecute(
-			t,
-			parsers.SelectStmt{
-				SelectItems: []parsers.SelectItem{
-					{
-						Path: []string{"c", "id"},
-						Type: parsers.SelectItemTypeField,
-					},
-				},
-				Table: parsers.Table{SelectItem: testutils.SelectItem_Path("c")},
-				Filters: parsers.SelectItem{
-					Type:   parsers.SelectItemTypeFunctionCall,
-					Invert: true,
-					Value: parsers.FunctionCall{
-						Type: parsers.FunctionCallIn,
-						Arguments: []interface{}{
-							parsers.SelectItem{
-								Path: []string{"c", "id"},
-								Type: parsers.SelectItemTypeField,
-							},
-							testutils.SelectItem_Constant_String("123"),
-							testutils.SelectItem_Constant_String("456"),
-						},
-					},
-				},
-			},
-			mockData,
-			[]memoryexecutor.RowType{
-				map[string]interface{}{"id": "12345"},
-				map[string]interface{}{"id": "67890"},
-			},
-		)
-	})
-
 	t.Run("Should execute IN function with function call", func(t *testing.T) {
 		testQueryExecute(
 			t,
@@ -2,7 +2,6 @@ package memoryexecutor

 import (
 	"fmt"
-	"regexp"
 	"strings"

 	"github.com/pikami/cosmium/internal/logger"
@@ -76,46 +75,6 @@ func (r rowContext) strings_StartsWith(arguments []interface{}) bool {
 	return strings.HasPrefix(str1, str2)
 }

-func (r rowContext) strings_RegexMatch(arguments []interface{}) bool {
-	value, valueOk := r.parseString(arguments[0])
-	pattern, patternOk := r.parseString(arguments[1])
-	if !valueOk || !patternOk {
-		return false
-	}
-
-	modifiers, ok := r.getStringFlag(arguments)
-	if !ok {
-		return false
-	}
-
-	regexPattern := pattern
-	if strings.Contains(modifiers, "x") {
-		regexPattern = stripRegexIgnoredWhitespace(regexPattern)
-	}
-
-	var flags strings.Builder
-	if strings.Contains(modifiers, "i") {
-		flags.WriteByte('i')
-	}
-	if strings.Contains(modifiers, "m") {
-		flags.WriteByte('m')
-	}
-	if strings.Contains(modifiers, "s") {
-		flags.WriteByte('s')
-	}
-	if flags.Len() > 0 {
-		regexPattern = "(?" + flags.String() + ")" + regexPattern
-	}
-
-	matched, err := regexp.MatchString(regexPattern, value)
-	if err != nil {
-		logger.Errorf("strings_RegexMatch - invalid pattern %q: %v", pattern, err)
-		return false
-	}
-
-	return matched
-}
-
 func (r rowContext) strings_Concat(arguments []interface{}) string {
 	result := ""

@@ -359,20 +318,6 @@ func (r rowContext) getBoolFlag(arguments []interface{}) bool {
 	return ignoreCase
 }

-func (r rowContext) getStringFlag(arguments []interface{}) (string, bool) {
-	if len(arguments) <= 2 || arguments[2] == nil {
-		return "", true
-	}
-
-	flagItem := arguments[2].(parsers.SelectItem)
-	if value, ok := r.resolveSelectItem(flagItem).(string); ok {
-		return value, true
-	}
-
-	logger.ErrorLn("getStringFlag - got parameters of wrong type")
-	return "", false
-}
-
 func (r rowContext) parseString(argument interface{}) (value string, ok bool) {
 	exItem := argument.(parsers.SelectItem)
 	ex := r.resolveSelectItem(exItem)
@@ -384,41 +329,6 @@ func (r rowContext) parseString(argument interface{}) (value string, ok bool) {
 	return "", false
 }

-func stripRegexIgnoredWhitespace(pattern string) string {
-	var result strings.Builder
-	inCharClass := false
-	escaped := false
-
-	for _, r := range pattern {
-		if escaped {
-			result.WriteRune(r)
-			escaped = false
-			continue
-		}
-
-		if r == '\\' {
-			result.WriteRune(r)
-			escaped = true
-			continue
-		}
-
-		switch r {
-		case '[':
-			inCharClass = true
-		case ']':
-			inCharClass = false
-		}
-
-		if !inCharClass && (r == ' ' || r == '\t' || r == '\n' || r == '\r' || r == '\f') {
-			continue
-		}
-
-		result.WriteRune(r)
-	}
-
-	return result.String()
-}
-
 func convertToString(value interface{}) string {
 	switch v := value.(type) {
 	case string:
@@ -231,42 +231,6 @@ func Test_Execute_StringFunctions(t *testing.T) {
 		)
 	})

-	t.Run("Should execute function REGEXMATCH()", func(t *testing.T) {
-		testQueryExecute(
-			t,
-			parsers.SelectStmt{
-				SelectItems: []parsers.SelectItem{
-					{
-						Path: []string{"c", "id"},
-						Type: parsers.SelectItemTypeField,
-					},
-					{
-						Alias: "regexMatch",
-						Type:  parsers.SelectItemTypeFunctionCall,
-						Value: parsers.FunctionCall{
-							Type: parsers.FunctionCallRegexMatch,
-							Arguments: []interface{}{
-								parsers.SelectItem{
-									Path: []string{"c", "str"},
-									Type: parsers.SelectItemTypeField,
-								},
-								testutils.SelectItem_Constant_String("COOL WORLD"),
-								testutils.SelectItem_Constant_String("i"),
-							},
-						},
-					},
-				},
-				Table: parsers.Table{SelectItem: testutils.SelectItem_Path("c")},
-			},
-			mockData,
-			[]memoryexecutor.RowType{
-				map[string]interface{}{"id": "123", "regexMatch": false},
-				map[string]interface{}{"id": "456", "regexMatch": false},
-				map[string]interface{}{"id": "789", "regexMatch": true},
-			},
-		)
-	})
-
 	t.Run("Should execute function INDEX_OF()", func(t *testing.T) {
 		testQueryExecute(
 			t,
@@ -38,7 +38,6 @@ func CreateServerInstance(serverName *C.char, configurationJSON *C.char) int {
 	switch configuration.DataStore {
 	case config.DataStoreBadger:
 		dataStore = badgerdatastore.NewBadgerDataStore(badgerdatastore.BadgerDataStoreOptions{
-			InitialDataFilePath: configuration.InitialDataFilePath,
 			PersistDataFilePath: configuration.PersistDataFilePath,
 		})
 	default:
@@ -7,9 +7,6 @@ int test_Databases();

 int main(int argc, char *argv[])
 {
-    /* Disable stdout buffering for CI environments without a real terminal */
-    setvbuf(stdout, NULL, _IONBF, 0);
-
    if (argc < 2)
    {
        fprintf(stderr, "Usage: %s <path_to_shared_library>\n", argv[0]);
@@ -17,20 +14,13 @@ int main(int argc, char *argv[])
    }

    const char *libPath = argv[1];
-    handle = load_library(libPath);
+    handle = dlopen(libPath, RTLD_LAZY);
    if (!handle)
    {
-        fprintf(stderr, "Failed to load shared library: %s\n", get_load_error());
+        fprintf(stderr, "Failed to load shared library: %s\n", dlerror());
        return EXIT_FAILURE;
    }

-    /* give the loaded library a short time to initialize */
-#ifdef _WIN32
-    Sleep(1000);
-#else
-    sleep(1);
-#endif
-
    printf("Running tests for library: %s\n", libPath);
    int results[] = {
        test_CreateServerInstance(),
@@ -51,15 +41,6 @@ int main(int argc, char *argv[])

    printf("Tests passed: %d/%d\n", numPassed, numTests);

-    /* Exit explicitly before unloading the library.
-       Go runtime cleanup during FreeLibrary can set a non-zero exit code on Windows. */
-    int exitCode = (numPassed == numTests) ? EXIT_SUCCESS : EXIT_FAILURE;
-
-#ifdef _WIN32
-    ExitProcess(exitCode);
-#else
-    close_library(handle);
-#endif
-
-    return exitCode;
+    dlclose(handle);
+    return EXIT_SUCCESS;
 }
@@ -1,50 +1,13 @@
 #include "shared.h"

-lib_handle_t handle = NULL;
-
-char *get_load_error(void)
-{
-#ifdef _WIN32
-    DWORD error = GetLastError();
-    static char buf[256];
-    FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
-                   NULL, error, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
-                   buf, sizeof(buf), NULL);
-    return buf;
-#else
-    return dlerror();
-#endif
-}
-
-lib_handle_t load_library(const char *path)
-{
-#ifdef _WIN32
-    return LoadLibraryA(path);
-#else
-    return dlopen(path, RTLD_LAZY);
-#endif
-}
-
-void close_library(lib_handle_t handle)
-{
-#ifdef _WIN32
-    FreeLibrary(handle);
-#else
-    dlclose(handle);
-#endif
-}
+void *handle = NULL;

 void *load_function(const char *func_name)
 {
-#ifdef _WIN32
-    void *func = (void *)GetProcAddress(handle, func_name);
-#else
    void *func = dlsym(handle, func_name);
-#endif
-    
    if (!func)
    {
-        fprintf(stderr, "Failed to load function %s: %s\n", func_name, get_load_error());
+        fprintf(stderr, "Failed to load function %s: %s\n", func_name, dlerror());
    }
    return func;
 }
@@ -3,24 +3,13 @@

 #include <stdio.h>
 #include <stdlib.h>
+#include <dlfcn.h>
 #include <string.h>
 #include <ctype.h>

-#ifdef _WIN32
-    #include <unistd.h>
-    #include <windows.h>
-    typedef HMODULE lib_handle_t;
-#else
-    #include <dlfcn.h>
-    typedef void* lib_handle_t;
-#endif
-
-extern lib_handle_t handle;
+extern void *handle;

 void *load_function(const char *func_name);
 char *compact_json(const char *json);
-char *get_load_error(void);
-lib_handle_t load_library(const char *path);
-void close_library(lib_handle_t handle);

 #endif
@@ -2,16 +2,6 @@

 int test_Databases()
 {
-    /* Load FreeMemory function - must use this to free memory allocated by the DLL
-       because the DLL may use a different C runtime heap than the test loader */
-    typedef void (*FreeMemoryFn)(char *);
-    FreeMemoryFn FreeMemory = (FreeMemoryFn)load_function("FreeMemory");
-    if (!FreeMemory)
-    {
-        fprintf(stderr, "Failed to find FreeMemory function\n");
-        return 0;
-    }
-
    typedef int (*CreateDatabaseFn)(char *, char *);
    CreateDatabaseFn CreateDatabase = (CreateDatabaseFn)load_function("CreateDatabase");
    if (!CreateDatabase)
@@ -46,7 +36,6 @@ int test_Databases()
    if (database)
    {
        printf("GetDatabase: SUCCESS (database = %s)\n", database);
-        FreeMemory(database);
    }
    else
    {
@@ -2,16 +2,6 @@

 int test_ServerInstanceStateMethods()
 {
-    /* Load FreeMemory function - must use this to free memory allocated by the DLL
-       because the DLL may use a different C runtime heap than the test loader */
-    typedef void (*FreeMemoryFn)(char *);
-    FreeMemoryFn FreeMemory = (FreeMemoryFn)load_function("FreeMemory");
-    if (!FreeMemory)
-    {
-        fprintf(stderr, "Failed to find FreeMemory function\n");
-        return 0;
-    }
-
    typedef int (*LoadServerInstanceStateFn)(char *, char *);
    LoadServerInstanceStateFn LoadServerInstanceState = (LoadServerInstanceStateFn)load_function("LoadServerInstanceState");
    if (!LoadServerInstanceState)
@@ -56,7 +46,7 @@ int test_ServerInstanceStateMethods()
    char *compact_state = compact_json(state);
    if (!compact_state)
    {
-        FreeMemory(state);
+        free(state);
        return 0;
    }

@@ -69,12 +59,10 @@ int test_ServerInstanceStateMethods()
        printf("GetServerInstanceState: State does not match expected value.\n");
        printf("Expected: %s\n", expected_state);
        printf("Actual:   %s\n", compact_state);
-        FreeMemory(state);
-        free(compact_state);
        return 0;
    }

-    FreeMemory(state);
+    free(state);
    free(compact_state);
    return 1;
 }