123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120 |
- import NIO
- import Foundation
- /// Performant method for generating RFC1123 date headers.
- internal final class RFC1123DateCache {
- static func on(_ eventLoop: EventLoop) -> RFC1123DateCache {
- assert(eventLoop.inEventLoop)
- if let existing = thread.currentValue {
- return existing
- } else {
- let new = RFC1123DateCache()
- let fracSeconds = 1.0 - Date().timeIntervalSince1970.truncatingRemainder(dividingBy: 1)
- let msToNextSecond = Int64(fracSeconds * 1000) + 1
- eventLoop.scheduleRepeatedTask(initialDelay: .milliseconds(msToNextSecond), delay: .seconds(1)) { task in
- new.updateTimestamp()
- }
- self.thread.currentValue = new
- return new
- }
- }
- /// Thread-specific RFC1123
- private static let thread: ThreadSpecificVariable<RFC1123DateCache> = .init()
- /// Currently cached time components.
- private var cachedTimeComponents: (key: time_t, components: tm)?
- /// Currently cached timestamp.
- private var timestamp: String
- /// Creates a new `RFC1123DateCache`.
- private init() {
- self.timestamp = ""
- self.updateTimestamp()
- }
- func currentTimestamp() -> String {
- return self.timestamp
- }
- /// Updates the current RFC 1123 date string.
- func updateTimestamp() {
- // get the current time
- var date = time(nil)
- // generate a key used for caching
- // this key is a unique id for each day
- let key = date / secondsInDay
- // get time components
- let dateComponents: tm
- if let cached = self.cachedTimeComponents, cached.key == key {
- dateComponents = cached.components
- } else {
- var tc = tm.init()
- gmtime_r(&date, &tc)
- dateComponents = tc
- self.cachedTimeComponents = (key: key, components: tc)
- }
- // parse components
- let year: Int = numericCast(dateComponents.tm_year) + 1900 // years since 1900
- let month: Int = numericCast(dateComponents.tm_mon) // months since January [0-11]
- let monthDay: Int = numericCast(dateComponents.tm_mday) // day of the month [1-31]
- let weekDay: Int = numericCast(dateComponents.tm_wday) // days since Sunday [0-6]
- // get basic time info
- let t: Int = date % secondsInDay
- let hours: Int = numericCast(t / 3600)
- let minutes: Int = numericCast((t / 60) % 60)
- let seconds: Int = numericCast(t % 60)
- // generate the RFC 1123 formatted string
- var rfc1123 = ""
- rfc1123.reserveCapacity(30)
- rfc1123.append(dayNames[weekDay])
- rfc1123.append(", ")
- rfc1123.append(stringNumbers[monthDay])
- rfc1123.append(" ")
- rfc1123.append(monthNames[month])
- rfc1123.append(" ")
- rfc1123.append(stringNumbers[year / 100])
- rfc1123.append(stringNumbers[year % 100])
- rfc1123.append(" ")
- rfc1123.append(stringNumbers[hours])
- rfc1123.append(":")
- rfc1123.append(stringNumbers[minutes])
- rfc1123.append(":")
- rfc1123.append(stringNumbers[seconds])
- rfc1123.append(" GMT")
- // cache the new timestamp
- self.timestamp = rfc1123
- }
- }
- // MARK: Private
- private let dayNames = [
- "Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"
- ]
- private let monthNames = [
- "Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
- ]
- private let stringNumbers = [
- "00", "01", "02", "03", "04", "05", "06", "07", "08", "09",
- "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", "43", "44", "45", "46", "47", "48", "49",
- "50", "51", "52", "53", "54", "55", "56", "57", "58", "59",
- "60", "61", "62", "63", "64", "65", "66", "67", "68", "69",
- "70", "71", "72", "73", "74", "75", "76", "77", "78", "79",
- "80", "81", "82", "83", "84", "85", "86", "87", "88", "89",
- "90", "91", "92", "93", "94", "95", "96", "97", "98", "99"
- ]
- private let secondsInDay = 60 * 60 * 24
|