Zhen Ye bb8d1ab3bf
enhance: make new go package to manage proto (#39114)
issue: #39095

---------

Signed-off-by: chyezh <chyezh@outlook.com>
2025-01-10 10:49:01 +08:00

52 lines
1.1 KiB
Go

package model
import pb "github.com/milvus-io/milvus/pkg/proto/etcdpb"
type Alias struct {
Name string
CollectionID int64
CreatedTime uint64
State pb.AliasState
DbID int64
}
func (a *Alias) Available() bool {
return a.State == pb.AliasState_AliasCreated
}
func (a *Alias) Clone() *Alias {
return &Alias{
Name: a.Name,
CollectionID: a.CollectionID,
CreatedTime: a.CreatedTime,
State: a.State,
DbID: a.DbID,
}
}
func (a *Alias) Equal(other Alias) bool {
return a.Name == other.Name &&
a.CollectionID == other.CollectionID &&
a.DbID == other.DbID
}
func MarshalAliasModel(alias *Alias) *pb.AliasInfo {
return &pb.AliasInfo{
AliasName: alias.Name,
CollectionId: alias.CollectionID,
CreatedTime: alias.CreatedTime,
State: alias.State,
DbId: alias.DbID,
}
}
func UnmarshalAliasModel(info *pb.AliasInfo) *Alias {
return &Alias{
Name: info.GetAliasName(),
CollectionID: info.GetCollectionId(),
CreatedTime: info.GetCreatedTime(),
State: info.GetState(),
DbID: info.GetDbId(),
}
}