simon987/task_tracker
1
0
Fork 0
mirror of https://github.com/simon987/task_tracker.git synced 2025-04-19 18:16:45 +00:00
Code Issues Projects Releases Wiki Activity

Make taskTrackerClient public

Browse Source
This commit is contained in:
simon987 2020-06-28 11:42:35 -04:00
parent ed8b153d3b
commit 961d692c61
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
client/client.go
View File

@ -11,7 +11,7 @@ import (
"strconv" "strconv"
) )
type taskTrackerClient struct { type TaskTrackerClient struct {
worker *Worker worker *Worker
httpClient http.Client httpClient http.Client
serverAddress string serverAddress string
@ -20,21 +20,21 @@ type taskTrackerClient struct {
workerIdStr string workerIdStr string
} }
func New(serverAddress string) *taskTrackerClient { func New(serverAddress string) *TaskTrackerClient {
client := new(taskTrackerClient) client := new(TaskTrackerClient)
client.serverAddress = serverAddress client.serverAddress = serverAddress
return client return client
} }
func (c *taskTrackerClient) SetWorker(worker *Worker) { func (c *TaskTrackerClient) SetWorker(worker *Worker) {
c.worker = worker c.worker = worker
c.secretB64 = base64.StdEncoding.EncodeToString(worker.Secret) c.secretB64 = base64.StdEncoding.EncodeToString(worker.Secret)
c.workerIdStr = strconv.FormatInt(worker.Id, 10) c.workerIdStr = strconv.FormatInt(worker.Id, 10)
} }
func (c *taskTrackerClient) get(path string) *http.Response { func (c *TaskTrackerClient) get(path string) *http.Response {
url := c.serverAddress + path url := c.serverAddress + path
req, err := http.NewRequest("GET", url, nil) req, err := http.NewRequest("GET", url, nil)
@ -50,7 +50,7 @@ func (c *taskTrackerClient) get(path string) *http.Response {
return r return r
} }
func (c *taskTrackerClient) post(path string, x interface{}) *http.Response { func (c *TaskTrackerClient) post(path string, x interface{}) *http.Response {
body, err := json.Marshal(x) body, err := json.Marshal(x)
buf := bytes.NewBuffer(body) buf := bytes.NewBuffer(body)
@ -81,7 +81,7 @@ func unmarshalResponse(r *http.Response, result interface{}) error {
return nil return nil
} }
func (c taskTrackerClient) MakeWorker(alias string) (*Worker, error) { func (c TaskTrackerClient) MakeWorker(alias string) (*Worker, error) {
httpResp := c.post("/worker/create", api.CreateWorkerRequest{ httpResp := c.post("/worker/create", api.CreateWorkerRequest{
Alias: alias, Alias: alias,
@ -101,7 +101,7 @@ func (c taskTrackerClient) MakeWorker(alias string) (*Worker, error) {
return nil, err return nil, err
} }
func (c taskTrackerClient) FetchTask(projectId int) (*AssignTaskResponse, error) { func (c TaskTrackerClient) FetchTask(projectId int) (*AssignTaskResponse, error) {
httpResp := c.get("/task/get/" + strconv.Itoa(projectId)) httpResp := c.get("/task/get/" + strconv.Itoa(projectId))
var jsonResp AssignTaskResponse var jsonResp AssignTaskResponse
@ -112,7 +112,7 @@ func (c taskTrackerClient) FetchTask(projectId int) (*AssignTaskResponse, error)
return &jsonResp, err return &jsonResp, err
} }
func (c taskTrackerClient) ReleaseTask(req api.ReleaseTaskRequest) (*ReleaseTaskResponse, error) { func (c TaskTrackerClient) ReleaseTask(req api.ReleaseTaskRequest) (*ReleaseTaskResponse, error) {
httpResp := c.post("/task/release", req) httpResp := c.post("/task/release", req)
var jsonResp ReleaseTaskResponse var jsonResp ReleaseTaskResponse
@ -121,7 +121,7 @@ func (c taskTrackerClient) ReleaseTask(req api.ReleaseTaskRequest) (*ReleaseTask
return &jsonResp, err return &jsonResp, err
} }
func (c taskTrackerClient) SubmitTask(req api.SubmitTaskRequest) (AssignTaskResponse, error) { func (c TaskTrackerClient) SubmitTask(req api.SubmitTaskRequest) (AssignTaskResponse, error) {
httpResp := c.post("/task/submit", req) httpResp := c.post("/task/submit", req)
var jsonResp AssignTaskResponse var jsonResp AssignTaskResponse
@ -132,7 +132,7 @@ func (c taskTrackerClient) SubmitTask(req api.SubmitTaskRequest) (AssignTaskResp
return jsonResp, err return jsonResp, err
} }
func (c taskTrackerClient) GetProjectSecret(projectId int) (string, error) { func (c TaskTrackerClient) GetProjectSecret(projectId int) (string, error) {
httpResp := c.get("/project/secret/" + strconv.Itoa(projectId)) httpResp := c.get("/project/secret/" + strconv.Itoa(projectId))
var jsonResp ProjectSecretResponse var jsonResp ProjectSecretResponse
@ -141,7 +141,7 @@ func (c taskTrackerClient) GetProjectSecret(projectId int) (string, error) {
return jsonResp.Content.Secret, err return jsonResp.Content.Secret, err
} }
func (c taskTrackerClient) GetProjectList() ([]storage.Project, error) { func (c TaskTrackerClient) GetProjectList() ([]storage.Project, error) {
httpResp := c.get("/project/list") httpResp := c.get("/project/list")
var jsonResp ProjectListResponse var jsonResp ProjectListResponse
@ -150,7 +150,7 @@ func (c taskTrackerClient) GetProjectList() ([]storage.Project, error) {
return jsonResp.Content.Projects, err return jsonResp.Content.Projects, err
} }
func (c taskTrackerClient) RequestAccess(req api.CreateWorkerAccessRequest) (api.JsonResponse, error) { func (c TaskTrackerClient) RequestAccess(req api.CreateWorkerAccessRequest) (api.JsonResponse, error) {
httpResp := c.post("/project/request_access", req) httpResp := c.post("/project/request_access", req)
var jsonResp api.JsonResponse var jsonResp api.JsonResponse