cmd

type AuditLogDecoder interface {
	CmdDecoder

	SetPSCloseStrategy(s PSCloseStrategy)
	SetIDAllocator(alloc *ConnIDAllocator)
	SetCommandEndTime(t time.Time)
	EnableFilterCommandWithRetry()
	SetUserAllowlist(users []string)
}

type AuditLogExtensionDecoder struct {
	// contains filtered or unexported fields
}

type AuditLogPluginDecoder struct {
	// contains filtered or unexported fields
}

type CmdDecoder interface {
	Decode(reader LineReader) (c *Command, err error)

	SetCommandStartTime(t time.Time)
}

type CmdEncoder interface {
	Encode(c *Command, writer *bytes.Buffer) error
}

type Command struct {
	PreparedStmt string
	// CapturedPsID is the prepared statement ID in capture.
	// The Execute command needs to update the prepared statement ID in replay.
	CapturedPsID uint32
	Params       []any

	// Payload starts with command type so that replay can reuse this byte array.
	Payload []byte
	StartTs time.Time
	// For audit log plugin, the decoder will allocate a new id to avoid id collision. To make it easier
	// to debug, we keep the upstream connection id here to store it in the exception report.
	UpstreamConnID uint64
	ConnID         uint64
	Type           pnet.Command
	// When the replay connection disconnects, the replayer reconnects based on the current DB of the next command.
	CurDB string
	// The place in the traffic file, used to report.
	FileName string
	Line     int
	// Logged only in audit log.
	StmtType string
	EndTs    time.Time

	// Logged only in native log.
	Success bool
	// contains filtered or unexported fields
}

type ConnIDAllocator struct {
	// contains filtered or unexported fields
}

type DeDup struct {
	sync.Mutex
	Items map[string]DupItem
}

type DupItem struct {
	Times      int
	MinOverlap time.Duration
	Cost       time.Duration
}

type LineReader interface {
	String() string
	ReadLine() ([]byte, string, int, error)
	Read([]byte) (string, int, error)
	Close()
}

type NativeDecoder struct {
	// contains filtered or unexported fields
}

type NativeEncoder struct {
}

type PSCloseStrategy string

type TrafficFormat string