utils.go

package manage

import (
	"errors"
	"io/ioutil"
	"net/http"

	"github.com/golang/glog"
	"golang.org/x/net/context"

	"github.com/cloudstax/firecamp/common"
	"github.com/cloudstax/firecamp/db"
)

func ConvertToHTTPError(err error) (errmsg string, errcode int) {
	switch err {
	case common.ErrServiceExist:
		return err.Error(), http.StatusConflict
	case common.ErrTimeout:
		return err.Error(), http.StatusRequestTimeout
	case db.ErrDBInvalidRequest:
		return http.StatusText(http.StatusBadRequest), http.StatusBadRequest
	case db.ErrDBRecordNotFound:
		return http.StatusText(http.StatusNotFound), http.StatusNotFound
	case db.ErrDBConditionalCheckFailed:
		return err.Error(), http.StatusPreconditionFailed
	case db.ErrDBTableNotFound:
		return http.StatusText(http.StatusNotFound), http.StatusNotFound
	}
	return err.Error(), http.StatusInternalServerError
}

func ConvertHTTPError(resp *http.Response) error {
	if resp.StatusCode == http.StatusOK {
		return nil
	}

	defer resp.Body.Close()
	body, _ := ioutil.ReadAll(resp.Body)

	switch resp.StatusCode {
	case http.StatusConflict:
		return errors.New("ServiceExist: " + string(body))
	case http.StatusRequestTimeout:
		return errors.New("Timeout: " + string(body))
	case http.StatusBadRequest:
		return errors.New("InvalidArgs: " + string(body))
	case http.StatusNotFound:
		return errors.New("NotFound: " + string(body))
	case http.StatusPreconditionFailed:
		return errors.New("ConditionalCheckFailed: " + string(body))
	default:
		return errors.New("InternalError: " + string(body))
	}
}

func CreateConfigFile(ctx context.Context, dbIns db.DB, cfgfile *common.ConfigFile, requuid string) (*common.ConfigFile, error) {
	err := dbIns.CreateConfigFile(ctx, cfgfile)
	if err == nil {
		glog.Infoln("created config file", db.PrintConfigFile(cfgfile), "requuid", requuid)
		return cfgfile, nil
	}

	if err != db.ErrDBConditionalCheckFailed {
		glog.Errorln("CreateConfigFile error", err, db.PrintConfigFile(cfgfile), "requuid", requuid)
		return nil, err
	}

	// config file exists, check whether it is a retry request.
	currcfg, err := dbIns.GetConfigFile(ctx, cfgfile.ServiceUUID, cfgfile.FileID)
	if err != nil {
		glog.Errorln("get existing config file error", err, db.PrintConfigFile(cfgfile), "requuid", requuid)
		return nil, err
	}

	skipMtime := true
	skipContent := true
	if !db.EqualConfigFile(currcfg, cfgfile, skipMtime, skipContent) {
		glog.Errorln("config file not match, current", currcfg, "new", cfgfile, "requuid", requuid)
		return nil, common.ErrConfigMismatch
	}

	glog.Infoln("config file exists", db.PrintConfigFile(currcfg), "requuid", requuid)
	return currcfg, nil
}