feat: super 包新增 TryWriteChannelByHandler 函数，支持尝试写入 channel，如果 channel…

… 无法写入则执行 handler
kercylan98 · Dec 29, 2023 · efbde3e · efbde3e
1 parent 80f38ff
commit efbde3e
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/utils/super/channel.go b/utils/super/channel.go
@@ -10,3 +10,13 @@ func TryWriteChannel[T any](ch chan<- T, data T) bool {
 		return false
 	}
 }
+
+// TryWriteChannelByHandler 尝试写入 channel，如果 channel 无法写入则执行 handler
+//   - 无法写入的情况包括：channel 已满、channel 已关闭
+func TryWriteChannelByHandler[T any](ch chan<- T, data T, handler func()) {
+	select {
+	case ch <- data:
+	default:
+		handler()
+	}
+}