Browse Source

[dataextract] 提交go后台服务器相关代码

yanxin 7 months ago
parent
commit
5e76e36792

+ 7 - 0
FastExtract/go/common/serve.go

@@ -0,0 +1,7 @@
+package common
+
+const (
+	GetNewTask       string = "/GetNewTask"
+	GetTaskStatus    string = "/GetTaskStatus"
+	UpdateTaskStatus string = "/UpdateTaskStatus"
+)

+ 22 - 0
FastExtract/go/common/task.go

@@ -0,0 +1,22 @@
+package common
+
+type Task struct {
+	ID   int    `json:"id"`
+	Path string `json:"path"`
+}
+
+type TaskStatus struct {
+	ReadyTaskCount     int `json:"ready_task_count"`
+	PendingTaskCount   int `json:"pending_task_count"`
+	CompletedTaskCount int `json:"completed_task_count"`
+	FailedTaskCount    int `json:"failed_task_count"`
+}
+
+type StatusCode int
+
+const (
+	ReadyTaskStatus StatusCode = 0
+	PendingStatus   StatusCode = 1
+	CompletedStatus StatusCode = 2
+	FailedStatus    StatusCode = 3
+)

+ 11 - 0
FastExtract/go/config/db_config.go

@@ -0,0 +1,11 @@
+package config
+
+type DBConfig struct {
+	MySQLConfig struct {
+		Host     string `yaml:"host"`
+		Port     string `yaml:"port"`
+		User     string `yaml:"user"`
+		Password string `yaml:"password"`
+		Database string `yaml:"database"`
+	} `yaml:"mysql"`
+}

+ 71 - 0
FastExtract/go/controller/task_controller.go

@@ -0,0 +1,71 @@
+package controller
+
+import (
+	"Zythum/dao"
+	"encoding/json"
+	"github.com/gin-gonic/gin"
+	"log"
+	"net/http"
+	"strconv"
+)
+
+func writeResponse(ctx *gin.Context, data map[string]interface{}) {
+	ctx.JSON(http.StatusOK, data)
+}
+
+func NewTaskHandler(ctx *gin.Context) {
+	newTask, err := dao.GetNewTask()
+	if err != nil {
+		return
+	}
+	taskJson, err := json.Marshal(newTask)
+	ctx.Header("Content-Type", "application/json")
+	_, err = ctx.Writer.Write(taskJson)
+	if err != nil {
+		log.Println("Failed to write response:", err)
+	} else {
+		//log.Println("Get /GetNewTask")
+	}
+}
+
+func UpdateTaskHandler(ctx *gin.Context) {
+	idString := ctx.Query("id")
+	statusString := ctx.Query("status")
+
+	data := map[string]interface{}{
+		"update_status": 0,
+	}
+
+	id, err := strconv.Atoi(idString)
+	if err != nil {
+		writeResponse(ctx, data)
+		return
+	}
+
+	status, err1 := strconv.Atoi(statusString)
+	if err1 != nil {
+		writeResponse(ctx, data)
+		return
+	}
+
+	updateResult := dao.UpdateTaskStatus(id, status)
+	if updateResult != 0 {
+		writeResponse(ctx, data)
+		return
+	}
+
+	data["update_status"] = 1
+	writeResponse(ctx, data)
+}
+
+func GetTaskStatusHandler(ctx *gin.Context) {
+	taskStatus := dao.GetTaskStatus()
+	taskStatusJson, err := json.Marshal(taskStatus)
+	ctx.Header("Content-Type", "application/json")
+	_, err = ctx.Writer.Write(taskStatusJson)
+	if err != nil {
+		log.Println("Failed to write response:", err)
+	} else {
+		//log.Println("Get /GetTaskStatus")
+	}
+}

+ 41 - 0
FastExtract/go/dao/global_db.go

