| 33 | } |
| 34 | |
| 35 | func (e *encoder) Encode(messages <-chan message.Message, storage storage.Writer) error { |
| 36 | header := newHeader(CurrentVersion).getBytes() |
| 37 | if _, err := storage.Write(header); err != nil { |
| 38 | return err |
| 39 | } |
| 40 | |
| 41 | var gzipHandle *gzip.Writer |
| 42 | var encoder *cbor.Encoder |
| 43 | gzipHandle = gzip.NewWriter(storage) |
| 44 | encoder = cbor.NewEncoder(gzipHandle, cbor.EncOptions{}) |
| 45 | |
| 46 | startTime := int64(0) |
| 47 | var ip = "" |
| 48 | var country = "XX" |
| 49 | var username *string |
| 50 | for { |
| 51 | msg, ok := <-messages |
| 52 | if !ok { |
| 53 | break |
| 54 | } |
| 55 | if startTime == 0 { |
| 56 | startTime = msg.Timestamp |
| 57 | } |
| 58 | ip, country, username = e.storeMetadata(msg, storage, startTime, ip, country, username) |
| 59 | if err := encoder.Encode(&msg); err != nil { |
| 60 | return fmt.Errorf("failed to encode audit log message (%w)", err) |
| 61 | } |
| 62 | _ = gzipHandle.Flush() |
| 63 | if msg.MessageType == message.TypeDisconnect { |
| 64 | break |
| 65 | } |
| 66 | } |
| 67 | if err := gzipHandle.Flush(); err != nil { |
| 68 | return fmt.Errorf("failed to flush audit log gzip stream (%w)", err) |
| 69 | } |
| 70 | if err := storage.Close(); err != nil { |
| 71 | return fmt.Errorf("failed to close audit log gzip stream (%w)", err) |
| 72 | } |
| 73 | return nil |
| 74 | } |
| 75 | |
| 76 | func (e *encoder) storeMetadata( |
| 77 | msg message.Message, |