tsk-5: Fetch token (#9)
Closes #5 Reviewed-on: #9 Co-authored-by: phoenix <kundeng00@pm.me> Co-committed-by: phoenix <kundeng00@pm.me>
This commit is contained in:
+19
-11
@@ -2,32 +2,31 @@ package config
|
||||
|
||||
import (
|
||||
"context"
|
||||
"flag"
|
||||
"fmt"
|
||||
"log"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"git.kundeng.us/phoenix/textsender-models/tx0/token"
|
||||
|
||||
"git.kundeng.us/phoenix/catapult/internal/app"
|
||||
"git.kundeng.us/phoenix/catapult/internal/auth"
|
||||
)
|
||||
|
||||
const App_Name = "catapult"
|
||||
|
||||
func CheckVersionFlag() *bool {
|
||||
versionFlag := flag.Bool("version", false, "Print version information")
|
||||
flag.Parse()
|
||||
|
||||
return versionFlag
|
||||
}
|
||||
|
||||
type Service struct {
|
||||
wg sync.WaitGroup
|
||||
ctx context.Context
|
||||
cancel context.CancelFunc
|
||||
App *app.App
|
||||
token *token.Login
|
||||
}
|
||||
|
||||
func NewService() *Service {
|
||||
func NewService(application *app.App) *Service {
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
return &Service{
|
||||
ctx: ctx,
|
||||
cancel: cancel,
|
||||
App: application,
|
||||
}
|
||||
}
|
||||
|
||||
@@ -55,6 +54,15 @@ func (s *Service) worker1() {
|
||||
case <-ticker.C:
|
||||
// Do some work
|
||||
log.Println("Worker 1: Processing...")
|
||||
if s.token == nil {
|
||||
catapultAuth := auth.Auth{Application: s.App}
|
||||
if token, err := catapultAuth.GetToken(); err != nil {
|
||||
fmt.Println("Error:", err)
|
||||
} else {
|
||||
fmt.Println("Access token:", token.AccessToken)
|
||||
s.token = token
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user