@@ -0,0 +1,41 @@
+package dao
+
+import (
+	"Zythum/config"
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/jmoiron/sqlx"
+	"gopkg.in/yaml.v3"
+	"os"
+)
+
+var (
+	db *sqlx.DB
+)
+
+func init() {
+	data, err := os.ReadFile("config/datasource.yaml")
+	if err != nil {
+		panic("Failed to read datasource.yaml, please check the file.")
+	}
+
+	var dbConfig config.DBConfig
+
+	err = yaml.Unmarshal(data, &dbConfig)
+	if err != nil {
+		panic("Failed to unmarshal datasource.yaml, please check the file.")
+	}
+
+	dsn := dbConfig.MySQLConfig.User + ":" +
+		dbConfig.MySQLConfig.Password + "@tcp(" +
+		dbConfig.MySQLConfig.Host + ":" +
+		dbConfig.MySQLConfig.Port + ")/" +
+		dbConfig.MySQLConfig.Database + "?charset=utf8mb4&parseTime=True"
+
+	db, err = sqlx.Open("mysql", dsn)
+	if err != nil {
+		panic("Failed to connect database, please check the configuration.")
+	}
+
+	db.SetMaxOpenConns(30)
+	db.SetMaxIdleConns(15)
+}

+ 99 - 0
FastExtract/go/dao/task_repository.go

@@ -0,0 +1,99 @@
+package dao
+
+import (
+	"Zythum/common"
+	"database/sql"
+	_ "github.com/go-sql-driver/mysql"
+	"log"
+)
+
+func GetNewTask() (common.Task, error) {
+	tx, err := db.Begin()
+	committed := false
+	if err != nil {
+		log.Println("Failed to begin transaction:", err)
+		return common.Task{ID: -1, Path: ""}, err
+	}
+	defer func(tx *sql.Tx) {
+		if !committed {
+			err := tx.Rollback()
+			if err != nil {
+				log.Println("Failed to rollback transaction:", err)
+			}
+		}
+	}(tx)
+
+	rows, err := tx.Query("SELECT id, path FROM financial_reports WHERE status = 0 ORDER BY id LIMIT 1 FOR UPDATE;")
+	if err != nil {
+		return common.Task{ID: -1, Path: ""}, err
+	}
+
+	var task common.Task
+	rows.Next()
+	err = rows.Scan(&task.ID, &task.Path)
+	if err != nil {
+		return common.Task{ID: -2, Path: ""}, err
+	}
+
+	err = rows.Close()
+	if err != nil {
+		log.Println("Failed to close rows:", err)
+	}
+
+	_, err = tx.Exec("UPDATE financial_reports SET status = 1 WHERE id = ?", task.ID)
+	if err != nil {
+		return common.Task{ID: -3, Path: ""}, err
+	}
+
+	if err := tx.Commit(); err != nil {
+		return common.Task{ID: -4, Path: ""}, err
+	} else {
+		committed = true
+	}
+
+	return task, nil
+}
+
+func UpdateTaskStatus(id int, status int) int {
+	_, err := db.Exec("UPDATE financial_reports SET status = ? WHERE id = ?;", status, id)
+	if err != nil {
+		log.Println("Database update error:", err)
+		return -1
+	}
+	return 0
+}
+
+func GetTaskStatus() common.TaskStatus {
+	var taskStatus common.TaskStatus
+	err := db.QueryRow(
+		"SELECT (SELECT COUNT(id) FROM financial_reports WHERE status = 0) AS ready_task_count, "+
+			"(SELECT COUNT(id) FROM financial_reports WHERE status = 1) AS pending_task_count, "+
+			"(SELECT COUNT(id) FROM financial_reports WHERE status = 2) AS completed_task_count, "+
+			"(SELECT COUNT(id) FROM financial_reports WHERE status = 3) AS failed_task_count;").Scan(&taskStatus.ReadyTaskCount, &taskStatus.PendingTaskCount, &taskStatus.CompletedTaskCount, &taskStatus.FailedTaskCount)
+
+	if err != nil {
+		log.Println("Database query error:", err)
+		return common.TaskStatus{}
+	}
+
+	return taskStatus
+}
+
+func getTasksByStatus(status int) {
+	rows, err := db.Query("SELECT id, path FROM financial_reports WHERE status = ?;", status)
+	if err != nil {
+		log.Println("Database query error:", err)
+		return
+	}
+
+	for rows.Next() {
+		var task common.Task
+		err := rows.Scan(&task.ID, &task.Path)
+		if err != nil {
+			log.Println("Database scan error:", err)
+			return
+		}
+		log.Println("Task:", task)
+	}
+	defer rows.Close()
+}

+ 37 - 0
FastExtract/go/go.mod

