Golang go-http-auth.AuthenticatedRequest类代码示例

func ServerActionsHandler(w http.ResponseWriter, r *auth.AuthenticatedRequest) {
	r.ParseForm()
	clientToken, ok := r.PostForm["token"]

	if ok == false {
		fmt.Fprint(w, "{\"error\": \"token wasn't specified\"}")
		return
	}

	if clientToken[0] != serverToken {
		fmt.Fprint(w, "{\"error\": \"wrong token\"}")
		return
	}

	clientAction, ok := r.PostForm["action"]

	if ok == false {
		fmt.Fprint(w, "{\"error\": \"action wasn't specified\"}")
		return
	}

	params := make(map[string]string)
	params["char_id"] = r.PostFormValue("params[char_id]")
	log.Println(params)
	processClientAction(clientAction[0], w, params)
}

// 编辑任务
func editTask(w http.ResponseWriter, r *auth.AuthenticatedRequest) {
	success := "true"
	msg := "更新成功！"

	servers := r.FormValue("servers")
	backupServers := r.FormValue("backup_servers")
	comment := r.FormValue("comment")
	logornot := r.FormValue("logornot")
	id := r.FormValue("id")
	now := time.Now().Format("2006-01-02 15:04:05")
	logOrNot, _ := strconv.Atoi(logornot)
	taskId, _ := strconv.Atoi(id)
	err := db.UpdateTaskInfo(servers, backupServers, comment, logOrNot, now, taskId)
	if err != nil {
		logger.Println(err)
		success = "false"
		msg = "数据存储操作出现问题!"
	}
	result := statusResult{
		Success: success,
		Msg:     msg,
	}
	rt, err := json.Marshal(result)
	if err != nil {
		fmt.Println(err)
	}
	fmt.Fprintf(w, string(rt))
	return
}

func ServerStatusHandler(w http.ResponseWriter, r *auth.AuthenticatedRequest) {
	r.ParseForm()

	clientToken, ok := r.PostForm["token"]

	if ok == false {
		fmt.Fprint(w, "{\"error\": \"token wasn't specified\"}")
		return
	}

	if clientToken[0] != serverToken {
		fmt.Fprint(w, "{\"error\": \"wrong token\"}")
		return
	}

	clientTopicVersion, ok := r.PostForm["topic_version"]
	clientTopicVersionInt, err := strconv.Atoi(clientTopicVersion[0])

	if err != nil {
		clientTopicVersionInt = 0
	}

	if ok && clientTopicVersionInt != topicVersion {
		response := getStatusResponse(true)
		fmt.Fprint(w, response)
		return
	}

	timeout := make(chan bool)
	responseChan := make(chan string)
	pollIndex := len(statusPolls)
	statusPolls[pollIndex] = responseChan

	go func() {
		time.Sleep(time.Second * 35)
		timeout <- true
	}()

	select {
	case responseBody := <-responseChan:
		fmt.Fprint(w, responseBody)
		return
	case <-timeout:
		delete(statusPolls, pollIndex)
		w.WriteHeader(503)
		return
	}
}

// 应用新HAProxy配置文件
func applyConf(w http.ResponseWriter, r *auth.AuthenticatedRequest) {
	success := "true"
	msg := "成功应用！"

	target := r.FormValue("target")
	rebuildHAProxyConf()
	if target == "master" {
		bytes, err := ioutil.ReadFile(appConf.NewHAProxyConfPath)
		masterConf, err := os.OpenFile(appConf.MasterConf, os.O_CREATE|os.O_RDWR|os.O_TRUNC, 0666)
		defer masterConf.Close()
		masterConf.Write(bytes)
		cmd := appConf.MasterRestartScript
		cmdToRun := exec.Command(cmd)
		err = cmdToRun.Run()
		if err != nil {
			logger.Println(err)
			success = "false"
			msg = fmt.Sprintf("应用失败！%s", err.Error())
		}
	} else {
		var err error
		if appConf.CopyMethod == 1 {
			err = sshoperation.ScpHaproxyConf(appConf)
		}
		if appConf.CopyMethod == 2 {
			err = sshoperation.ConfidentialScpHAProxyConf(appConf)
		}
		if err != nil {
			success = "false"
			msg = fmt.Sprintf("应用失败！%s", err.Error())
		}
	}
	result := statusResult{
		Success: success,
		Msg:     msg,
	}
	rt, err := json.Marshal(result)
	if err != nil {
		fmt.Println(err)
	}
	fmt.Fprintf(w, string(rt))
	return
}

