Cloudpods/backend/vendor/yunion.io/x/pkg/util/compare/syncresults.go

// Copyright 2019 Yunion
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package compare

import (
	"fmt"
	"strings"

	"yunion.io/x/pkg/errors"
)

type SyncResult struct {
	AddCnt       int `json:"add_cnt,omitzero"`
	AddErrCnt    int `json:"add_err_cnt,omitzero"`
	UpdateCnt    int `json:"update_cnt,omitzero"`
	UpdateErrCnt int `json:"update_err_cnt,omitzero"`
	DelCnt       int `json:"del_cnt,omitzero"`
	DelErrCnt    int `json:"del_err_cnt,omitzero"`
	errors       []error
}

func (self *SyncResult) Error(msg error) {
	if self.errors == nil {
		self.errors = make([]error, 0)
	}
	self.errors = append(self.errors, msg)
}

func (self *SyncResult) Add() {
	self.AddCnt += 1
}

func (self *SyncResult) AddError(msg error) {
	self.AddErrCnt += 1
	self.Error(msg)
}

func (self *SyncResult) Update() {
	self.UpdateCnt += 1
}

func (self *SyncResult) UpdateError(msg error) {
	self.UpdateErrCnt += 1
	self.Error(msg)
}

func (self *SyncResult) Delete() {
	self.DelCnt += 1
}

func (self *SyncResult) DeleteError(msg error) {
	self.DelErrCnt += 1
	self.Error(msg)
}

func (self *SyncResult) AllError() error {
	msgs := make(map[string]bool)
	for _, e := range self.errors {
		msg := e.Error()
		msgs[msg] = true
	}
	ret := make([]string, len(msgs))
	i := 0
	for m := range msgs {
		ret[i] = m
		i += 1
	}
	return fmt.Errorf(strings.Join(ret, ";"))
}

func (self *SyncResult) IsError() bool {
	return self.errors != nil && len(self.errors) > 0
}

func (self *SyncResult) IsGenerateError() bool {
	for _, err := range self.errors {
		if e := errors.Cause(err); e != errors.ErrNotImplemented && e != errors.ErrNotSupported && e != errors.ErrNotEmpty {
			return true
		}
	}
	return false
}

func (self *SyncResult) Result() string {
	msg := fmt.Sprintf("removed %d failed %d updated %d failed %d added %d failed %d", self.DelCnt, self.DelErrCnt, self.UpdateCnt, self.UpdateErrCnt, self.AddCnt, self.AddErrCnt)
	if self.IsError() {
		msg = fmt.Sprintf("%s;%s", msg, self.AllError())
	}
	return msg
}