refactor tcp worker

This commit is contained in:
Darien Raymond committed 2017-02-26 14:38:41 +01:00
1 parent 122461647a
commit d93ff628bc
10 files changed
+127 -238

No files matched your search

+6 -3
View File
@@ -19,9 +19,12 @@ func ContextWithStreamSettings(ctx context.Context, streamSettings *StreamConfig
return context.WithValue(ctx, streamSettingsKey, streamSettings)
}
func StreamSettingsFromContext(ctx context.Context) (*StreamConfig, bool) {
ss, ok := ctx.Value(streamSettingsKey).(*StreamConfig)
return ss, ok
func StreamSettingsFromContext(ctx context.Context) *StreamConfig {
ss := ctx.Value(streamSettingsKey)
if ss == nil {
return nil
}
return ss.(*StreamConfig)
}
func ContextWithDialerSource(ctx context.Context, addr net.Address) context.Context {