@@ -0,0 +1,37 @@
+module Zythum
+
+go 1.19
+
+require (
+	github.com/gin-gonic/gin v1.9.1
+	github.com/go-sql-driver/mysql v1.8.1
+	github.com/jmoiron/sqlx v1.4.0
+	gopkg.in/yaml.v3 v3.0.1
+)
+
+require (
+	filippo.io/edwards25519 v1.1.0 // indirect
+	github.com/bytedance/sonic v1.9.1 // indirect
+	github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 // indirect
+	github.com/gabriel-vasile/mimetype v1.4.2 // indirect
+	github.com/gin-contrib/sse v0.1.0 // indirect
+	github.com/go-playground/locales v0.14.1 // indirect
+	github.com/go-playground/universal-translator v0.18.1 // indirect
+	github.com/go-playground/validator/v10 v10.14.0 // indirect
+	github.com/goccy/go-json v0.10.2 // indirect
+	github.com/json-iterator/go v1.1.12 // indirect
+	github.com/klauspost/cpuid/v2 v2.2.4 // indirect
+	github.com/leodido/go-urn v1.2.4 // indirect
+	github.com/mattn/go-isatty v0.0.19 // indirect
+	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
+	github.com/modern-go/reflect2 v1.0.2 // indirect
+	github.com/pelletier/go-toml/v2 v2.0.8 // indirect
+	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
+	github.com/ugorji/go/codec v1.2.11 // indirect
+	golang.org/x/arch v0.3.0 // indirect
+	golang.org/x/crypto v0.9.0 // indirect
+	golang.org/x/net v0.10.0 // indirect
+	golang.org/x/sys v0.8.0 // indirect
+	golang.org/x/text v0.9.0 // indirect
+	google.golang.org/protobuf v1.30.0 // indirect
+)

+ 95 - 0
FastExtract/go/go.sum

