Swift3으로 TCPIP Socket 을 구현하려면 막막한데요. Stream Class를 이용하면 간단히 구현 할 수 있습니다.
아래와 같이 Deleate를 등록하면 에러 이벤트 처리도 가능합니다.
import Foundation
class TcpSocket: NSObject, StreamDelegate {
var host:String?
var port:Int?
var inputStream: InputStream?
var outputStream: OutputStream?
func connect(host: String, port: Int) {
self.host = host
self.port = port
Stream.getStreamsToHost(withName:host, port : port, inputStream: &inputStream, outputStream: &outputStream)
if inputStream != nil && outputStream != nil {
// Set delegate
inputStream!.delegate = self
outputStream!.delegate = self
// Schedule
inputStream!.schedule(in: .main, forMode: RunLoopMode.defaultRunLoopMode)
outputStream!.schedule(in: .main, forMode: RunLoopMode.defaultRunLoopMode)
print("Start open()")
// Open!
inputStream!.open()
outputStream!.open()
}
}
func send(data: Data) -> Int {
let bytesWritten = data.withUnsafeBytes { outputStream?.write($0, maxLength: data.count) }
return bytesWritten!
}
// func send(data: String) -> Int {
// let bytesWritten = outputStream?.write(data, maxLength:data.characters.count)
// return bytesWritten!
// }
func recv(buffersize: Int) -> Data {
var buffer = [UInt8](repeating :0, count : buffersize)
let bytesRead = inputStream?.read(&buffer, maxLength: buffersize)
var dropCount = buffersize - bytesRead!
if dropCount < 0 {
dropCount = 0
}
let chunk = buffer.dropLast(dropCount)
return Data(chunk)
}
func disconnect() {
inputStream?.close()
outputStream?.close()
}
func stream(_ stream: Stream, handle eventCode: Stream.Event) {
print("event:\(eventCode)")
if stream === inputStream {
switch eventCode {
case Stream.Event.errorOccurred:
print("inputStream:ErrorOccurred")
case Stream.Event.openCompleted:
print("inputStream:OpenCompleted")
case Stream.Event.hasBytesAvailable:
print("inputStream:HasBytesAvailable")
default:
break
}
}
else if stream === outputStream {
switch eventCode {
case Stream.Event.errorOccurred:
print("outputStream:ErrorOccurred")
case Stream.Event.openCompleted:
print("outputStream:OpenCompleted")
case Stream.Event.hasSpaceAvailable:
print("outputStream:HasSpaceAvailable")
default:
break
}
}
}
}
사용법
let socket = TcpSocket()
....
socket.connect(host: "118.34.167.155", port: 5552)
let query = "HELLO SWIFT SOCKET!"
let dataQuery = query.data(using: String.Encoding.utf8, allowLossyConversion: true)
let sentCount = socket.send(data: dataQuery!)
//let sentCount = socket.send(data: query)
print("sentCount : \(sentCount)")
let buffersize = 1024
let chunk = socket.recv(buffersize: buffersize)
var getString : String?
if(chunk.count > 0){
getString = String(bytes: chunk, encoding: String.Encoding.utf8)!
print("received : \(getString!)")
}
connect.disconnect()
