[Udemy iOS & Swift Bootcamp] extension

extension ?

기존 클래스, 구조체, 열거형 타입에 새로운 기능을 추가하는 것.

Swift가 오픈소스가 아니라서 기존 소스코드에서 접근하지 못하는 타입들을 확장할 수 있다.

 

 

코드 기본형

extension someType{
	//Add new functionality
}

 

예시 1)

import UIKit

var myDouble = 3.14519

let myRoundedDouble = String(format: "%.1f", myDouble) // 3.1
myDouble.round() //3
//myDouble.round(to: 3) //???? round에 to값을 넣는 함수는 없으므로 error를 발생시킨다.

3.14519 라는 값을 가진 myDouble 변수가 있다.

 

이 걸 String 함수를 이용해서 소수점 1의 자리까지만 출력할 수 있다.

round() 함수를 이용하면 정수만 출력할 수도 있다.

 

이 때, myDouble.round(to:3) 이렇게 사용할 수도 있을까?

round 함수에는 to 값을 보내는 함수는 없다. 그래서 error 가 난다.

 

하지만 extension 으로 직접 만든다면 myDouble.round(to:3)을 사용할 수도 있다.

import UIKit

extension Double{
    //(외부인자 내부인자 : 타입) -> 리턴타입
    func round(to places: Int) -> Double
    {
        let precisionNumber = pow(10, Double(places))
        var n = self
        n = n * precisionNumber
        n.round()
        n = n / precisionNumber
        return n
    }
}

var myDouble = 3.14519

myDouble = myDouble * 1000
myDouble.round()
myDouble = myDouble / 1000

myDouble.round(to: 1)

Double extension 에 round(to places: Int) -> Double 함수를 만들었다.

이제 myDouble.round(to: 1) 요런 형식이 가능하다.

myDouble.round(to:1) 은 1의 자리수까지 반환하는 것!

 

예시2)

버튼 함수에도 extension 을 만들 수 있다.

위는 코드로 UIButton을 만들고 배경색을 입혀준 모습이다.

그냥 버튼 말고 둥근 버튼을 만들고 싶은데, 그런 함수는 없다. 근데 extension 으로 만들면 된다.

UIButton extension 을 만들고 그 안에 makeCircular() 함수를 추가했다.

button에 makeCircular() 함수가 생기고 이를 호출하면 위와 각이 둥근 버튼이 만들어 진다.

 

그 외 프로토콜도 가능 !

 

---

Clima 프로젝트에 적용

 

//이전 코드

import UIKit

class WeatherViewController: UIViewController, UITextFieldDelegate, WeatherManagerDelegate {

    @IBOutlet weak var conditionImageView: UIImageView!
    @IBOutlet weak var temperatureLabel: UILabel!
    @IBOutlet weak var cityLabel: UILabel!
    @IBOutlet weak var searchTextField: UITextField!
    
    var weatherManager = WeatherManager()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        weatherManager.delegate = self
        searchTextField.delegate = self
    }

    @IBAction func serarchPressed(_ sender: UIButton) {
        searchTextField.endEditing(true)
    }
    
    func textFieldShouldReturn(_ textField: UITextField) -> Bool {
        searchTextField.endEditing(false)
        print(searchTextField.text!)
        return true
    }
    
    func textFieldShouldEndEditing(_ textField: UITextField) -> Bool {
        if textField.text != ""{
            return true
        }
        else {
            textField.placeholder = "Type Something"
            return false
        }
    }
    
    func textFieldDidEndEditing(_ textField: UITextField) {
        if let city = searchTextField.text {
            weatherManager.fetchWeather(cityName: city)
        }
        searchTextField.text = ""
    }
    
    func didUpdateWeather(_ weatherManger: WeatherManager, weather: WeatherModel){
        //DispatchQueue 로 안기다리면 에러 남.
        DispatchQueue.main.async{
            self.temperatureLabel.text = weather.temperatureName
            self.conditionImageView.image = UIImage(systemName: weather.conditionName)
        }
        print(weather.temperature)
    }
    
    func didFailWithError(error: Error){
        print(error)
    }
}

 

//extension을 적용해서 코드를 단순화 한 것.

import UIKit

class WeatherViewController: UIViewController {

    @IBOutlet weak var conditionImageView: UIImageView!
    @IBOutlet weak var temperatureLabel: UILabel!
    @IBOutlet weak var cityLabel: UILabel!
    @IBOutlet weak var searchTextField: UITextField!
    
    var weatherManager = WeatherManager()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        weatherManager.delegate = self
        searchTextField.delegate = self
    }
}

extension WeatherViewController: UITextFieldDelegate{
    
    @IBAction func serarchPressed(_ sender: UIButton) {
        searchTextField.endEditing(true)
    }
    
    func textFieldShouldReturn(_ textField: UITextField) -> Bool {
        searchTextField.endEditing(false)
        print(searchTextField.text!)
        return true
    }
    
    func textFieldShouldEndEditing(_ textField: UITextField) -> Bool {
        if textField.text != ""{
            return true
        }
        else {
            textField.placeholder = "Type Something"
            return false
        }
    }
    
    func textFieldDidEndEditing(_ textField: UITextField) {
        if let city = searchTextField.text {
            weatherManager.fetchWeather(cityName: city)
        }
        searchTextField.text = ""
    }
}

extension WeatherViewController: WeatherManagerDelegate{
    
    func didUpdateWeather(_ weatherManger: WeatherManager, weather: WeatherModel){
        //DispatchQueue 로 안기다리면 에러 남.
        DispatchQueue.main.async{
            self.temperatureLabel.text = weather.temperatureName
            self.conditionImageView.image = UIImage(systemName: weather.conditionName)
        }
        print(weather.temperature)
    }
    
    func didFailWithError(error: Error){
        print(error)
    }
}