Add underRepair for allocation

Author: KrishnaDeqode

zboxcore/sdk/allocation.go

@@ -26,6 +26,7 @@ import (
 var (
 	noBLOBBERS     = errors.New("No Blobbers set in this allocation")
 	notInitialized = common.NewError("sdk_not_initialized", "Please call InitStorageSDK Init and use GetAllocation to get the allocation object")
+	underRepair    = common.NewError("allocaton_under_repair", "Allocation is under repair, Please try again later")
 )
 
 type MetaOperation struct {
@@ -83,12 +84,21 @@ type Allocation struct {
 	mutex               *sync.Mutex
 	downloadProgressMap map[string]*DownloadRequest
 	initialized         bool
+	underRepair         bool
+}
+
+func (a *Allocation) UnderRepair() bool {
+	return a.underRepair
 }
 
 func (a *Allocation) GetStats() *AllocationStats {
 	return a.Stats
 }
 
+func (a *Allocation) UpdateRepairStatus(value bool) {
+	a.underRepair = value
+}
+
 func (a *Allocation) InitAllocation() {
 	// if a.uploadChan != nil {
 	// 	close(a.uploadChan)
@@ -173,6 +183,9 @@ func (a *Allocation) uploadOrUpdateFile(localpath string, remotepath string, sta
 	if !a.isInitialized() {
 		return notInitialized
 	}
+	if a.UnderRepair() {
+		return underRepair
+	}
 	fileInfo, err := os.Stat(localpath)
 	if err != nil {
 		return fmt.Errorf("Local file error: %s", err.Error())
@@ -236,6 +249,9 @@ func (a *Allocation) downloadFile(localPath string, remotePath string, contentMo
 	if !a.isInitialized() {
 		return notInitialized
 	}
+	if a.UnderRepair() {
+		return underRepair
+	}
 	if stat, err := os.Stat(localPath); err == nil {
 		if !stat.IsDir() {
 			return fmt.Errorf("Local path is not a directory '%s'", localPath)
@@ -437,6 +453,9 @@ func (a *Allocation) DeleteFile(path string) error {
 	if !a.isInitialized() {
 		return notInitialized
 	}
+	if a.UnderRepair() {
+		return underRepair
+	}
 	if len(path) == 0 {
 		return common.NewError("invalid_path", "Invalid path for the list")
 	}
@@ -577,6 +596,9 @@ func (a *Allocation) downloadFromAuthTicket(localPath string, authTicket string,
 	if !a.isInitialized() {
 		return notInitialized
 	}
+	if a.UnderRepair() {
+		return underRepair
+	}
 	sEnc, err := base64.StdEncoding.DecodeString(authTicket)
 	if err != nil {
 		return common.NewError("auth_ticket_decode_error", "Error decoding the auth ticket."+err.Error())