Swift iOS - как отменить DispatchGroup () от управления циклом

Question

Я перебираю несколько Url с, преобразую их в Data, затем отправляю данные в Firebase Storage, а затем, когда все сделано, отправляю собранную информацию в Firebase Database

Я использую DispatchGroup() * .enter(), чтобы запустить цикл, и как только я отправляю данные в хранилище и получаю строку URL-адреса значения absoluteString Я использую .leave(), чтобы начать следующую итерацию.

Я понял, чтово время цикла есть несколько точек, где могут возникать ошибки:

1. один раз внутри UrlSession
2. один раз внутри функции Storage .putData
3. один раз внутри Storage.downloadURL(completion:... обработчик завершения
4. и снова, если конечный параметр downloadURL ?.absoluteString равен nil

Если я получаю ошибку в любой из этих точек, я показываю функцию предупреждения showAlert()который показывает предупреждение и отменяет UrlSession с session.invalidateAndCancel().Я отменяю все, потому что я хочу, чтобы пользователь начал все заново.

Поскольку DispatchGroup () остается висеть на .enter(), как мне отменить DispatchGroup (), чтобы остановить цикл?

var urls = [URL]()
var picUUID = UUID().uuidString
var dict = [String:Any]()

let session = URLSession.shared
let myGroup = DispatchGroup()
var count = 0

for url in urls{

    myGroup.enter()
    session.dataTask(with: url!, completionHandler: {
            (data, response, error) in

            if error != nil { 
                self.showAlert() // 1st point of error
                return 
            }

            DispatchQueue.main.async{
                self.sendDataToStorage("\(self.picUUID)_\(self.count).jpg", picData: data)
                self.count += 1
            }
    }).resume()

    myGroup.notify(queue: .global(qos: .background) {
        self.sendDataFromDictToFirebaseDatabase()
        self.count = 0
        self.session.invalidateAndCancel()
   }
}

func sendDataToStorage(_ picId: String, picData: Data?){

    dict.updateValue(picId, forKey:"picId_\(count)")

    let picRef = storageRoot.child("pics")
    picRef.putData(picData!, metadata: nil, completion: { (metadata, error) in

        if error != nil{
            self.showAlert()  // 2nd point of error
            return
        }

        picRef?.downloadURL(completion: { (url, error) in

            if error != nil{
                self.showAlert()  // 3rd point of error
                return
            }

            if let picUrl = url?.absoluteString{

               self.dict.updateValue(picUrl, forKey:"picUrl_\(count)")
               self.myGroup.leave() //only leave the group if a Url string was obtained
            }else{
               self.showAlert()  // 4th point of error
            }
        })
    })
}

func showAlert(){
    // the DispatchGroup() should get cancelled here
    session.invalidateAndCancel()
    count = 0
    UIAlertController...
}

func sendDataFromDictToFirebaseDatabase(){
}

Answer 1

В комментариях ниже вопрос @rmaddy сказал: «Вам нужно позвонить leave, успешно это или нет».Я сделал это, но цикл все еще работал, и sendDataFromDictToFirebaseDatabase() все еще вызывал событие, хотя произошла ошибка.

Единственное, что я мог найти, - это поместить цикл в функцию с обработчиком завершения и использовать bool решить, стоит ли или нет sendDataFromDictToFirebaseDatabase() должен выстрелить:

var urls = [URL]()
var picUUID = UUID().uuidString
var dict = [String:Any]()

let session = URLSession.shared
let myGroup = DispatchGroup()
var count = 0
var wasThereAnError = false // use this bool to find out if there was an error at any of the error points

func loopUrls(_ urls: [URL?], completion: @escaping ()->()){

    for url in urls{

        myGroup.enter()
        session.dataTask(with: url!, completionHandler: {
            (data, response, error) in

            if error != nil {
                self.showAlert() // 1st point of error. If there is an error set wasThereAnError = true
                return
            }

            DispatchQueue.main.async{
                self.sendDataToStorage("\(self.picUUID)_\(self.count).jpg", picData: data)
                self.count += 1
            }
        }).resume()

        myGroup.notify(queue: .global(qos: .background) {
            completion()
        }
    }
}

// will run in completion handler
func loopWasSuccessful(){

    // after the loop finished this only runs if there wasn't an error
    if wasThereAnError = false {
        sendDataFromDictToFirebaseDatabase()
        count = 0
        session.invalidateAndCancel()
    }
}

func sendDataToStorage(_ picId: String, picData: Data?){

    dict.updateValue(picId, forKey:"picId_\(count)")

    let picRef = storageRoot.child("pics")
    picRef.putData(picData!, metadata: nil, completion: { (metadata, error) in

        if error != nil{
            self.showAlert()  // 2nd point of error. If there is an error set wasThereAnError = true
            return
        }

        picRef?.downloadURL(completion: { (url, error) in

            if error != nil{
                self.showAlert()  // 3rd point of error. If there is an error set wasThereAnError = true
                return
            }

            if let picUrl = url?.absoluteString{

                self.dict.updateValue(picUrl, forKey:"picUrl_\(count)")
                self.myGroup.leave() // leave group here if all good on this iteration
            }else{
                self.showAlert()  // 4th point of error. If there is an error set wasThereAnError = true
            }
        })
    })
}

func showAlert(){
    wasThereAnError = true // since there was an error set this to true
    myGroup.leave() // even though there is an error still leave the group
    session.invalidateAndCancel()
    count = 0
    UIAlertController...
}

func sendDataFromDictToFirebaseDatabase(){
}

И использовать его:

@IBAction fileprivate func postButtonPressed(_ sender: UIButton) {    

    wasThereAnError = false // set this back to false because if there was an error it was never reset

    loopUrls(urls, completion: loopWasSuccessful)
}