mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-30 15:35:33 +08:00
0a465bb5b7
issue: #41544 - add lock interceptor into wal. - use recovery and shardmanager to replace the original implementation of segment assignment. - remove redundant implementation and unittest. - remove redundant proto definition. - use 2 streamingnode in e2e. --------- Signed-off-by: chyezh <chyezh@outlook.com>
51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
package message
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/milvus-io/milvus/pkg/v2/proto/messagespb"
|
|
)
|
|
|
|
type (
|
|
TxnState = messagespb.TxnState
|
|
TxnID int64
|
|
)
|
|
|
|
const (
|
|
TxnStateInFlight TxnState = messagespb.TxnState_TxnInFlight
|
|
TxnStateOnCommit TxnState = messagespb.TxnState_TxnOnCommit
|
|
TxnStateCommitted TxnState = messagespb.TxnState_TxnCommitted
|
|
TxnStateOnRollback TxnState = messagespb.TxnState_TxnOnRollback
|
|
TxnStateRollbacked TxnState = messagespb.TxnState_TxnRollbacked
|
|
|
|
NonTxnID = TxnID(-1)
|
|
)
|
|
|
|
// NewTxnContextFromProto generates TxnContext from proto message.
|
|
func NewTxnContextFromProto(proto *messagespb.TxnContext) *TxnContext {
|
|
if proto == nil {
|
|
return nil
|
|
}
|
|
return &TxnContext{
|
|
TxnID: TxnID(proto.TxnId),
|
|
Keepalive: time.Duration(proto.KeepaliveMilliseconds) * time.Millisecond,
|
|
}
|
|
}
|
|
|
|
// TxnContext is the transaction context for message.
|
|
type TxnContext struct {
|
|
TxnID TxnID
|
|
Keepalive time.Duration
|
|
}
|
|
|
|
// IntoProto converts TxnContext to proto message.
|
|
func (t *TxnContext) IntoProto() *messagespb.TxnContext {
|
|
if t == nil {
|
|
return nil
|
|
}
|
|
return &messagespb.TxnContext{
|
|
TxnId: int64(t.TxnID),
|
|
KeepaliveMilliseconds: t.Keepalive.Milliseconds(),
|
|
}
|
|
}
|