notify/bulk_snapshot.go

package notify

import (
	"errors"
	"sort"
	"time"
)

type BulkSnapshot struct {
	ID                                  string
	DataID                              uint64
	FastPathVersion                     uint8
	Scope                               string
	Range                               BulkRange
	Metadata                            BulkMetadata
	BindingOwner                        string
	BindingAlive                        bool
	BindingCurrent                      bool
	BindingReason                       string
	BindingError                        string
	BindingBulkAdaptiveSoftPayloadBytes int
	Dedicated                           bool
	DedicatedLaneID                     uint32
	DedicatedAttached                   bool
	DedicatedAttachState                string
	DedicatedAttachAttempts             uint32
	DedicatedAttachLastCode             string
	DedicatedDataStarted                bool
	SessionEpoch                        uint64
	LogicalClientID                     string
	TransportGeneration                 uint64
	TransportAttached                   bool
	TransportHasRuntimeConn             bool
	TransportCurrent                    bool
	TransportDetachReason               string
	TransportDetachKind                 string
	TransportDetachGeneration           uint64
	TransportDetachError                string
	TransportDetachedAt                 time.Time
	ReattachEligible                    bool
	LocalClosed                         bool
	LocalReadClosed                     bool
	RemoteClosed                        bool
	PeerReadClosed                      bool
	BufferedChunks                      int
	BufferedBytes                       int
	ReadTimeout                         time.Duration
	WriteTimeout                        time.Duration
	ChunkSize                           int
	WindowBytes                         int
	MaxInFlight                         int
	BytesRead                           int64
	BytesWritten                        int64
	ReadCalls                           int64
	WriteCalls                          int64
	OpenedAt                            time.Time
	LastReadAt                          time.Time
	LastWriteAt                         time.Time
	ResetError                          string
}

type clientBulkSnapshotReader interface {
	clientBulkSnapshots() []BulkSnapshot
}

type serverBulkSnapshotReader interface {
	serverBulkSnapshots() []BulkSnapshot
}

var (
	errClientBulkSnapshotNil         = errors.New("client bulk snapshot target is nil")
	errServerBulkSnapshotNil         = errors.New("server bulk snapshot target is nil")
	errClientBulkSnapshotUnsupported = errors.New("client bulk snapshot target type is unsupported")
	errServerBulkSnapshotUnsupported = errors.New("server bulk snapshot target type is unsupported")
)

func GetClientBulkSnapshots(c Client) ([]BulkSnapshot, error) {
	if c == nil {
		return nil, errClientBulkSnapshotNil
	}
	reader, ok := any(c).(clientBulkSnapshotReader)
	if !ok {
		return nil, errClientBulkSnapshotUnsupported
	}
	return reader.clientBulkSnapshots(), nil
}

func GetServerBulkSnapshots(s Server) ([]BulkSnapshot, error) {
	if s == nil {
		return nil, errServerBulkSnapshotNil
	}
	reader, ok := any(s).(serverBulkSnapshotReader)
	if !ok {
		return nil, errServerBulkSnapshotUnsupported
	}
	return reader.serverBulkSnapshots(), nil
}

func (c *ClientCommon) clientBulkSnapshots() []BulkSnapshot {
	return bulkSnapshotsFromRuntime(c.getBulkRuntime())
}

func (s *ServerCommon) serverBulkSnapshots() []BulkSnapshot {
	return bulkSnapshotsFromRuntime(s.getBulkRuntime())
}

func bulkSnapshotsFromRuntime(runtime *bulkRuntime) []BulkSnapshot {
	if runtime == nil {
		return nil
	}
	return runtime.snapshots()
}

func sortBulkSnapshots(src []BulkSnapshot) {
	sort.Slice(src, func(i, j int) bool {
		if src[i].Scope != src[j].Scope {
			return src[i].Scope < src[j].Scope
		}
		if src[i].ID != src[j].ID {
			return src[i].ID < src[j].ID
		}
		if src[i].DataID != src[j].DataID {
			return src[i].DataID < src[j].DataID
		}
		return src[i].TransportGeneration < src[j].TransportGeneration
	})
}