mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-16 09:15:26 +01:00
92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
|
package logstorage
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"net/http"
|
||
|
"strconv"
|
||
|
|
||
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/encoding"
|
||
|
)
|
||
|
|
||
|
// TenantID is an id of a tenant for log streams.
|
||
|
//
|
||
|
// Each log stream is associated with a single TenantID.
|
||
|
type TenantID struct {
|
||
|
// AccountID is the id of the account for the log stream.
|
||
|
AccountID uint32
|
||
|
|
||
|
// ProjectID is the id of the project for the log stream.
|
||
|
ProjectID uint32
|
||
|
}
|
||
|
|
||
|
// Reset resets tid.
|
||
|
func (tid *TenantID) Reset() {
|
||
|
tid.AccountID = 0
|
||
|
tid.ProjectID = 0
|
||
|
}
|
||
|
|
||
|
// String returns human-readable representation of tid
|
||
|
func (tid *TenantID) String() string {
|
||
|
return fmt.Sprintf("{accountID=%d,projectID=%d}", tid.AccountID, tid.ProjectID)
|
||
|
}
|
||
|
|
||
|
// equal returns true if tid equals to a.
|
||
|
func (tid *TenantID) equal(a *TenantID) bool {
|
||
|
return tid.AccountID == a.AccountID && tid.ProjectID == a.ProjectID
|
||
|
}
|
||
|
|
||
|
// less returns true if tid is less than a.
|
||
|
func (tid *TenantID) less(a *TenantID) bool {
|
||
|
if tid.AccountID != a.AccountID {
|
||
|
return tid.AccountID < a.AccountID
|
||
|
}
|
||
|
return tid.ProjectID < a.ProjectID
|
||
|
}
|
||
|
|
||
|
// marshal appends the marshaled tid to dst and returns the result
|
||
|
func (tid *TenantID) marshal(dst []byte) []byte {
|
||
|
dst = encoding.MarshalUint32(dst, tid.AccountID)
|
||
|
dst = encoding.MarshalUint32(dst, tid.ProjectID)
|
||
|
return dst
|
||
|
}
|
||
|
|
||
|
// unmarshal unmarshals tid from src and returns the remaining tail.
|
||
|
func (tid *TenantID) unmarshal(src []byte) ([]byte, error) {
|
||
|
if len(src) < 8 {
|
||
|
return src, fmt.Errorf("cannot unmarshal tenantID from %d bytes; need at least 8 bytes", len(src))
|
||
|
}
|
||
|
tid.AccountID = encoding.UnmarshalUint32(src[:4])
|
||
|
tid.ProjectID = encoding.UnmarshalUint32(src[4:])
|
||
|
return src[8:], nil
|
||
|
}
|
||
|
|
||
|
// GetTenantIDFromRequest returns tenantID from r.
|
||
|
func GetTenantIDFromRequest(r *http.Request) (TenantID, error) {
|
||
|
var tenantID TenantID
|
||
|
|
||
|
accountID, err := getUint32FromHeader(r, "AccountID")
|
||
|
if err != nil {
|
||
|
return tenantID, err
|
||
|
}
|
||
|
projectID, err := getUint32FromHeader(r, "ProjectID")
|
||
|
if err != nil {
|
||
|
return tenantID, err
|
||
|
}
|
||
|
|
||
|
tenantID.AccountID = accountID
|
||
|
tenantID.ProjectID = projectID
|
||
|
return tenantID, nil
|
||
|
}
|
||
|
|
||
|
func getUint32FromHeader(r *http.Request, headerName string) (uint32, error) {
|
||
|
s := r.Header.Get(headerName)
|
||
|
if len(s) == 0 {
|
||
|
return 0, nil
|
||
|
}
|
||
|
n, err := strconv.ParseUint(s, 10, 32)
|
||
|
if err != nil {
|
||
|
return 0, fmt.Errorf("cannot parse %s header %q: %w", headerName, s, err)
|
||
|
}
|
||
|
return uint32(n), nil
|
||
|
}
|