Golang util.HttpResponse类代码示例

/*
function:
	check if the user provides valid taskname.
return:

*/
func checkTaskName(w *http.ResponseWriter, r *http.Request, Args []string) (thisTask Task, err error) {
	var response util.HttpResponse
	if len(Args) < 1 {
		response.Set(util.INVALID_INPUT, "You must provide one taskname to the command.")
		io.WriteString(*w, response.String())
		return thisTask, errors.New("Taskname is not provided.")
	}

	thistask, exists := Tasks[Args[0]]
	if !exists {
		response.Set(util.INVALID_INPUT, "There is no task with the name of "+`'`+Args[0]+`'`)
		io.WriteString(*w, response.String())
		return thistask, errors.New("Taskname is invalid.")
	}
	return thistask, nil
}

/*Function:
get the information in http.Request to 'inputInfo'.
return value:
	1)succeed: the data was stored 'inputInfo'
	2)Fail:the error information is write into 'w'
*/
func GetInfoFromRequest(w *http.ResponseWriter, r *http.Request) (inputInfo util.SendCmd, err error) {
	var response util.HttpResponse

	var input []byte
	input, err = util.ReadContentFromHttpRequest(r)
	if err != nil {
		response.Set(util.SERVER_ERROR, err.Error()+"in server Handler()")
		io.WriteString(*w, response.String())
		return inputInfo, err
	}

	err = json.Unmarshal(input, &inputInfo)
	if err != nil {
		response.Set(util.SERVER_ERROR, err.Error()+"in Server TaskHandler()")
		io.WriteString(*w, response.String())
		return inputInfo, err
	}
	return inputInfo, nil
}

