mirror of
https://github.com/hibiken/asynq.git
synced 2024-09-20 02:55:54 +08:00
116 lines
2.3 KiB
Go
116 lines
2.3 KiB
Go
// Copyright 2020 Kentaro Hibino. All rights reserved.
|
|
// Use of this source code is governed by a MIT license
|
|
// that can be found in the LICENSE file.
|
|
|
|
package asynq_test
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"github.com/hibiken/asynq"
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
func ExampleServer_Run() {
|
|
srv := asynq.NewServer(
|
|
asynq.RedisClientOpt{Addr: ":6379"},
|
|
asynq.Config{Concurrency: 20},
|
|
)
|
|
|
|
h := asynq.NewServeMux()
|
|
// ... Register handlers
|
|
|
|
// Run blocks and waits for os signal to terminate the program.
|
|
if err := srv.Run(h); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func ExampleServer_Stop() {
|
|
srv := asynq.NewServer(
|
|
asynq.RedisClientOpt{Addr: ":6379"},
|
|
asynq.Config{Concurrency: 20},
|
|
)
|
|
|
|
h := asynq.NewServeMux()
|
|
// ... Register handlers
|
|
|
|
if err := srv.Start(h); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
sigs := make(chan os.Signal, 1)
|
|
signal.Notify(sigs, unix.SIGTERM, unix.SIGINT)
|
|
<-sigs // wait for termination signal
|
|
|
|
srv.Stop()
|
|
}
|
|
|
|
func ExampleServer_Quiet() {
|
|
srv := asynq.NewServer(
|
|
asynq.RedisClientOpt{Addr: ":6379"},
|
|
asynq.Config{Concurrency: 20},
|
|
)
|
|
|
|
h := asynq.NewServeMux()
|
|
// ... Register handlers
|
|
|
|
if err := srv.Start(h); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
sigs := make(chan os.Signal, 1)
|
|
signal.Notify(sigs, unix.SIGTERM, unix.SIGINT, unix.SIGTSTP)
|
|
// Handle SIGTERM, SIGINT to exit the program.
|
|
// Handle SIGTSTP to stop processing new tasks.
|
|
for {
|
|
s := <-sigs
|
|
if s == unix.SIGTSTP {
|
|
srv.Quiet() // stop processing new tasks
|
|
continue
|
|
}
|
|
break
|
|
}
|
|
|
|
srv.Stop()
|
|
}
|
|
|
|
func ExampleScheduler() {
|
|
scheduler := asynq.NewScheduler(
|
|
asynq.RedisClientOpt{Addr: ":6379"},
|
|
&asynq.SchedulerOpts{Location: time.Local},
|
|
)
|
|
|
|
if _, err := scheduler.Register("* * * * *", asynq.NewTask("task1", nil)); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if _, err := scheduler.Register("@every 30s", asynq.NewTask("task2", nil)); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// Run blocks and waits for os signal to terminate the program.
|
|
if err := scheduler.Run(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func ExampleParseRedisURI() {
|
|
rconn, err := asynq.ParseRedisURI("redis://localhost:6379/10")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
r, ok := rconn.(asynq.RedisClientOpt)
|
|
if !ok {
|
|
log.Fatal("unexpected type")
|
|
}
|
|
fmt.Println(r.Addr)
|
|
fmt.Println(r.DB)
|
|
// Output:
|
|
// localhost:6379
|
|
// 10
|
|
}
|