mirror of
https://github.com/pikami/cosmium.git
synced 2026-06-12 15:27:06 +01:00
Compare commits
7 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 90de21c7b0 | |||
| 7c9c8ec9dc | |||
| be761badae | |||
| 36fd7f48cc | |||
| 05e8cd2842 | |||
| c3726a6633 | |||
| d76cc88175 |
@@ -12,13 +12,13 @@ jobs:
|
||||
|
||||
steps:
|
||||
- name: Checkout code
|
||||
uses: actions/checkout@v3
|
||||
uses: actions/checkout@v6
|
||||
|
||||
- name: Cross-Compile with xgo
|
||||
uses: crazy-max/ghaction-xgo@e22d3c8b089adba750d5a74738b8e95d96f0c991 # v3.1.0
|
||||
uses: crazy-max/ghaction-xgo@de82f877ff4552f03b66c146f608233849e9c3dc # v4.0.0
|
||||
with:
|
||||
xgo_version: latest
|
||||
go_version: 1.24.7
|
||||
go_version: 1.26.3
|
||||
dest: dist
|
||||
pkg: sharedlibrary
|
||||
prefix: cosmium
|
||||
@@ -28,7 +28,7 @@ jobs:
|
||||
buildvcs: true
|
||||
|
||||
- name: Upload artifact
|
||||
uses: actions/upload-artifact@v4
|
||||
uses: actions/upload-artifact@v7
|
||||
with:
|
||||
name: shared-libraries
|
||||
path: dist/*
|
||||
@@ -48,10 +48,10 @@ jobs:
|
||||
lib_ext: arm64.dylib
|
||||
|
||||
steps:
|
||||
- uses: actions/checkout@v3
|
||||
- uses: actions/checkout@v6
|
||||
|
||||
- name: Download shared libraries
|
||||
uses: actions/download-artifact@v4
|
||||
uses: actions/download-artifact@v8
|
||||
with:
|
||||
name: shared-libraries
|
||||
path: libs
|
||||
|
||||
@@ -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@v4
|
||||
uses: actions/checkout@v6
|
||||
with:
|
||||
fetch-depth: 0
|
||||
|
||||
- name: Set up Go
|
||||
uses: actions/setup-go@v5
|
||||
with:
|
||||
go-version: 1.24.7
|
||||
go-version: 1.26.3
|
||||
|
||||
- name: Cross-Compile with xgo
|
||||
uses: crazy-max/ghaction-xgo@e22d3c8b089adba750d5a74738b8e95d96f0c991 # v3.1.0
|
||||
uses: crazy-max/ghaction-xgo@de82f877ff4552f03b66c146f608233849e9c3dc # v4.0.0
|
||||
with:
|
||||
xgo_version: latest
|
||||
go_version: 1.24.7
|
||||
go_version: 1.26.3
|
||||
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.24.7
|
||||
GOVERSION=1.26.3
|
||||
|
||||
DIST_DIR=dist
|
||||
|
||||
|
||||
@@ -23,6 +23,7 @@ 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")
|
||||
@@ -35,6 +36,7 @@ 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()
|
||||
@@ -42,6 +44,7 @@ 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
|
||||
@@ -52,6 +55,7 @@ func ParseFlags() ServerConfig {
|
||||
config.AccountKey = *accountKey
|
||||
config.LogLevel = logLevel.value
|
||||
config.DataStore = dataStore.value
|
||||
config.EnableRntbd = *enableRntbd
|
||||
|
||||
config.PopulateCalculatedFields()
|
||||
|
||||
@@ -62,6 +66,7 @@ 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 {
|
||||
|
||||
@@ -4,10 +4,12 @@ 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"`
|
||||
@@ -17,6 +19,7 @@ type ServerConfig struct {
|
||||
DisableTls bool `json:"disableTls"`
|
||||
LogLevel string `json:"logLevel"`
|
||||
ExplorerBaseUrlLocation string `json:"explorerBaseUrlLocation"`
|
||||
EnableRntbd bool `json:"enableRntbd"`
|
||||
|
||||
DataStore string `json:"dataStore"`
|
||||
}
|
||||
|
||||
@@ -47,6 +47,9 @@ 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
|
||||
}
|
||||
@@ -90,7 +93,25 @@ func (h *Handlers) ReplaceDocument(c *gin.Context) {
|
||||
return
|
||||
}
|
||||
|
||||
status := h.dataStore.DeleteDocument(databaseId, collectionId, documentId)
|
||||
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)
|
||||
if status == datastore.StatusNotFound {
|
||||
c.IndentedJSON(http.StatusNotFound, constants.NotFoundResponse)
|
||||
return
|
||||
|
||||
@@ -3,6 +3,7 @@ package handlers
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
@@ -41,3 +42,46 @@ 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),
|
||||
})
|
||||
}
|
||||
|
||||
@@ -4,8 +4,10 @@ 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"
|
||||
@@ -19,4 +21,10 @@ const (
|
||||
// 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,6 +31,7 @@ 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" {
|
||||
@@ -79,6 +80,7 @@ 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)
|
||||
|
||||
|
||||
@@ -0,0 +1,47 @@
|
||||
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},
|
||||
)
|
||||
})
|
||||
})
|
||||
}
|
||||
@@ -0,0 +1,73 @@
|
||||
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,6 +5,7 @@ import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"reflect"
|
||||
"sync"
|
||||
@@ -379,6 +380,80 @@ 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)
|
||||
|
||||
|
||||
@@ -0,0 +1,63 @@
|
||||
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)
|
||||
}
|
||||
}
|
||||
}
|
||||
+10
-1
@@ -11,6 +11,7 @@ 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() {
|
||||
@@ -38,6 +39,15 @@ 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)
|
||||
}
|
||||
|
||||
@@ -50,6 +60,5 @@ func waitForExit(server *api.ApiServer, dataStore datastore.DataStore) {
|
||||
|
||||
// Stop the server
|
||||
server.Stop()
|
||||
|
||||
dataStore.Close()
|
||||
}
|
||||
|
||||
@@ -1,64 +1,64 @@
|
||||
module github.com/pikami/cosmium
|
||||
|
||||
go 1.24.7
|
||||
go 1.26.3
|
||||
|
||||
require (
|
||||
github.com/Azure/azure-sdk-for-go/sdk/azcore v1.21.0
|
||||
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/cosmiumdev/json-patch/v5 v5.9.11
|
||||
github.com/dgraph-io/badger/v4 v4.9.0
|
||||
github.com/gin-gonic/gin v1.11.0
|
||||
github.com/dgraph-io/badger/v4 v4.9.2
|
||||
github.com/gin-gonic/gin v1.12.0
|
||||
github.com/google/uuid v1.6.0
|
||||
github.com/stretchr/testify v1.11.1
|
||||
github.com/vmihailenco/msgpack/v5 v5.4.1
|
||||
golang.org/x/exp v0.0.0-20260112195511-716be5621a96
|
||||
golang.org/x/exp v0.0.0-20260603202125-055de637280b
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/Azure/azure-sdk-for-go v68.0.0+incompatible // indirect
|
||||
github.com/Azure/azure-sdk-for-go/sdk/internal v1.11.2 // indirect
|
||||
github.com/bytedance/gopkg v0.1.3 // indirect
|
||||
github.com/bytedance/sonic v1.15.0 // indirect
|
||||
github.com/bytedance/sonic/loader v0.5.0 // 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/cespare/xxhash/v2 v2.3.0 // indirect
|
||||
github.com/cloudwego/base64x v0.1.6 // indirect
|
||||
github.com/cloudwego/base64x v0.1.7 // indirect
|
||||
github.com/davecgh/go-spew v1.1.1 // indirect
|
||||
github.com/dgraph-io/ristretto/v2 v2.4.0 // indirect
|
||||
github.com/dustin/go-humanize v1.0.1 // indirect
|
||||
github.com/gabriel-vasile/mimetype v1.4.12 // indirect
|
||||
github.com/gin-contrib/sse v1.1.0 // indirect
|
||||
github.com/gabriel-vasile/mimetype v1.4.13 // indirect
|
||||
github.com/gin-contrib/sse v1.1.1 // 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.1 // indirect
|
||||
github.com/goccy/go-json v0.10.5 // 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/json-iterator/go v1.1.12 // indirect
|
||||
github.com/klauspost/compress v1.18.3 // indirect
|
||||
github.com/klauspost/compress v1.18.6 // 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.20 // indirect
|
||||
github.com/mattn/go-isatty v0.0.22 // 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.2.4 // indirect
|
||||
github.com/pelletier/go-toml/v2 v2.3.1 // 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.59.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/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.39.0 // indirect
|
||||
go.opentelemetry.io/otel/metric v1.39.0 // indirect
|
||||
go.opentelemetry.io/otel/trace v1.39.0 // indirect
|
||||
go.uber.org/mock v0.6.0 // indirect
|
||||
golang.org/x/arch v0.23.0 // indirect
|
||||
golang.org/x/crypto v0.47.0 // indirect
|
||||
golang.org/x/net v0.49.0 // indirect
|
||||
golang.org/x/sys v0.40.0 // indirect
|
||||
golang.org/x/text v0.33.0 // 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
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
)
|
||||
|
||||
@@ -1,44 +1,44 @@
|
||||
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.21.0 h1:fou+2+WFTib47nS+nz/ozhEBnvU96bKHy6LjRsY4E28=
|
||||
github.com/Azure/azure-sdk-for-go/sdk/azcore v1.21.0/go.mod h1:t76Ruy8AHvUAC8GfMWJMa0ElSbuIcO03NLpynfbgsPA=
|
||||
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/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.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/gopkg v0.1.3 h1:TPBSwH8RsouGCBcMBktLt1AymVo2TVsBVCY4b6TnZ/M=
|
||||
github.com/bytedance/gopkg v0.1.3/go.mod h1:576VvJ+eJgyCzdjS+c4+77QF3p7ubbtiKARP3TxducM=
|
||||
github.com/bytedance/sonic v1.15.0 h1:/PXeWFaR5ElNcVE84U0dOHjiMHQOwNIx3K4ymzh/uSE=
|
||||
github.com/bytedance/sonic v1.15.0/go.mod h1:tFkWrPz0/CUCLEF4ri4UkHekCIcdnkqXw9VduqpJh0k=
|
||||
github.com/bytedance/sonic/loader v0.5.0 h1:gXH3KVnatgY7loH5/TkeVyXPfESoqSBSBEiDd5VjlgE=
|
||||
github.com/bytedance/sonic/loader v0.5.0/go.mod h1:AR4NYCk5DdzZizZ5djGqQ92eEhCCcdf5x77udYiSJRo=
|
||||
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/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.6 h1:t11wG9AECkCDk5fMSoxmufanudBtJ+/HemLstXDLI2M=
|
||||
github.com/cloudwego/base64x v0.1.6/go.mod h1:OFcloc187FXDaYHvrNIjxSe8ncn0OOM8gEHfghB2IPU=
|
||||
github.com/cloudwego/base64x v0.1.7 h1:NppS+Fgzg5ovhn4NkUXaDT3x9jldgH5ToMCqzBSi2zI=
|
||||
github.com/cloudwego/base64x v0.1.7/go.mod h1:Cu1PV9zfrSf7ET2tIbWbbEy7jO7HHJ13q4X2SQ8aWYg=
|
||||
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.0 h1:tpqWb0NewSrCYqTvywbcXOhQdWcqephkVkbBmaaqHzc=
|
||||
github.com/dgraph-io/badger/v4 v4.9.0/go.mod h1:5/MEx97uzdPUHR4KtkNt8asfI2T4JiEiQlV7kWUo8c0=
|
||||
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/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.12 h1:e9hWvmLYvtp846tLHam2o++qitpguFiYCKbn0w9jyqw=
|
||||
github.com/gabriel-vasile/mimetype v1.4.12/go.mod h1:d+9Oxyo1wTzWdyVUPMmXFvp4F9tea18J8ufA774AB3s=
|
||||
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.11.0 h1:OW/6PLjyusp2PPXtyxKHU0RbX6I/l28FTdDlae5ueWk=
|
||||
github.com/gin-gonic/gin v1.11.0/go.mod h1:+iq/FyxlGzII0KHiBGjuNn4UNENUlKbGlNmc+W50Dls=
|
||||
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/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,14 +50,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.1 h1:f3zDSN/zOma+w6+1Wswgd9fLkdwy06ntQJp0BBvFG0w=
|
||||
github.com/go-playground/validator/v10 v10.30.1/go.mod h1:oSuBIQzuJxL//3MelwSLD5hc2Tu889bF0Idm9Dg26cM=
|
||||
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/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.2.2 h1:Rl4B7itRWVtYIHFrSNd7vhTiz9UpLdi6gZhZ3wEeDy8=
|
||||
github.com/golang-jwt/jwt/v5 v5.2.2/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk=
|
||||
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/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
|
||||
@@ -67,8 +67,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.3 h1:9PJRvfbmTabkOX8moIpXPbMMbYN60bWImDDU7L+/6zw=
|
||||
github.com/klauspost/compress v1.18.3/go.mod h1:R0h/fSBs8DE4ENlcrlib3PsXS61voFxhIs2DeRhCvJ4=
|
||||
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/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,23 +79,25 @@ 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.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
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/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.2.4 h1:mye9XuhQ6gvn5h28+VilKrrPoQVanw5PMw/TB0t5Ec4=
|
||||
github.com/pelletier/go-toml/v2 v2.2.4/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY=
|
||||
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/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.59.0 h1:OLJkp1Mlm/aS7dpKgTc6cnpynnD2Xg7C1pwL6vy/SAw=
|
||||
github.com/quic-go/quic-go v0.59.0/go.mod h1:upnsH4Ju1YkqpLXC305eW3yDZ4NfnNbmQRCMWS58IKU=
|
||||
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/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
@@ -117,29 +119,30 @@ github.com/vmihailenco/msgpack/v5 v5.4.1 h1:cQriyiUvjTwOHg8QZaPihLWeRAAVoCpE00IU
|
||||
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.39.0 h1:8yPrr/S0ND9QEfTfdP9V+SiwT4E0G7Y5MO7p85nis48=
|
||||
go.opentelemetry.io/otel v1.39.0/go.mod h1:kLlFTywNWrFyEdH0oj2xK0bFYZtHRYUdv1NklR/tgc8=
|
||||
go.opentelemetry.io/otel/metric v1.39.0 h1:d1UzonvEZriVfpNKEVmHXbdf909uGTOQjA0HF0Ls5Q0=
|
||||
go.opentelemetry.io/otel/metric v1.39.0/go.mod h1:jrZSWL33sD7bBxg1xjrqyDjnuzTUB0x1nBERXd7Ftcs=
|
||||
go.opentelemetry.io/otel/trace v1.39.0 h1:2d2vfpEDmCJ5zVYz7ijaJdOF59xLomrvj7bjt6/qCJI=
|
||||
go.opentelemetry.io/otel/trace v1.39.0/go.mod h1:88w4/PnZSazkGzz/w84VHpQafiU4EtqqlVdxWy+rNOA=
|
||||
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.23.0 h1:lKF64A2jF6Zd8L0knGltUnegD62JMFBiCPBmQpToHhg=
|
||||
golang.org/x/arch v0.23.0/go.mod h1:dNHoOeKiyja7GTvF9NJS1l3Z2yntpQNzgrjh1cU103A=
|
||||
golang.org/x/crypto v0.47.0 h1:V6e3FRj+n4dbpw86FJ8Fv7XVOql7TEwpHapKoMJ/GO8=
|
||||
golang.org/x/crypto v0.47.0/go.mod h1:ff3Y9VzzKbwSSEzWqJsJVBnWmRwRSHt/6Op5n9bQc4A=
|
||||
golang.org/x/exp v0.0.0-20260112195511-716be5621a96 h1:Z/6YuSHTLOHfNFdb8zVZomZr7cqNgTJvA8+Qz75D8gU=
|
||||
golang.org/x/exp v0.0.0-20260112195511-716be5621a96/go.mod h1:nzimsREAkjBCIEFtHiYkrJyT+2uy9YZJB7H1k68CXZU=
|
||||
golang.org/x/net v0.49.0 h1:eeHFmOGUTtaaPSGNmjBKpbng9MulQsJURQUAfUwY++o=
|
||||
golang.org/x/net v0.49.0/go.mod h1:/ysNB2EvaqvesRkuLAyjI1ycPZlQHM3q01F02UY/MV8=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.40.0 h1:DBZZqJ2Rkml6QMQsZywtnjnnGvHza6BTfYFWY9kjEWQ=
|
||||
golang.org/x/sys v0.40.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||
golang.org/x/text v0.33.0 h1:B3njUFyqtHDUI5jMn1YIr5B0IE2U0qck04r6d4KPAxE=
|
||||
golang.org/x/text v0.33.0/go.mod h1:LuMebE6+rBincTi9+xWTY8TztLzKHc/9C1uBCG27+q8=
|
||||
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=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
|
||||
@@ -35,3 +35,7 @@ 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.",
|
||||
}
|
||||
|
||||
@@ -12,8 +12,10 @@ import (
|
||||
)
|
||||
|
||||
type BadgerDataStore struct {
|
||||
db *badger.DB
|
||||
gcTicker *time.Ticker
|
||||
db *badger.DB
|
||||
gcTicker *time.Ticker
|
||||
gcDone chan struct{}
|
||||
gcStopped chan struct{}
|
||||
}
|
||||
|
||||
type BadgerDataStoreOptions struct {
|
||||
@@ -36,8 +38,10 @@ func NewBadgerDataStore(options BadgerDataStoreOptions) *BadgerDataStore {
|
||||
gcTicker := time.NewTicker(5 * time.Minute)
|
||||
|
||||
ds := &BadgerDataStore{
|
||||
db: db,
|
||||
gcTicker: gcTicker,
|
||||
db: db,
|
||||
gcTicker: gcTicker,
|
||||
gcDone: make(chan struct{}),
|
||||
gcStopped: make(chan struct{}),
|
||||
}
|
||||
|
||||
ds.initializeDataStore(options.InitialDataFilePath)
|
||||
@@ -50,7 +54,8 @@ func NewBadgerDataStore(options BadgerDataStoreOptions) *BadgerDataStore {
|
||||
func (r *BadgerDataStore) Close() {
|
||||
if r.gcTicker != nil {
|
||||
r.gcTicker.Stop()
|
||||
r.gcTicker = nil
|
||||
close(r.gcDone)
|
||||
<-r.gcStopped
|
||||
}
|
||||
|
||||
r.db.Close()
|
||||
@@ -63,11 +68,19 @@ func (r *BadgerDataStore) DumpToJson() (string, error) {
|
||||
}
|
||||
|
||||
func (r *BadgerDataStore) runGarbageCollector() {
|
||||
for range r.gcTicker.C {
|
||||
again:
|
||||
err := r.db.RunValueLogGC(0.7)
|
||||
if err == nil {
|
||||
goto again
|
||||
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
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,138 @@
|
||||
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()
|
||||
}
|
||||
@@ -0,0 +1,746 @@
|
||||
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)
|
||||
}
|
||||
@@ -0,0 +1,128 @@
|
||||
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
|
||||
}
|
||||
@@ -0,0 +1,212 @@
|
||||
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
|
||||
}
|
||||
@@ -0,0 +1,120 @@
|
||||
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,6 +107,7 @@ const (
|
||||
FunctionCallContains FunctionCallType = "Contains"
|
||||
FunctionCallEndsWith FunctionCallType = "EndsWith"
|
||||
FunctionCallStartsWith FunctionCallType = "StartsWith"
|
||||
FunctionCallRegexMatch FunctionCallType = "RegexMatch"
|
||||
FunctionCallIndexOf FunctionCallType = "IndexOf"
|
||||
FunctionCallToString FunctionCallType = "ToString"
|
||||
FunctionCallUpper FunctionCallType = "Upper"
|
||||
|
||||
@@ -143,6 +143,34 @@ func Test_Parse(t *testing.T) {
|
||||
)
|
||||
})
|
||||
|
||||
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,
|
||||
|
||||
+1448
-1414
File diff suppressed because it is too large
Load Diff
+12
-2
@@ -425,7 +425,15 @@ MulDivExpression <- left:SelectItemWithParentheses operations:(ws op:MultiplyOrD
|
||||
return makeMathExpression(left, operations)
|
||||
}
|
||||
|
||||
SelectItemWithParentheses <- "(" ws ex:OrExpression ws ")" { return ex, nil }
|
||||
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
|
||||
}
|
||||
/ inv:(Not ws)? ex:SelectItem {
|
||||
if inv != nil {
|
||||
ex1 := ex.(parsers.SelectItem)
|
||||
@@ -673,6 +681,8 @@ ThreeArgumentStringFunctionExpression <- function:ThreeArgumentStringFunction ws
|
||||
functionType = parsers.FunctionCallEndsWith
|
||||
case "STARTSWITH":
|
||||
functionType = parsers.FunctionCallStartsWith
|
||||
case "REGEXMATCH":
|
||||
functionType = parsers.FunctionCallRegexMatch
|
||||
case "INDEX_OF":
|
||||
functionType = parsers.FunctionCallIndexOf
|
||||
}
|
||||
@@ -680,7 +690,7 @@ ThreeArgumentStringFunctionExpression <- function:ThreeArgumentStringFunction ws
|
||||
return createFunctionCall(functionType, []interface{}{ex1, ex2, ignoreCase})
|
||||
}
|
||||
|
||||
ThreeArgumentStringFunction <- ("CONTAINS"i / "ENDSWITH"i / "STARTSWITH"i / "INDEX_OF"i) {
|
||||
ThreeArgumentStringFunction <- ("CONTAINS"i / "ENDSWITH"i / "STARTSWITH"i / "REGEXMATCH"i / "INDEX_OF"i) {
|
||||
return string(c.text), nil
|
||||
}
|
||||
|
||||
|
||||
@@ -168,6 +168,32 @@ 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 {
|
||||
if len(arguments) > 2 && arguments[2] != nil {
|
||||
boolExpr := r.resolveSelectItem(arguments[2].(parsers.SelectItem))
|
||||
if boolValue, ok := boolExpr.(bool); ok {
|
||||
partialSearch = boolValue
|
||||
|
||||
@@ -162,6 +162,8 @@ 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:
|
||||
|
||||
@@ -2,6 +2,7 @@ package memoryexecutor
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"regexp"
|
||||
"strings"
|
||||
|
||||
"github.com/pikami/cosmium/internal/logger"
|
||||
@@ -75,6 +76,46 @@ 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 := ""
|
||||
|
||||
@@ -318,6 +359,20 @@ 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)
|
||||
@@ -329,6 +384,41 @@ 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,6 +231,42 @@ 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,
|
||||
|
||||
Reference in New Issue
Block a user