forked from vapor/postgres-nio
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathPostgresMessageEncoder.swift
42 lines (34 loc) · 1.28 KB
/
PostgresMessageEncoder.swift
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import NIOCore
import Logging
@available(*, deprecated, message: "Will be removed from public API")
public final class PostgresMessageEncoder: MessageToByteEncoder {
/// See `MessageToByteEncoder`.
public typealias OutboundIn = PostgresMessage
/// Logger to send debug messages to.
let logger: Logger?
/// Creates a new `PostgresMessageEncoder`.
public init(logger: Logger? = nil) {
self.logger = logger
}
/// See `MessageToByteEncoder`.
public func encode(data message: PostgresMessage, out: inout ByteBuffer) throws {
// serialize identifier
var message = message
switch message.identifier {
case .none: break
default:
out.write(identifier: message.identifier)
}
// leave room for identifier and size
let messageSizeIndex = out.writerIndex
out.moveWriterIndex(forwardBy: 4)
// serialize the message data
out.writeBuffer(&message.data)
// set message size
out.setInteger(Int32(out.writerIndex - messageSizeIndex), at: messageSizeIndex)
self.logger?.trace("Encoded: PostgresMessage (\(message.identifier))")
}
}
protocol ByteBufferSerializable {
func serialize(into buffer: inout ByteBuffer)
}