migrate assign volume to grpc API on volume server

weed/topology/allocate_volume.go

@@ -1,13 +1,11 @@
 package topology
 
 import (
-	"encoding/json"
-	"errors"
-	"fmt"
-	"net/url"
+	"context"
 
 	"github.com/chrislusf/seaweedfs/weed/storage"
-	"github.com/chrislusf/seaweedfs/weed/util"
+	"github.com/chrislusf/seaweedfs/weed/operation"
+	"github.com/chrislusf/seaweedfs/weed/pb/volume_server_pb"
 )
 
 type AllocateVolumeResult struct {
@@ -15,22 +13,16 @@ type AllocateVolumeResult struct {
 }
 
 func AllocateVolume(dn *DataNode, vid storage.VolumeId, option *VolumeGrowOption) error {
-	values := make(url.Values)
-	values.Add("volume", vid.String())
-	values.Add("collection", option.Collection)
-	values.Add("replication", option.ReplicaPlacement.String())
-	values.Add("ttl", option.Ttl.String())
-	values.Add("preallocate", fmt.Sprintf("%d", option.Prealloacte))
-	jsonBlob, err := util.Post("http://"+dn.Url()+"/admin/assign_volume", values)
-	if err != nil {
-		return err
-	}
-	var ret AllocateVolumeResult
-	if err := json.Unmarshal(jsonBlob, &ret); err != nil {
-		return fmt.Errorf("Invalid JSON result for %s: %s", "/admin/assign_volum", string(jsonBlob))
-	}
-	if ret.Error != "" {
-		return errors.New(ret.Error)
-	}
-	return nil
+
+	return operation.WithVolumeServerClient(dn.Url(), func(client volume_server_pb.VolumeServerClient) error {
+		_, deleteErr := client.AssignVolume(context.Background(), &volume_server_pb.AssignVolumeRequest{
+			VolumdId:    uint32(vid),
+			Collection:  option.Collection,
+			Replication: option.ReplicaPlacement.String(),
+			Ttl:         option.Ttl.String(),
+			Preallocate: option.Prealloacte,
+		})
+		return deleteErr
+	})
+
 }