mirror of https://github.com/chrislusf/seaweedfs
45 lines
1.4 KiB
Go
45 lines
1.4 KiB
Go
package topology
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/seaweedfs/seaweedfs/weed/operation"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/volume_server_pb"
|
|
"github.com/seaweedfs/seaweedfs/weed/storage/needle"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type AllocateVolumeResult struct {
|
|
Error string
|
|
}
|
|
|
|
func AllocateVolume(dn *DataNode, grpcDialOption grpc.DialOption, vid needle.VolumeId, option *VolumeGrowOption) error {
|
|
|
|
return operation.WithVolumeServerClient(false, dn.ServerAddress(), grpcDialOption, func(client volume_server_pb.VolumeServerClient) error {
|
|
|
|
_, allocateErr := client.AllocateVolume(context.Background(), &volume_server_pb.AllocateVolumeRequest{
|
|
VolumeId: uint32(vid),
|
|
Collection: option.Collection,
|
|
Replication: option.ReplicaPlacement.String(),
|
|
Ttl: option.Ttl.String(),
|
|
Preallocate: option.Preallocate,
|
|
MemoryMapMaxSizeMb: option.MemoryMapMaxSizeMb,
|
|
DiskType: string(option.DiskType),
|
|
})
|
|
return allocateErr
|
|
})
|
|
|
|
}
|
|
|
|
func DeleteVolume(dn *DataNode, grpcDialOption grpc.DialOption, vid needle.VolumeId) error {
|
|
|
|
return operation.WithVolumeServerClient(false, dn.ServerAddress(), grpcDialOption, func(client volume_server_pb.VolumeServerClient) error {
|
|
|
|
_, allocateErr := client.VolumeDelete(context.Background(), &volume_server_pb.VolumeDeleteRequest{
|
|
VolumeId: uint32(vid),
|
|
})
|
|
return allocateErr
|
|
})
|
|
|
|
}
|