// 删除任务
func delListenTask(w http.ResponseWriter, r *auth.AuthenticatedRequest) {
	success := "true"
	msg := "已成功删除"

	taskId := r.FormValue("taskid")
	id, _ := strconv.Atoi(taskId)
	result, err := db.DeleteTask(id)
	if err != nil {
		logger.Fatalln(err)
		success = "false"
		msg = "删除数据出错！"
	}
	rowsAffected, err := result.RowsAffected()
	if rowsAffected != 1 {
		success = "false"
		msg = fmt.Sprintf("数据删除有问题，删除了%d条", rowsAffected)
	}
	rt, _ := json.Marshal(statusResult{Success: success, Msg: msg})
	fmt.Fprintf(w, string(rt))
	return
}

// 申请虚拟ip端口请求处理函数
func applyVPort(w http.ResponseWriter, r *auth.AuthenticatedRequest) {
	autoOrNot, _ := strconv.Atoi(r.FormValue("autoornot"))
	business := -1
	var port int
	if autoOrNot == 1 && appConf.BusinessList != "" {
		business, _ = strconv.Atoi(r.FormValue("business"))
	}
	if autoOrNot == 0 {
		port, _ = strconv.Atoi(r.FormValue("port"))
	}
	servers := r.FormValue("servers")
	backupServers := r.FormValue("backup_servers")
	comment := strings.TrimSpace(r.FormValue("comment"))
	logOrNot := r.FormValue("logornot")

	rows, err := db.QueryVPort()
	if err != nil {
		logger.Println(err)
	}
	rowNum := len(rows)

	vportToAssign := -1

	dupOrNot := 0
	noAvailablePort := false

	// 若是指定端口方式，则判断指定的端口是否在1000-99999范围内
	inPortRange := true
	if autoOrNot == 0 && port < 1000 || port > 99999 {
		inPortRange = false
	}

	if inPortRange == true {
		/*
			将1000到已分配的最大端口号之间所有未占用和已占用的端口映射到一个真假值数组
		*/
		var portSlots []bool
		var assignedBiggest int
		if rowNum > 0 {
			assignedBiggest = rows[rowNum-1]
			allowedSmallest := 1000
			mayAssignedPortRange := assignedBiggest - allowedSmallest + 1
			//fmt.Println(mayAssignedPortRange)
			portSlots = make([]bool, mayAssignedPortRange)
			/*
			* 注意上一句make的用法-容量和长度都为mayAssignedPortRange,
			* 并且所有bool类型元素都自动初始化为false，所以下面的几行初始化代码不再需要。
			 */
			/*
				fmt.Println(portSlots[0])
				fmt.Println(portSlots[1])
				fmt.Println(portSlots[mayAssignedPortRange-1])
				for index := 0; index < mayAssignedPortRange; index++ {
					portSlots[index] = false
				}
			*/
			for index := 0; index < rowNum; index++ {
				port := rows[index]
				portSlots[port-1000] = true
			}
		}
		// 自动分配端口
		if autoOrNot == 1 {
			// 未指定业务
			if business == -1 {
				/*
					虚拟ip端口自动分配算法(不指定业务)
					可分配端口范围：10000 - 19999
				*/
				if rowNum == 0 {
					vportToAssign = 10000
				} else {
					vportToAssign, noAvailablePort = autoAssignPort(10000, 19999, assignedBiggest, portSlots)
				}
			} else {
				// 指定业务
				var portRange string
				businesses := strings.Split(appConf.BusinessList, ";")
				for index, bToPortRange := range businesses {
					if index == business {
						thisBToPortRange := strings.Split(bToPortRange, ",")
						if comment != "" {
							comment += "<br />"
						}
						comment += "业务：" + thisBToPortRange[0]
						portRange = thisBToPortRange[1]
						break
					}
				}
				firstAndLast := strings.Split(portRange, "-")
				firstPort, _ := strconv.Atoi(firstAndLast[0])
				lastPort, _ := strconv.Atoi(firstAndLast[1])
				if rowNum == 0 {
					vportToAssign = firstPort
				} else {
					vportToAssign, noAvailablePort = autoAssignPort(firstPort, lastPort, assignedBiggest, portSlots)
				}
			}
		} else {
//.........这里部分代码省略.........

func CaseHandler(w http.ResponseWriter, r *auth.AuthenticatedRequest) {
	var err error
	path := r.URL.Path
	pattern, err := regexp.Compile("/case/([^/]*)")
	if err != nil {
		panic(err)
	}
	matches := pattern.FindStringSubmatch(path)
	if len(matches) != 2 {
		resourceNotFound(w, &r.Request)
		return
	}
	id := matches[1]

	switch r.Method {
	case "GET":
		baileyCase := record.FetchRecord(id)
		if baileyCase == nil {
			resourceNotFound(w, &r.Request)
			return
		}
		responseBody := mustache.RenderFile("view/case.html", baileyCase)
		w.Write([]byte(responseBody))

	case "POST":
		baileyCase := record.FetchRecord(id)
		if baileyCase == nil {
			resourceNotFound(w, &r.Request)
			return
		}
		var jsonUpdate JsonUpdate
		err = json.Unmarshal([]byte(r.FormValue("json")), &jsonUpdate)
		if err != nil {
			log.Println(err)
			badRequest(w, &r.Request)
			return
		}

		// this is really ugly code, and should be fixed asap
		baileyCase.OfInterest = jsonUpdate.OfInterest
		baileyCase.NotOfInterest = jsonUpdate.NotOfInterest
		baileyCase.Clothing = jsonUpdate.Clothing
		baileyCase.ClothingCount = jsonUpdate.ClothingCount
		baileyCase.RawTextiles = jsonUpdate.RawTextiles
		baileyCase.RawTextilesCount = jsonUpdate.RawTextilesCount
		baileyCase.OtherTextiles = jsonUpdate.OtherTextiles
		baileyCase.Accessories = jsonUpdate.Accessories
		baileyCase.AccessoriesCount = jsonUpdate.AccessoriesCount
		baileyCase.HouseholdLinen = jsonUpdate.HouseholdLinen
		baileyCase.HouseholdLinenCount = jsonUpdate.HouseholdLinenCount
		baileyCase.Other = jsonUpdate.Other
		baileyCase.OtherCount = jsonUpdate.OtherCount
		baileyCase.OtherNotSpecified = jsonUpdate.OtherNotSpecified

		baileyCase.Save()
		successNoContent(w, &r.Request)

	default:
		methodNotAllowed(w, &r.Request)
	}
}

func insertHandler(w http.ResponseWriter, r *auth.AuthenticatedRequest) {
	file, header, err := r.FormFile("file")
	if err != nil {
		log.Error("Invalid form data %v:%s", r.Header, err.Error())
		fmt.Fprintln(w, err)
		return
	}
	log.Debug("Insert handler started for file %s", header.Filename)
	tag := r.Request.Header.Get("File-Tag")
	if len(tag) == 0 {
		log.Error("No file-tag header")
		fmt.Fprintf(w, "No file-tag header")
		return
	}

	bytes, err := base64.StdEncoding.DecodeString(header.Filename)
	if err != nil {
		log.Error("Filename %s is not valid base64 string", header.Filename)
		fmt.Fprintf(w, "Filename is not valid base64 string")
		return
	}
	filename := string(bytes)

	id, err := RegisterFile(filename, tag, time.Now())
	if err != nil {
		log.Error("Unable to register file %s from %s in DB:%s", filename, r.Host, err.Error())
		fmt.Fprintf(w, "Unable to register file")
		return
	}

	path, dir, err := CreateFilePath(id)

	if err != nil {
		log.Error("Unable to create file path for %s from %s in DB:%s", filename, r.Host, err.Error())
		fmt.Fprintf(w, "Unable to create file path")
		return
	}

	out, err := os.Create(path)
	if err != nil {
		log.Error("Unable to save file %s from %s in DB:%s", filename, r.Host, err.Error())
		fmt.Fprintf(w, "Unable to save file")
		return
	}

	_, err = io.Copy(out, file)
	if err != nil {
		fmt.Fprintln(w, err)
	}

	out.Close()
	file.Close()
	err = SaveMetaData(id, filename, tag, dir)
	if err != nil {
		log.Error("can't save meta for %d:%s", id, err.Error())
	}
	err = ConfirmFile(id)
	if err != nil {
		log.Error("Unable to commit file %s from %s in DB:%s", filename, r.Host, err.Error())
		fmt.Fprintf(w, "Unable to commit file")
		return
	}
	w.Header().Add("File-Id", strconv.FormatInt(id, 10))
	fmt.Fprintf(w, "%d", id)
	log.Info("File %s from %s saved succesfully", filename, r.Host)
}