| | |
| | | |
| | | import ( |
| | | "context" |
| | | "os" |
| | | "sync" |
| | | "time" |
| | | |
| | |
| | | |
| | | "basic.com/valib/gogpu.git" |
| | | ) |
| | | |
| | | const maxTryBeforeReboot = 10 |
| | | |
| | | type face struct { |
| | | handle *SDKFace |
| | |
| | | ipc2Rule string |
| | | ruleMsgMaxCacheSize int |
| | | reserved map[string]interface{} |
| | | |
| | | running bool |
| | | rebootUntil int |
| | | mtxRunning sync.Mutex |
| | | } |
| | | |
| | | func (f *face) maybeReboot(ctx context.Context) { |
| | | for { |
| | | select { |
| | | case <-ctx.Done(): |
| | | return |
| | | default: |
| | | f.mtxRunning.Lock() |
| | | running := f.running |
| | | f.mtxRunning.Unlock() |
| | | |
| | | if running { |
| | | f.rebootUntil = 0 |
| | | |
| | | f.mtxRunning.Lock() |
| | | f.running = false |
| | | f.mtxRunning.Unlock() |
| | | |
| | | } else { |
| | | f.rebootUntil++ |
| | | f.fnLogger("Face No Running: ", f.rebootUntil) |
| | | if f.rebootUntil > maxTryBeforeReboot { |
| | | f.fnLogger("Face Too Long Running, Reboot") |
| | | os.Exit(0) |
| | | } |
| | | } |
| | | time.Sleep(time.Second) |
| | | } |
| | | } |
| | | } |
| | | |
| | | // Create create sdk |
| | |
| | | ipc2Rule: ipc2Rule, |
| | | ruleMsgMaxCacheSize: ruleMaxSize, |
| | | reserved: reserved, |
| | | |
| | | running: true, |
| | | rebootUntil: maxTryBeforeReboot, |
| | | } |
| | | } |
| | | |
| | |
| | | go torule.Run(ctx) |
| | | |
| | | go s.run(ctx, chRcv, chSnd) |
| | | |
| | | go s.maybeReboot(ctx) |
| | | } |
| | | |
| | | ////////////////////////////////////////////////////////////////// |
| | |
| | | count, data, _ := f.handle.Run(i.Data, imgW, imgH, 3, dtchn) |
| | | |
| | | common.EjectResult(data, rMsg, out) |
| | | f.mtxRunning.Lock() |
| | | f.running = true |
| | | f.mtxRunning.Unlock() |
| | | |
| | | var id, name string |
| | | if rMsg.Msg.Tasklab != nil { |