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
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
|
//
// TextFileOutputStream.swift
// MullvadVPN
//
// Created by pronebird on 02/08/2020.
// Copyright © 2020 Mullvad VPN AB. All rights reserved.
//
import Foundation
class TextFileOutputStream: TextOutputStream {
private let writer: DispatchIO
private let encoding: String.Encoding
private let queue = DispatchQueue.global(qos: .utility)
class func standardOutputStream(encoding: String.Encoding = .utf8) -> TextFileOutputStream {
return TextFileOutputStream(
fileDescriptor: FileHandle.standardOutput.fileDescriptor,
encoding: encoding
)
}
init(fileDescriptor: Int32, encoding: String.Encoding = .utf8) {
self.encoding = encoding
writer = DispatchIO(type: .stream, fileDescriptor: fileDescriptor, queue: queue) { errno in
if errno != 0 {
print("TextFileOutputStream: closed channel with error: \(errno)")
}
}
}
init?(
fileURL: URL,
createFile: Bool,
filePermissions: mode_t = S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH,
encoding: String.Encoding = .utf8
) {
var oflag: Int32 = O_WRONLY
var mode: mode_t = .zero
if createFile {
oflag |= O_CREAT
mode = filePermissions
}
let queue = queue
let writer = fileURL.path.withCString { filePathPointer -> DispatchIO? in
return DispatchIO(
type: .stream,
path: filePathPointer,
oflag: oflag,
mode: mode,
queue: queue,
cleanupHandler: { errno in
if errno != 0 {
print("TextFileOutputStream: closed channel with error: \(errno)")
}
}
)
}
if let writer = writer {
self.writer = writer
self.encoding = encoding
} else {
return nil
}
}
deinit {
writer.close()
}
func write(_ string: String) {
string.data(using: encoding)?.withUnsafeBytes { bytes in
writer
.write(
offset: .zero,
data: DispatchData(bytes: bytes),
queue: queue
) { done, data, errno in
if errno != 0 {
print("TextFileOutputStream: write error: \(errno)")
}
}
}
}
}
|
