replication.go

package hdfs

import (
	"errors"
	"os"

	hdfs "github.com/colinmarc/hdfs/v2/internal/protocol/hadoop_hdfs"
	"google.golang.org/protobuf/proto"
)

// Sets replication factor for a file
func (c *Client) SetReplication(name string, replication int16) (bool, error) {
	return setReplication(c, name, replication)
}

func setReplication(c *Client, name string, replication int16) (bool, error) {
	_, err := c.getFileInfo(name)
	if err != nil {
		return false, &os.PathError{Op: "remove", Path: name, Err: err}
	}

	req := &hdfs.SetReplicationRequestProto{
		Src:         proto.String(name),
		Replication: proto.Uint32(uint32(replication)),
	}
	resp := &hdfs.SetReplicationResponseProto{}

	err = c.leaderNamenode.Execute("setReplication", req, resp)
	if err != nil {
		return false, &os.PathError{Op: "setReplication", Path: name,
			Err: interpretException(err)}
	} else if resp.Result == nil {
		return false, &os.PathError{Op: "setReplication", Path: name,
			Err: errors.New("unexpected empty response")}
	}

	return *resp.Result, nil
}