Exemple #1
0
func doAndStop(sg stopgroup.StopGroup, f func()) {
	defer func() {
		v := recover()
		err, ok := v.(error)
		if ok {
			sg.Stop(err)
		} else {
			sg.Stop(nil)
		}
	}()
	f()
}
Exemple #2
0
func Manage(upstream msg.MessageChans, sg stopgroup.StopGroup, ds Durations) (downstream msg.MessageChans) {
	downstream, myMsgs := msg.MessageChansPair()
	go func() {
		defer close(myMsgs.W)
		onStop := sg.NotifyOnStop()
		for {
			select {
			case <-onStop:
				return
			case <-time.After(ds.ReadTimeout):
				sg.Stop(ErrReadTimeout)
				return
			case msg, ok := <-upstream.R:
				if !ok {
					return
				}
				myMsgs.W <- msg
			}
		}
	}()
	go func() {
		defer close(upstream.W)
		onStop := sg.NotifyOnStop()
		for msg := range myMsgs.R {
			select {
			case <-onStop:
				return
			case <-time.After(ds.WriteTimeout):
				sg.Stop(ErrWriteTimeout)
				return
			case upstream.W <- msg:
			}
		}
	}()
	return
}