@@ -0,0 +1,95 @@
+filippo.io/edwards25519 v1.1.0 h1:FNf4tywRC1HmFuKW5xopWpigGjJKiJSV0Cqo0cJWDaA=
+filippo.io/edwards25519 v1.1.0/go.mod h1:BxyFTGdWcka3PhytdK4V28tE5sGfRvvvRV7EaN4VDT4=
+github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM=
+github.com/bytedance/sonic v1.9.1 h1:6iJ6NqdoxCDr6mbY8h18oSO+cShGSMRGCEo7F2h0x8s=
+github.com/bytedance/sonic v1.9.1/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U=
+github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY=
+github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 h1:qSGYFH7+jGhDF8vLC+iwCD4WpbV1EBDSzWkJODFLams=
+github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/gabriel-vasile/mimetype v1.4.2 h1:w5qFW6JKBz9Y393Y4q372O9A7cUSequkh1Q7OhCmWKU=
+github.com/gabriel-vasile/mimetype v1.4.2/go.mod h1:zApsH/mKG4w07erKIaJPFiX0Tsq9BFQgN3qGY5GnNgA=
+github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
+github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
+github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg=
+github.com/gin-gonic/gin v1.9.1/go.mod h1:hPrL7YrpYKXt5YId3A/Tnip5kqbEAP+KLuI3SUcPTeU=
+github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
+github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
+github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
+github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
+github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
+github.com/go-playground/validator/v10 v10.14.0 h1:vgvQWe3XCz3gIeFDm/HnTIbj6UGmg/+t63MyGU2n5js=
+github.com/go-playground/validator/v10 v10.14.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU=
+github.com/go-sql-driver/mysql v1.8.1 h1:LedoTUt/eveggdHS9qUFC1EFSa8bU2+1pZjSRpvNJ1Y=
+github.com/go-sql-driver/mysql v1.8.1/go.mod h1:wEBSXgmK//2ZFJyE+qWnIsVGmvmEKlqwuVSjsCm7DZg=
+github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
+github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
+github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
+github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
+github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
+github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
+github.com/jmoiron/sqlx v1.4.0 h1:1PLqN7S1UYp5t4SrVVnt4nUVNemrDAtxlulVe+Qgm3o=
+github.com/jmoiron/sqlx v1.4.0/go.mod h1:ZrZ7UsYB/weZdl2Bxg6jCRO9c3YHl8r3ahlKmRT4JLY=
+github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
+github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
+github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
+github.com/klauspost/cpuid/v2 v2.2.4 h1:acbojRNwl3o09bUq+yDCtZFc1aiwaAAxtcn8YkZXnvk=
+github.com/klauspost/cpuid/v2 v2.2.4/go.mod h1:RVVoqg1df56z8g3pUjL/3lE5UfnlrJX8tyFgg4nqhuY=
+github.com/leodido/go-urn v1.2.4 h1:XlAE/cm/ms7TE/VMVoduSpNBoyc2dOxHs5MZSwAN63Q=
+github.com/leodido/go-urn v1.2.4/go.mod h1:7ZrI8mTSeBSHl/UaRyKQW1qZeMgak41ANeCNaVckg+4=
+github.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw=
+github.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=
+github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA=
+github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
+github.com/mattn/go-sqlite3 v1.14.22 h1:2gZY6PC6kBnID23Tichd1K+Z0oS6nE/XwU+Vz/5o4kU=
+github.com/mattn/go-sqlite3 v1.14.22/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
+github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
+github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
+github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
+github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
+github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
+github.com/pelletier/go-toml/v2 v2.0.8 h1:0ctb6s9mE31h0/lhu+J6OPmVeDxJn+kYnJc2jZR9tGQ=
+github.com/pelletier/go-toml/v2 v2.0.8/go.mod h1:vuYfssBdrU2XDZ9bYydBu6t+6a6PYNcZljzZR9VXg+4=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
+github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
+github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
+github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
+github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
+github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
+github.com/stretchr/testify v1.8.3 h1:RP3t2pwF7cMEbC1dqtB6poj3niw/9gnV4Cjg5oW5gtY=
+github.com/stretchr/testify v1.8.3/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
+github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
+github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
+github.com/ugorji/go/codec v1.2.11 h1:BMaWp1Bb6fHwEtbplGBGJ498wD+LKlNSl25MjdZY4dU=
+github.com/ugorji/go/codec v1.2.11/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
+golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
+golang.org/x/arch v0.3.0 h1:02VY4/ZcO/gBOH6PUaoiptASxtXU10jazRCP865E97k=
+golang.org/x/arch v0.3.0/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
+golang.org/x/crypto v0.9.0 h1:LF6fAI+IutBocDJ2OT0Q1g8plpYljMZ4+lty+dsqw3g=
+golang.org/x/crypto v0.9.0/go.mod h1:yrmDGqONDYtNj3tH8X9dzUun2m2lzPa9ngI6/RUPGR0=
+golang.org/x/net v0.10.0 h1:X2//UzNDwYmtCLn7To6G58Wr6f5ahEAQgKNzv9Y951M=
+golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
+golang.org/x/sys v0.0.0-20220704084225-05e143d24a9e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.8.0 h1:EBmGv8NaZBZTWvrbjNoL6HVt+IVy3QDQpJs7VRIw3tU=
+golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/text v0.9.0 h1:2sjJmO8cDvYveuX97RDLsxlyUxLl+GHoLxBiRdHllBE=
+golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
+golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
+golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
+google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng=
+google.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=

+ 12 - 0
FastExtract/go/main.go

@@ -0,0 +1,12 @@
+package main
+
+import "Zythum/router"
+
+func main() {
+	r := router.App()
+
+	err := r.Run()
+	if err != nil {
+		panic(err)
+	}
+}

+ 13 - 0
FastExtract/go/router/app.go

@@ -0,0 +1,13 @@
+package router
+
+import (
+	"Zythum/service"
+	"github.com/gin-gonic/gin"
+)
+
+func App() *gin.Engine {
+	r := gin.Default()
+	service.TaskBasicService(r)
+
+	return r
+}

+ 13 - 0
FastExtract/go/service/task_service.go

@@ -0,0 +1,13 @@
+package service
+
+import (
+	"Zythum/common"
+	"Zythum/controller"
+	"github.com/gin-gonic/gin"
+)
+
+func TaskBasicService(engine *gin.Engine) {
+	engine.GET(common.GetNewTask, controller.NewTaskHandler)
+	engine.GET(common.UpdateTaskStatus, controller.UpdateTaskHandler)
+	engine.GET(common.GetTaskStatus, controller.GetTaskStatusHandler)
+}