test/intimate
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add: 优雅停机

Browse Source
This commit is contained in:
eson
2020-07-16 18:31:13 +08:00
parent fe25921f9a
commit 93efe36d0b
10 changed files with 226 additions and 255 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tasks/openrec/openrec_task1/.gitignore vendored
View File

@@ -1 +1,2 @@
openrec_task1
openrec_task1
log

15
tasks/openrec/openrec_task1/task_openrec.go
View File

@@ -4,7 +4,11 @@ import (
"database/sql"
"intimate"
"log"
"os"
"os/signal"
"strconv"
"sync/atomic"
"syscall"
"time"
"github.com/474420502/hunter"
@@ -42,7 +46,16 @@ type OpenrecRanking struct {
// Execute 执行方法
func (or *OpenrecRanking) Execute(cxt *hunter.TaskContext) {
for {
var loop int32 = 1
go func() {
signalchan := make(chan os.Signal)
signal.Notify(signalchan, syscall.SIGKILL, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGSTOP)
log.Println("accept stop command:", <-signalchan)
atomic.StoreInt32(&loop, 0)
}()
for atomic.LoadInt32(&loop) > 0 {
resp, err := cxt.Hunt()
if err != nil {

15
tasks/openrec/openrec_task2/task_openrec.go
View File

@@ -5,7 +5,11 @@ import (
"encoding/json"
"intimate"
"log"
"os"
"os/signal"
"strconv"
"sync/atomic"
"syscall"
"time"
"github.com/474420502/gcurl"
@@ -31,7 +35,16 @@ type OpenrecExtratorRanking struct {
// Execute 执行方法
func (oer *OpenrecExtratorRanking) Execute(cxt *hunter.TaskContext) {
for {
var loop int32 = 1
go func() {
signalchan := make(chan os.Signal)
signal.Notify(signalchan, syscall.SIGKILL, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGSTOP)
log.Println("accept stop command:", <-signalchan)
atomic.StoreInt32(&loop, 0)
}()
for atomic.LoadInt32(&loop) > 0 {
source, err := store.Pop(string(intimate.TTOpenrecUser))