SagerNet/sing
1
0
Fork 0
mirror of https://github.com/SagerNet/sing.git synced 2025-04-03 11:57:39 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add Cast[Reader/Writer] interface

Browse source
This commit is contained in:
世界 2023-05-19 10:54:17 +08:00
parent be60138936
commit 49166ac427
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
1 changed files with 64 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

64
common/network/conn.go
View file

@ -135,6 +135,22 @@ func UnwrapReader(reader io.Reader) io.Reader {
return reader
}
func CastReader[T io.Reader](reader io.Reader) (T, bool) {
if c, ok := reader.(T); ok {
return c, true
}
if u, ok := reader.(ReaderWithUpstream); !ok || !u.ReaderReplaceable() {
return common.DefaultValue[T](), false
}
if u, ok := reader.(WithUpstreamReader); ok {
return CastReader[T](u.UpstreamReader().(io.Reader))
}
if u, ok := reader.(common.WithUpstream); ok {
return CastReader[T](u.Upstream().(io.Reader))
}
return common.DefaultValue[T](), false
}
func UnwrapPacketReader(reader PacketReader) PacketReader {
if u, ok := reader.(ReaderWithUpstream); !ok || !u.ReaderReplaceable() {
return reader
@ -148,6 +164,22 @@ func UnwrapPacketReader(reader PacketReader) PacketReader {
return reader
}
func CastPacketReader[T PacketReader](reader PacketReader) (T, bool) {
if c, ok := reader.(T); ok {
return c, true
}
if u, ok := reader.(ReaderWithUpstream); !ok || !u.ReaderReplaceable() {
return common.DefaultValue[T](), false
}
if u, ok := reader.(WithUpstreamReader); ok {
return CastPacketReader[T](u.UpstreamReader().(PacketReader))
}
if u, ok := reader.(common.WithUpstream); ok {
return CastPacketReader[T](u.Upstream().(PacketReader))
}
return common.DefaultValue[T](), false
}
func UnwrapWriter(writer io.Writer) io.Writer {
if u, ok := writer.(WriterWithUpstream); !ok || !u.WriterReplaceable() {
return writer
@ -161,6 +193,22 @@ func UnwrapWriter(writer io.Writer) io.Writer {
return writer
}
func CastWriter[T io.Writer](writer io.Writer) (T, bool) {
if c, ok := writer.(T); ok {
return c, true
}
if u, ok := writer.(WriterWithUpstream); !ok || !u.WriterReplaceable() {
return common.DefaultValue[T](), false
}
if u, ok := writer.(WithUpstreamWriter); ok {
return CastWriter[T](u.UpstreamWriter().(io.Writer))
}
if u, ok := writer.(common.WithUpstream); ok {
return CastWriter[T](u.Upstream().(io.Writer))
}
return common.DefaultValue[T](), false
}
func UnwrapPacketWriter(writer PacketWriter) PacketWriter {
if u, ok := writer.(WriterWithUpstream); !ok || !u.WriterReplaceable() {
return writer
@ -173,3 +221,19 @@ func UnwrapPacketWriter(writer PacketWriter) PacketWriter {
}
return writer
}
func CastPacketWriter[T PacketWriter](writer PacketWriter) (T, bool) {
if c, ok := writer.(T); ok {
return c, true
}
if u, ok := writer.(WriterWithUpstream); !ok || !u.WriterReplaceable() {
return common.DefaultValue[T](), false
}
if u, ok := writer.(WithUpstreamWriter); ok {
return CastPacketWriter[T](u.UpstreamWriter().(PacketWriter))
}
if u, ok := writer.(common.WithUpstream); ok {
return CastPacketWriter[T](u.Upstream().(PacketWriter))
}
return common.DefaultValue[T](), false
}