add TaskPool

Author: HFO4

conf.yaml

@@ -0,0 +1,3 @@
+token: sadsadasdasdasd
+api: http://127.0.0.1/Queue
+taskNum: 4

main.go

@@ -0,0 +1,63 @@
+package main
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io/ioutil"
+	"log"
+
+	"./task"
+
+	"gopkg.in/yaml.v2"
+)
+
+type taskConfig struct {
+	TOKEN   string `yaml:"token"`
+	APIURL  string `yaml:"api"`
+	TASKNUM int    `yaml:"taskNum"`
+}
+
+func (c *taskConfig) getConf() (*taskConfig, error) {
+
+	yamlFile, err := ioutil.ReadFile("conf.yaml")
+	if err != nil {
+		log.Printf("[ERROR] yamlFile.Get err   #%v ", err)
+		return nil, errors.New("Cant not read config.yaml")
+	}
+	err = yaml.Unmarshal(yamlFile, c)
+	if err != nil {
+		return nil, errors.New("Cant not prase config.yaml")
+	}
+
+	return c, nil
+}
+
+func main() {
+
+	fmt.Println("Cloudreve Queue Go Version")
+	fmt.Println("Author: AaronLiu <abslant@foxmail.com>")
+
+	var config taskConfig
+	_, err := config.getConf()
+	if err == nil {
+		log.Printf("[INFO] Config information:  %v ", config)
+		api := task.ApiInfo{TOKEN: config.TOKEN, APIURL: config.APIURL}
+		basicInfo := api.GetBasicInfo()
+		if basicInfo != "" {
+			log.Printf("[INFO] Basic Info:  %v ", basicInfo)
+			var siteInfo map[string]string
+			err := json.Unmarshal([]byte(basicInfo), &siteInfo)
+			if err != nil {
+				log.Printf("[ERROR] Failed to decode basic infomation,  %v ", err.Error())
+			}
+			for {
+				taskListContent := api.GetTaskList(config.TASKNUM)
+				task.Init(taskListContent, api)
+				break
+			}
+		}
+
+	}
+
+}

task/CloudreveApi.go

@@ -0,0 +1,55 @@
+package task
+
+import (
+	"io/ioutil"
+	"log"
+	"net/http"
+	"strconv"
+)
+
+//ApiInfo Api请求配置
+type ApiInfo struct {
+	TOKEN    string
+	APIURL   string
+	BASEPATH string
+}
+
+//GetBasicInfo 获取目标站点基本信息
+func (apiInfo *ApiInfo) GetBasicInfo() string {
+	return apiInfo.apiGet("basicInfo")
+}
+
+//GetTaskList 获取待处理任务列表
+func (apiInfo *ApiInfo) GetTaskList(num int) string {
+	return apiInfo.apiGet("getList?num=" + strconv.Itoa(num))
+}
+
+//apiGet 发送GET请求
+func (apiInfo *ApiInfo) apiGet(controller string) string {
+	client := &http.Client{}
+	request, err := http.NewRequest("GET", apiInfo.APIURL+"/"+controller, nil)
+	if err != nil {
+		log.Printf("[ERROR] Failed to create GET requetst, #%v ", err)
+	}
+	request.Header.Set("Authorization", "Bearer "+apiInfo.TOKEN)
+	response, err := client.Do(request)
+	if err != nil {
+		log.Printf("[ERROR] Failed to send GET requetst, #%v ", err)
+		return ""
+	}
+	defer response.Body.Close()
+	if response.StatusCode == 200 {
+		r, err := ioutil.ReadAll(response.Body)
+		if err != nil {
+			log.Printf("[ERROR] Failed to get GET requetst body, #%v ", err)
+		}
+		return string(r)
+	} else if response.StatusCode == 403 {
+		log.Printf("[ERROR] Auth failed, please verify your token, HTTP ERROR %v ", response.StatusCode)
+		return ""
+	} else {
+		log.Printf("[ERROR] Failed to get respond, HTTP ERROR %v ", response.StatusCode)
+		return ""
+	}
+
+}

task/TaskPool.go

@@ -0,0 +1,29 @@
+package task
+
+import (
+	"encoding/json"
+	"fmt"
+	"log"
+)
+
+//SingleTaskInfo 单个任务
+type SingleTaskInfo struct {
+	ID       int         `json:"id"`
+	TaskName string      `json:"task_name"`
+	Attr     interface{} `json:"attr"`
+	TaskType string      `json:"task"`
+	Status   string      `json:"status"`
+	Addtime  string      `json:"addtime"`
+}
+
+//Init 初始化任务线程池
+func Init(taskListContent string, apiInfo ApiInfo) {
+	var taskStringList []SingleTaskInfo
+	err := json.Unmarshal([]byte(taskListContent), &taskStringList)
+	if err != nil {
+		log.Printf("[ERROR] Failed to decode basic infomation,  %v ", err.Error())
+	}
+	for _, v := range taskStringList {
+		fmt.Println(v)
+	}
+}