mirror of
https://github.com/matrix-org/dendrite.git
synced 2025-12-29 01:33:10 -06:00
49f8c7fe38
- Use the $$NULL$$ for the known nullable use cases Co-authored-by: alexf@example.com <alexf@example.com>
51 lines
1.3 KiB
Go
51 lines
1.3 KiB
Go
package cosmosdbapi
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
type CosmosDocument struct {
|
|
Id string `json:"id"`
|
|
Pk string `json:"_pk"`
|
|
Tn string `json:"_sid"`
|
|
Cn string `json:"_cn"`
|
|
Ct string `json:"_ct"`
|
|
Ut string `json:"_ut"`
|
|
ETag string `json:"_etag"`
|
|
Timestamp int64 `json:"_ts"`
|
|
}
|
|
|
|
var DocumentIdPartNullString string = "$$NULL$$"
|
|
|
|
func removeSpecialChars(docId string) string {
|
|
// The following characters are restricted and cannot be used in the Id property: '/', '\', '?', '#'
|
|
invalidChars := [4]string{"/", "\\", "?", "#"}
|
|
replaceChar := ","
|
|
result := docId
|
|
for _, invalidChar := range invalidChars {
|
|
result = strings.ReplaceAll(result, invalidChar, replaceChar)
|
|
}
|
|
return result
|
|
}
|
|
|
|
func EnsureIdPart(idPart string) string {
|
|
if len(idPart) == 0 {
|
|
return DocumentIdPartNullString
|
|
}
|
|
return idPart
|
|
}
|
|
|
|
func GetDocumentId(tenantName string, collectionName string, id string) string {
|
|
safeId := removeSpecialChars(id)
|
|
return fmt.Sprintf("%s,%s,%s", collectionName, tenantName, safeId)
|
|
}
|
|
|
|
func GetPartitionKeyByCollection(tenantName string, collectionName string) string {
|
|
return fmt.Sprintf("%s,%s", collectionName, tenantName)
|
|
}
|
|
|
|
func GetPartitionKeyByUniqueId(tenantName string, collectionName string, uniqueId string) string {
|
|
return fmt.Sprintf("%s,%s,%s", collectionName, tenantName, uniqueId)
|
|
}
|