/*function:create task from the user command.
	we will create the task and store it in Tasks
param:
	r: r.Body include the string of the user command
	w:	w.Body include the result of the create.response will be marshal to w.
*/
func CreateTaskHandler(w http.ResponseWriter, r *http.Request) {
	var response util.HttpResponse
	var thisTask Task
	response.Status = util.INVALID_INPUT

	var contents []byte
	contents = make([]byte, 1000)
	length, err := r.Body.Read(contents)
	if err != nil && err != io.EOF {
		fmt.Fprintf(os.Stderr, "CreateHandler: can not read from http resposeWriter\n")
		fmt.Fprintf(os.Stderr, "%s", err)
		response.Warnings = []string{"CreateHandler: can not read from http resposeWriter\n" + err.Error()}
		io.WriteString(w, response.String())
		return
	}
	var res util.SendCmd
	if *FlagDebug {
		fmt.Println("contents:", string(contents))
	}
	//make sure the char in contents should not include '\0'
	contentsRight := contents[:length]
	errunmarshal := json.Unmarshal(contentsRight, &res)
	if errunmarshal != nil {
		fmt.Fprintf(os.Stderr, "CreateHandler: Unmarshal failed for contents: ")
		fmt.Fprintf(os.Stderr, "%s", errunmarshal)
		response.Warnings = []string{"CreateHandler: Unmarshal failed for contents: " + errunmarshal.Error()}
		io.WriteString(w, response.String())
		return
	} else {
		//now we will create our task here: filter, allocator, (image server) ,scheduler
		var userRequest CreateRequest
		userRequest.Init()
		for thisflag, flagvalue := range res.Data {
			switch {
			case strings.EqualFold(thisflag, "i"):
				userRequest.ImageName = flagvalue
			case strings.EqualFold(thisflag, "t"):
				userRequest.TypeName = flagvalue
				if !strings.EqualFold(flagvalue, "mpi") && !strings.EqualFold(flagvalue, "single") {
					fmt.Fprintf(os.Stderr, `the type of the task is not supported yet. Only "single" and "mpi" is supported.`)
					response.Warnings = []string{`the type of the task is not supported yet. Only "single" and "mpi" is supported.`}
					io.WriteString(w, response.String())
					return
				}
			case strings.EqualFold(thisflag, "n"):
				userRequest.TaskName = flagvalue
			case strings.EqualFold(thisflag, "s"):
				userRequest.Stratergy = flagvalue
				if !strings.EqualFold(thisflag, "MEM") && !strings.EqualFold(thisflag, "COM") {
					response.Warnings = []string{`Only MEM and COM are valid for -s flag`}
					io.WriteString(w, response.String())
					return
				}
			case strings.EqualFold(thisflag, "c"):
				userRequest.TotalCpuNum, _ = strconv.Atoi(flagvalue)
			case strings.EqualFold(thisflag, "C"):
				userRequest.ContainerNumMax, _ = strconv.Atoi(flagvalue)
			case strings.EqualFold(thisflag, "l"):
				value, _ := strconv.ParseFloat(flagvalue, 32)
				userRequest.OverLoadMax = float32(value)
			case strings.EqualFold(thisflag, "f"):
				filename := flagvalue
				readerme, openerr := os.Open(filename)
				if openerr != nil {
					fmt.Fprintf(os.Stderr, "CreateHandler:%s", openerr)
					response.Warnings = []string{"CreateHandler" + openerr.Error()}
					io.WriteString(w, response.String())
					return
				}
				unmarshalerr := util.UnmarshalReader(readerme, &(userRequest.ResNode))
				if unmarshalerr != nil {
					response.Warnings = []string{unmarshalerr.Error()}
					io.WriteString(w, response.String())
				}
			default:
				fmt.Fprintf(os.Stderr, "CreateHandler: %s flag invalid", thisflag)
				response.Warnings = []string{"CreateHandler: " + thisflag + "flag invalid"}
				io.WriteString(w, response.String())
				return
			}
		}

		var err error
		endpoint := []string{"http://" + MasterConfig.EtcdNode.Ip + ":" + MasterConfig.EtcdNode.Port}
		err = UpdateEtcdForUse(endpoint, MasterConfig.EtcdNode.Key, true, true)
		if err != nil {
			if *FlagDebug {
				util.PrintErr("Failded to Etcd data!")
			}
			response.Warnings = []string{err.Error()}
			io.WriteString(w, response.String())
			return
		}
		//Debug
//.........这里部分代码省略.........

/*List the information about one or more tasks.
The information is not in detai */
func ListTaskHandler(w http.ResponseWriter, r *http.Request) {
	//the return value :response
	var response util.HttpResponse
	inputInfo, err := GetInfoFromRequest(&w, r)
	if err != nil {
		return
	}
	var userRequest PsRequest
	userRequest.Init()
	for thisFlag, flagValue := range inputInfo.Data {
		switch thisFlag {
		case "a":
			userRequest.All, _ = strconv.ParseBool(flagValue)
		case "l":
			userRequest.Latest, _ = strconv.ParseBool(flagValue)
		case "n":
			userRequest.Name = flagValue
		case "i":
			userRequest.Image = flagValue
		case "t":
			userRequest.Type = flagValue
		default:
			response.Set(util.INVALID_INPUT, "invalid flag of wharf ps.")
			io.WriteString(w, response.String())
			return
		}
	}
	//flags conflict
	if userRequest.All && userRequest.Latest ||
		userRequest.All && userRequest.Name != "" ||
		userRequest.Latest && userRequest.Name != "" {
		response.Set(util.INVALID_INPUT, "flags are conflicting with each other.")
		io.WriteString(w, response.String())
		return
	}

	var psIterm PsOutput
	//task name is set || latest is set
	if userRequest.Name != "" || userRequest.Latest {
		if userRequest.Latest {
			userRequest.Name = LatestTask
		}
		thisTask, exists := Tasks[userRequest.Name]
		if !exists {
			response.Set(util.INVALID_INPUT, `No such task with name of "`+userRequest.Name+`"`)
			io.WriteString(w, response.String())
			return
		} else {
			fillPSOutpusWithTask(&psIterm, &thisTask)
			data, _ := json.Marshal(psIterm)
			response.Set(util.OK, string(data))
			io.WriteString(w, response.String())
			return
		}
	}
	//traverse all the task in Tasks
	for _, thisTask := range Tasks {
		response.Status = util.OK
		if (userRequest.All == true || thisTask.Status == RUNNING) &&
			(userRequest.Image == "" || thisTask.Cmd.ImageName == userRequest.Image) &&
			(userRequest.Type == "" || thisTask.Cmd.TypeName == userRequest.Type) {
			fillPSOutpusWithTask(&psIterm, &thisTask)
			data, _ := json.Marshal(psIterm)
			response.Append(string(data))
		}
		io.WriteString(w, response.String())
	}
	return
}

/*insect the detail informatin about only one task.*/
func InspectTaskHandler(w http.ResponseWriter, r *http.Request) {
	var response util.HttpResponse
	inputInfo, err := GetInfoFromRequest(&w, r)
	if err != nil {
		return
	}

	var userRequest InspectReqest
	userRequest.Init()
	for thisFlag, flagValue := range inputInfo.Data {
		switch thisFlag {
		case "f":
			userRequest.Field = flagValue
		default:
			response.Set(util.INVALID_INPUT, "invalid flag of wharf inspect.")
			io.WriteString(w, response.String())
			return
		}
	}
	//must contain one taskname
	if len(inputInfo.Args) != 1 {
		response.Set(util.INVALID_INPUT, "You must provide one and only  taskname to insepect.")
		io.WriteString(w, response.String())
		return
	}
	thistask, exists := Tasks[inputInfo.Args[0]]
	if !exists {
		response.Set(util.INVALID_INPUT, "There is no task with the name of "+`'`+inputInfo.Args[0]+`'`)
		io.WriteString(w, response.String())
		return
	}

	var outputData InspectOutput
	outputData.GetData(thistask)
	if userRequest.Field == "" {
		var outputByte []byte
		outputByte, err = json.Marshal(outputData)
		if err != nil {
			response.Set(util.SERVER_ERROR, err.Error())
		} else {
			response.Set(util.OK, string(outputByte))
		}
	} else {
		immutable := reflect.ValueOf(outputData)
		fieldValue := userRequest.Field

		if fieldValue[0:1] == `.` {
			fieldValue = fieldValue[1:]
		}

		itermsStr := strings.Split(fieldValue, ".")

		for i := range itermsStr {
			immutable = immutable.FieldByName(itermsStr[i])
		}

		response.Set(util.OK, immutable.String())

	}
	io.WriteString(w, response.String())
	return
}

/*
function:
	stop some tasks.
param:
	flags , task1, task2...
return:

*/
func StopTaskHandler(w http.ResponseWriter, r *http.Request) {
	var response util.HttpResponse
	inputInfo, err := GetInfoFromRequest(&w, r)
	if err != nil {
		return
	}

	var userRequest StopRequest
	userRequest.Init(&inputInfo)
	for thisFlag, flagValue := range inputInfo.Data {
		switch thisFlag {
		case "h":
			userRequest.FsName["Help"], _ = strconv.ParseBool(flagValue)
		default:
			response.Set(util.INVALID_INPUT, "invalid flag of wharf stop.")
			io.WriteString(w, response.String())
			return
		}
	}

	thisTask, err := checkTaskName(&w, r, userRequest.Args)
	if err != nil {
		return
	}

	if thisTask.Status == DOWN {
		response.Set(util.OK, "The task is already stoped.")
		io.WriteString(w, response.String())
		return
	}

	var isStoped bool
	isStoped = true
	var outputData StopOutput
	for i := range thisTask.Set {
		unit := thisTask.Set[i]
		stop2delErr := StopContainer2deleteDev(unit)
		if stop2delErr != nil {
			isStoped = false
			outputData.Append(unit.ContainerDesc.Id, unit.HostIp, stop2delErr.Error())
		}
	}

	if isStoped == true {
		thisTask.Status = DOWN
		delete(Tasks, userRequest.Args[0])
		Tasks[userRequest.Args[0]] = thisTask

		outputData.Warning = "task " + userRequest.Args[0] + " is stopped."
		response.Set(util.OK, outputData.String())
		io.WriteString(w, response.String())
	} else {
		outputData.Warning = "Failed to stop " + userRequest.Args[0]
		response.Set(util.SERVER_ERROR, outputData.String())
		io.WriteString(w, response.String())
	}
	return
}

/*
function:
	Rm some tasks.
param:
	flags , task1, task2...
*/
func RmTaskHandler(w http.ResponseWriter, r *http.Request) {
	var response util.HttpResponse
	inputInfo, err := GetInfoFromRequest(&w, r)
	if err != nil {
		return
	}

	//flag parse
	var userRequest RmRequest
	userRequest.Init(&inputInfo)
	for thisFlag, flagValue := range inputInfo.Data {
		switch thisFlag {
		case "h":
			userRequest.FsName["Help"], _ = strconv.ParseBool(flagValue)
		default:
			response.Set(util.INVALID_INPUT, "invalid flag of wharf stop.")
			io.WriteString(w, response.String())
			return
		}
	}

	thisTask, err := checkTaskName(&w, r, userRequest.Args)
	if err != nil {
		return
	}

	if thisTask.Status != DOWN {
		response.Set(util.INVALID_INPUT, "The task is not stoped yet.")
		io.WriteString(w, response.String())
		return
	}

	var isRemoved bool
	isRemoved = true
	var outputData StopOutput
	for i := range thisTask.Set {
		unit := thisTask.Set[i]
		rmErr := removeContainerOnHost(unit.ContainerDesc.Id, unit.HostIp)
		if rmErr != nil {
			isRemoved = false
			outputData.Append(unit.ContainerDesc.Id, unit.HostIp, rmErr.Error())
		}
		clearTaskInGres(unit)
	}

	if isRemoved == true {
		delete(Tasks, userRequest.Args[0])

		outputData.Warning = "task " + userRequest.Args[0] + " is removed."
		response.Set(util.OK, outputData.String())
		io.WriteString(w, response.String())
	} else {
		outputData.Warning = "Failed to remove" + userRequest.Args[0]
		response.Set(util.SERVER_ERROR, outputData.String())
		io.WriteString(w, response.String())
	}
	return
}

/*
function:
	start some tasks.
param:
	flags , task1, task2...
*/
func StartTaskHandler(w http.ResponseWriter, r *http.Request) {
	var response util.HttpResponse
	inputInfo, err := GetInfoFromRequest(&w, r)
	if err != nil {
		return
	}

	var userRequest StartRequest
	userRequest.Init(&inputInfo)
	for thisFlag, flagValue := range inputInfo.Data {
		switch thisFlag {
		case "h":
			userRequest.FsName["Help"], _ = strconv.ParseBool(flagValue)
		default:
			response.Set("404-invalid input", "invalid flag of wharf start.")
			io.WriteString(w, response.String())
			return
		}
	}

	thisTask, err := checkTaskName(&w, r, userRequest.Args)
	if err != nil {
		return
	}
	if thisTask.Status == RUNNING {
		response.Set(util.OK, "The task is already started.")
		io.WriteString(w, response.String())
		return
	}

	//start each ct accoring to the task
	var isStarted bool
	isStarted = true
	var outputData StartOutput
	for i := range thisTask.Set {
		unit := thisTask.Set[i]
		start2delErr := startContainerWithIP(&(thisTask.Set[i])) //hostip , contaienr id, container ip
		if start2delErr != nil {
			isStarted = false
			outputData.Append(unit.ContainerDesc.Id, unit.HostIp, start2delErr.Error())
		}
	}

	if isStarted == true {
		thisTask.Status = RUNNING
		delete(Tasks, userRequest.Args[0])
		Tasks[userRequest.Args[0]] = thisTask

		outputData.Warning = "task " + userRequest.Args[0] + " is started."
		response.Set(util.OK, outputData.String())
		io.WriteString(w, response.String())
	} else {
		outputData.Warning = "Failed to start " + userRequest.Args[0]
		response.Set(util.SERVER_ERROR, outputData.String())
		io.WriteString(w, response.String())
	}
	return
}