sample.swift

songsService.callApiGetArtists(onSuccess: { (response) in
            self.artists = response.getArtists()
            self.tableView.reloadData()
        }) { (error) in
            self.handleError(error: error)
        }
        
        //=--------------------------
        
        func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        if (artistButton.isActive) {
            return artists.count
        } else {
            return songs.count
        }
    }
    
    //---------------
    
    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        if (artistButton.isActive) {
                let cell = tableView.dequeueReusableCell(withIdentifier: "ArtistCell")!
                cell.textLabel?.text = presenter.searchArtists[indexPath.row].name
                return cell
        } else {
                return createSongCell(song: presenter.songs[indexPath.row], lastRow: indexPath.row == songs.count-1)           
        }
    }