package rest import ( "io" "reflect" "git.ali33.ru/fcg-xvii/go-tools/json" ) // конверторы // Int64ToBytes упаковывает int64 в срез байтов заданной длины func Int64ToBytes(num int64, byteCount int) []byte { bytes := make([]byte, byteCount) for i := 0; i < byteCount; i++ { shift := uint((byteCount - 1 - i) * 8) bytes[i] = byte(num >> shift) } return bytes } // BytesToInt64 конвертирует срез байтов в int64 func BytesToInt64(bytes []byte) int64 { var num int64 for _, b := range bytes { num = (num << 8) | int64(b) } return num } // ioError возвращает ошибку ввода выввода func ioError(field string, err error) IErrorArgs { return NewError( "ErrIO", json.Map{ "field": field, "error": err.Error(), }, ) } // чтение // ReadBuf считывает срез заданного колличества байт и возвращает его func ReadBuf(r io.Reader, size int, field string) ([]byte, IErrorArgs) { buf := make([]byte, size) if _, err := r.Read(buf); err != nil { return nil, ioError(field, err) } return buf, nil } // ReadBufSize считывает размер байтового среза, затем сам срез и возвращает его. // lenSize - количество байтов, которыми описывается размер основного среза (см. ReadInt64) func ReadBufSize(r io.Reader, lenSize int, field string) ([]byte, IErrorArgs) { var size int64 if err := ReadInt64(r, lenSize, field+"_size", &size); err != nil { return nil, err } buf, err := ReadBuf(r, int(size), field) return buf, err } // ReadInt64 Считывает срез заданного размера, который конвертируется в число типа int64. // size - размер среза (может быть от 1 до 8 байт). // Результат записывается в значение указателя result func ReadInt64(r io.Reader, size int, field string, result *int64) IErrorArgs { buf := make([]byte, size) if _, err := r.Read(buf); err != nil { return ioError(field, err) } *result = BytesToInt64(buf) return nil } // ReadString считывает строку (длина строки, затем строку), см. ReadBufSize func ReadString(r io.Reader, lenSize int, field string, result *string) IErrorArgs { strBuf, err := ReadBufSize(r, lenSize, field) if err != nil { return err } *result = string(strBuf) return nil } // ReadByte считывает 1 байт func ReadByte(r io.Reader, field string, result *byte) IErrorArgs { buf, err := ReadBuf(r, 1, field) if err != nil { return err } *result = buf[0] return nil } // запись // WriteBuf записывает срез байтов func WriteBuf(w io.Writer, buf []byte, field string) IErrorArgs { if _, err := w.Write(buf); err != nil { return ioError(field, err) } return nil } // WriteInt64 конвертирует число типа int64 в срез заданного размера и записывает его func WriteInt64(w io.Writer, val int64, size int, field string) IErrorArgs { buf := Int64ToBytes(val, size) return WriteBuf(w, buf, field) } // WriteString записывает строку в поток. // Сначала записывается размер строки (lenSize - количество байт в срезе размера). // Далее строка конвертируется в байтовый срез и записывается в поток func WriteString(w io.Writer, val, field string, lenSize int) IErrorArgs { // длина if err := WriteInt64(w, int64(len(val)), lenSize, field+"_size"); err != nil { return err } // строка if len(val) == 0 { return nil } return WriteBuf(w, []byte(val), field) } // WriteBuf записывает 1 байт в поток func WriteByte(w io.Writer, val byte, field string) IErrorArgs { return WriteBuf(w, []byte{val}, field) } // WriteBufSize записывает срез байтов в поток. // Сначала записывается размер среза (lenSize - количество байт в срезе размера). // Далее в поток записывается сам срез func WriteBufSize(w io.Writer, val []byte, lenSize int, field string) IErrorArgs { // длина if err := WriteInt64(w, int64(len(val)), lenSize, field+"_size"); err != nil { return err } // буфер return WriteBuf(w, val, field) } // ObjectFieldKeys позволяет рекурсивно получить список вложенных объектов запроса для предварительного формирования их в контексте ответа func ObjectFieldKeys(l []any) (res json.Map) { res = make(json.Map) t := reflect.TypeOf(res) for _, v := range l { rv := reflect.ValueOf(v) if reflect.ValueOf(v).Kind() == reflect.Map { if rv.CanConvert(t) { m := rv.Convert(t).Interface().(json.Map) if name := m.String("name", ""); len(name) > 0 { res[name] = ObjectFieldKeys(m.Slice("fields", []any{})) } } } } return }