nodeGroup.go

package node

import (
	"net/http"
	"strconv"

	goku_handler "github.com/eolinker/goku-api-gateway/goku-handler"

	"github.com/eolinker/goku-api-gateway/console/controller"
	"github.com/eolinker/goku-api-gateway/console/module/cluster"
	"github.com/eolinker/goku-api-gateway/console/module/node"
)

//GroupHandlers groupHandlers
type GroupHandlers struct {
}

//Handlers handlers
func (h *GroupHandlers) Handlers(factory *goku_handler.AccountHandlerFactory) map[string]http.Handler {
	return map[string]http.Handler{
		"/add":     factory.NewAccountHandleFunction(operationNode, true, AddNodeGroup),
		"/edit":    factory.NewAccountHandleFunction(operationNode, true, EditNodeGroup),
		"/delete":  factory.NewAccountHandleFunction(operationNode, true, DeleteNodeGroup),
		"/getInfo": factory.NewAccountHandleFunction(operationNode, false, GetNodeGroupInfo),
		"/getList": factory.NewAccountHandleFunction(operationNode, false, GetNodeGroupList),
	}
}

//NewGroupHandlers new groupHandlers
func NewGroupHandlers() *GroupHandlers {
	return &GroupHandlers{}
}

//AddNodeGroup 新增节点分组
func AddNodeGroup(httpResponse http.ResponseWriter, httpRequest *http.Request) {

	cluserName := httpRequest.PostFormValue("cluster")

	clusterID := cluster.GetClusterIDByName(cluserName)
	if clusterID == 0 {
		controller.WriteError(httpResponse, "340003", "", "[ERROR]Illegal cluster!", nil)
		return
	}
	groupName := httpRequest.PostFormValue("groupName")
	if groupName == "" {
		controller.WriteError(httpResponse,
			"280002",
			"nodeGroup",
			"[ERROR]Illegal groupName!",
			nil)
		return

	}
	flag, result, err := node.AddNodeGroup(groupName, clusterID)
	if !flag {

		controller.WriteError(httpResponse,
			"280000",
			"nodeGroup",
			result.(string),
			err)
		return

	}
	controller.WriteResultInfo(httpResponse, "node", "groupID", result)

	return
}

//EditNodeGroup 修改节点分组信息
func EditNodeGroup(httpResponse http.ResponseWriter, httpRequest *http.Request) {

	groupName := httpRequest.PostFormValue("groupName")
	groupID := httpRequest.PostFormValue("groupID")
	if groupName == "" {
		controller.WriteError(httpResponse,
			"280002",
			"nodeGroup",
			"[ERROR]Illegal groupName!",
			nil)
		return

	}
	id, err := strconv.Atoi(groupID)
	if err != nil {
		controller.WriteError(httpResponse,
			"280001",
			"nodeGroup",
			"[ERROR]Illegal groupID!",
			err)
		return

	}

	flag, result, err := node.EditNodeGroup(groupName, id)
	if !flag {

		controller.WriteError(httpResponse,
			"280000",
			"nodeGroup",
			result,
			err)
		return

	}

	controller.WriteResultInfo(httpResponse, "nodeGroup", "", nil)
}

//DeleteNodeGroup 删除节点分组
func DeleteNodeGroup(httpResponse http.ResponseWriter, httpRequest *http.Request) {

	groupID := httpRequest.PostFormValue("groupID")

	id, err := strconv.Atoi(groupID)
	if err != nil {
		controller.WriteError(httpResponse,
			"280001",
			"nodeGroup",
			"[ERROR]Illegal groupID!",
			err)
		return

	}
	flag, result, err := node.GetRunningNodeCount(id)
	if !flag {
		controller.WriteError(httpResponse,
			"280013",
			"nodeGroup",
			result.(string),
			err)
		return

	}
	if result.(int) > 0 {
		controller.WriteError(httpResponse,
			"280013",
			"nodeGroup",
			"[ERROR]Contains running nodes",
			err)
		return
	}
	flag, result, err = node.DeleteNodeGroup(id)
	if !flag {

		controller.WriteError(httpResponse,
			"280000",
			"nodeGroup",
			result.(string),
			err)
		return
	}

	controller.WriteResultInfo(httpResponse, "nodeGroup", "", nil)

}

//GetNodeGroupInfo 获取节点分组信息
func GetNodeGroupInfo(httpResponse http.ResponseWriter, httpRequest *http.Request) {

	groupID := httpRequest.PostFormValue("groupID")

	id, err := strconv.Atoi(groupID)
	if err != nil {
		controller.WriteError(httpResponse,
			"280001",
			"nodeGroup",
			"[ERROR]Illegal groupID!",
			err)
		return

	}
	flag, result, err := node.GetNodeGroupInfo(id)
	if !flag {

		controller.WriteError(httpResponse,
			"280000",
			"nodeGroup",
			"[ERROR]The node group information does not exist!",
			err)
		return
	}

	controller.WriteResultInfo(httpResponse, "node", "groupInfo", result)
}

//GetNodeGroupList 获取节点分组列表
func GetNodeGroupList(httpResponse http.ResponseWriter, httpRequest *http.Request) {

	cluserName := httpRequest.FormValue("cluster")
	clusterID := cluster.GetClusterIDByName(cluserName)
	if clusterID == 0 {
		controller.WriteError(httpResponse,
			"280001",
			"nodeGroup",
			"[ERROR]Illegal cluster!",
			nil)
		return
	}
	flag, result, err := node.GetNodeGroupList(clusterID)
	if !flag {

		controller.WriteError(httpResponse,
			"280000",
			"nodeGroup",
			"[ERROR]Empty group list!",
			err)
		return
	}
	controller.WriteResultInfo(httpResponse, "node", "groupList", result)
}