From ad50218e9499d10f9bf5787e2be406bd0224098d Mon Sep 17 00:00:00 2001
From: tursom <tursom@foxmail.com>
Date: Mon, 20 Jul 2020 08:41:48 +0800
Subject: [PATCH] add AsyncChannelEnhance.kt

---
 .../cn/tursom/channel/AsyncChannelEnhance.kt  | 23 +++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 AsyncSocket/src/main/kotlin/cn/tursom/channel/AsyncChannelEnhance.kt

diff --git a/AsyncSocket/src/main/kotlin/cn/tursom/channel/AsyncChannelEnhance.kt b/AsyncSocket/src/main/kotlin/cn/tursom/channel/AsyncChannelEnhance.kt
new file mode 100644
index 0000000..23eb5ac
--- /dev/null
+++ b/AsyncSocket/src/main/kotlin/cn/tursom/channel/AsyncChannelEnhance.kt
@@ -0,0 +1,23 @@
+package cn.tursom.channel
+
+import cn.tursom.channel.enhance.ChannelReader
+import cn.tursom.channel.enhance.ChannelWriter
+import cn.tursom.channel.enhance.EnhanceChannel
+import cn.tursom.channel.enhance.impl.*
+import cn.tursom.core.buffer.ByteBuffer
+
+fun AsyncChannel.reader(): ChannelReader<ByteBuffer> = ChannelReaderImpl(this)
+fun AsyncChannel.writer(): ChannelWriter<ByteBuffer> = ChannelWriterImpl(this)
+fun AsyncChannel.enhance(): EnhanceChannel<ByteBuffer, ByteBuffer> = EnhanceChannel(this)
+
+fun ChannelReader<ByteBuffer>.lengthField(): ChannelReader<ByteBuffer> = LengthFieldBasedFrameReader(this)
+fun ChannelWriter<ByteBuffer>.lengthField(): ChannelWriter<ByteBuffer> = LengthFieldPrependWriter(this)
+fun EnhanceChannel<ByteBuffer, ByteBuffer>.lengthField(): EnhanceChannel<ByteBuffer, ByteBuffer> = LengthFieldChannel(this)
+
+fun ChannelReader<ByteBuffer>.string() = StringReader(this)
+fun ChannelWriter<ByteBuffer>.string() = StringWriter(this)
+fun EnhanceChannel<ByteBuffer, ByteBuffer>.string() = StringChannel(this)
+
+fun ChannelWriter<String>.stringObjectWriter(
+  toString: (obj: Any) -> String = { it.toString() }
+): StringObjectWriter = StringObjectWriter(this, toString)
\ No newline at end of file