[swift] Swift Moya에서 네트워크 스로틀링 구현하기

28 Nov 2023

swift

Swift Moya는 iOS 애플리케이션에서 네트워크 요청을 쉽게 처리할 수 있는 라이브러리입니다. 이번에는 Moya를 사용하여 네트워크 스로틀링을 구현하는 방법에 대해 알아보겠습니다.

스로틀링은 네트워크 요청을 일정한 속도로 제한하는 것을 의미합니다. 이를 통해 동시에 많은 요청이 발생하는 상황에서도 서버에 과부하가 걸리지 않도록 할 수 있습니다.

1. Moya의 스로틀링 기능 사용

Moya는 네트워크 요청을 처리하는 Provider를 구현할 때 스로틀링 기능을 제공합니다. 다음은 Moya에서 스로틀링을 사용하는 예제 코드입니다.

import Moya

// API 스로틀링 기간을 정의하는 enum
enum ThrottleInterval {
    case none
    case short
    case long
    
    var interval: TimeInterval {
        switch self {
        case .none: return 0
        case .short: return 0.5
        case .long: return 2.0
        }
    }
}

// Provider 생성 시 스로틀링 설정
let provider = MoyaProvider<MyAPI>(plugins: [NetworkLoggerPlugin(),
                                            NetworkActivityIndicatorPlugin(throttleInterval: .short)])

// 네트워크 요청 시 스로틀링 설정
provider.request(.getUserInfo) { result in
    // 요청 결과 처리
}

위 예제에서는 MoyaProvider를 생성할 때 NetworkActivityIndicatorPlugin을 통해 스로틀링을 설정하고 있습니다. throttleInterval 매개변수에는 스로틀링 기간을 정의한 enum 값을 전달하여 원하는 스로틀링 설정을 할 수 있습니다.

2. 커스텀 스로틀링 구현

만약 Moya에서 제공하는 스로틀링 설정이 아닌 독자적인 스로틀링 로직을 구현하고 싶다면, 다음과 같이 커스텀 스로틀러를 만들어 사용할 수 있습니다.

import Moya

class CustomThrottler: Moya.PluginType {
    let interval: TimeInterval
    
    init(interval: TimeInterval) {
        self.interval = interval
    }
    
    func prepare(_ request: URLRequest, target: TargetType) -> URLRequest {
        var modifiedRequest = request
        // 스로틀링 로직 구현
        // interval 동안 다음 네트워크 요청이 발생하지 않도록 제한
        return modifiedRequest
    }
}

// Provider 생성 시 커스텀 스로틀러 설정
let provider = MoyaProvider<MyAPI>(plugins: [NetworkLoggerPlugin(),
                                            CustomThrottler(interval: 1.0)])

// 네트워크 요청 시 커스텀 스로틀러 설정
provider.request(.getUserInfo) { result in
    // 요청 결과 처리
}

위 예제에서는 CustomThrottler 클래스를 생성하여 Moya.PluginType을 구현한 뒤 prepare(_:target:) 메서드를 통해 커스텀 스로틀링 로직을 구현하고 있습니다. 이를 통해 원하는 스로틀링 동작을 정의할 수 있습니다.

3. 마치며

Swift Moya를 사용하여 네트워크 스로틀링을 구현하는 방법에 대해 알아보았습니다. Moya를 통해 간편하게 스로틀링 설정을 할 수 있으며, 필요한 경우 커스텀 스로틀러를 만들어 독자적인 스로틀링 로직을 구현할 수도 있습니다.

더 많은 정보를 원하신다면 Moya GitHub 저장소를 참고하시기 바랍니다.

참고: 이 예제는 Moya 14.0.0 버전을 기준으로 작성되었습니다.