mirror of
https://github.com/nabbar/golib.git
synced 2025-09-27 04:06:05 +08:00
9ca3530b1d
- replace mutex by atomic - fix minor bugs & optimize code Package Size: - add function 'mul' allowing do a multiplaction on the size value by a float64 given value (could use another size.float64) - add function 'div' allowing do a division of the size value by a float64 given value (could use another size.float64) - add function 'add' allowing do an addtion of the size value with a uint64 given value (could use another size.uint64) - add function 'sub' allowing do an substraction of the size value with a uint64 given value (could use another size.uint64) Package Encoding: - rework completly crypt package to new package encoding - part aes of old crypt become new subpackage encoding/aes - part hex of old crypt (associated with aes) become new standalone subpackage encoding/hexa - adding new subpackage mux to use multiplexer/de-multiplexer of io stream - mux is different of multireader / multiwriter in way as io send from mux in one channel will be received only in the coresponding channel of de-mux - mux is encoding subpackage becauxse based on hex + cbor marshalling / unmarshalling - all subpackage can be used directly of by io reader / writer Package Socket/server - rework server to use as possible direct io to con - rework server to optimize & fix managment (context / shutdown...) - rework server to stop connection befor stopping server - rework server to not load incoming and send it to handler by allowing handler to manager io input/output directly - server not more used buff internally (less memory consuption) - adding counter of opened connection - adding some testing who's checking server & client for each protocol Package Socket/client - rework client to allow connect to server before reading/writing on io stream - rework client to allow consum direct io stream instead of reading locally / call function - rework client to allow having less memory consumption by removing internal buffer - rework client allowing to use root io function (bufio, io copy, ...) Package Socket/multi - adding new subpackage multi to simplfy using of 1 reader / x writer - if no writer is define, using an extended io discard - allowing asynch instancing, populate writer and calling main writer - allowing calling writer / reader to expose the main io reader / writer Package Socket/delim: - adding new subpackage delim to simplify using of bufio reader with delimiter - using 1 writer / 1 reader - allowing root io function like readwritecloser, writeto, ... - use local buffer to prevent race if reader and writer are not is same process Package PProf: - fix missing \n in message Package HttpServer: - fix bug with init TLS & http2 Other: - bump dependencies
119 lines
2.7 KiB
Go
119 lines
2.7 KiB
Go
/*
|
|
* MIT License
|
|
*
|
|
* Copyright (c) 2023 Nicolas JUHEL
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in all
|
|
* copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
* SOFTWARE.
|
|
*
|
|
*
|
|
*/
|
|
|
|
package mux
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"errors"
|
|
"io"
|
|
|
|
libcbr "github.com/fxamacker/cbor/v2"
|
|
enchex "github.com/nabbar/golib/encoding/hexa"
|
|
)
|
|
|
|
type dmux struct {
|
|
r *bufio.Reader
|
|
d byte
|
|
m map[rune]io.Writer
|
|
}
|
|
|
|
// Read defines a Read method for the dmux type in Go.
|
|
// It reads data from dmux, handles error conditions, and performs data manipulation before writing the data to a specified channel.
|
|
func (o *dmux) Read(p []byte) (n int, err error) {
|
|
if o == nil {
|
|
return 0, ErrInvalidInstance
|
|
}
|
|
|
|
var (
|
|
b []byte
|
|
d = data{}
|
|
h = enchex.New()
|
|
)
|
|
|
|
b, err = o.r.ReadBytes(o.d)
|
|
b = bytes.TrimSuffix(b, []byte{o.d})
|
|
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
if len(b) < 1 {
|
|
return 0, nil
|
|
}
|
|
|
|
err = libcbr.Unmarshal(b, &d)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
if len(d.D) < 1 {
|
|
return 0, nil
|
|
} else if p, err = h.Decode(d.D); err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
if len(o.m) < 1 {
|
|
return 0, ErrInvalidChannel
|
|
} else if w, k := o.m[d.K]; !k {
|
|
return 0, ErrInvalidChannel
|
|
} else {
|
|
return w.Write(p)
|
|
}
|
|
}
|
|
|
|
// Copy defines a method Copy for a type dmux
|
|
// that continuously reads data into a buffer p in chunks of 32KB until an error occurs,
|
|
// and then returns that error except the error is io.EOF.
|
|
func (o *dmux) Copy() error {
|
|
var (
|
|
e error
|
|
p = make([]byte, 32*1024)
|
|
)
|
|
|
|
for e == nil {
|
|
clear(p)
|
|
_, e = o.Read(p)
|
|
}
|
|
|
|
if errors.Is(e, io.EOF) {
|
|
e = nil
|
|
}
|
|
|
|
return e
|
|
}
|
|
|
|
// NewChannel defines a method NewChannel for a type dmux,
|
|
// which adds a new channel to the dmux by mapping the given key to the provided writer in the dmux's map.
|
|
func (o *dmux) NewChannel(key rune, w io.Writer) {
|
|
if o == nil {
|
|
return
|
|
}
|
|
|
|
o.m[key] = w
|
|
}
|