[swift] 서브스크립트

 

책 <스위프트 프로그래밍 3판 ;야곰 지음> 을 정리한 글입니다.

 

- 서브스크립트

- 서브스크립트 문법

- 서브스크립트 구현

- 복수 서브스크립트

- 타입 서브스크립트

 

---

 

서브스크립트

- 서브크스크립트 (Subscript)를 통해 클래스, 구조체, 열거형 등의 요소에 접근 가능

- 별도의 Setter, Getter 메서드를 구현하지 않아도 인덱스를 통해 값을 설정하거나 가져올 수 있음

 

 

서브스크립트 문법

- 인스턴스의 이름 뒤에 대괄호로 감싼 값을 써줌으로써 인스턴스 내부의 특정 값에 접근 가능

- 서브스크립트를 정의하는 코드는 각 타입의 구현부 또는 타입의 익스텐션 구현부에 위치해야 함

// 서브스크립트 정의 문법

subscript(index: Int) -> Int {
    get {
        // 적절한 스크립트 결괏값 변환
    }
    
    set(newValue) {
        // 적절한 설정자 역할 수행
    }
}

// 읽기 전용 서브스크립트 정의 문법

subscript(index: Int) -> Int {
    get {
        // 적절한 서브스크립트 값 반환
    }
}

subscript(index: Int) -> Int {
    // 적절한 서브스크립트 결괏값 변환
}

 

 

서브스크립트 구현

- 학생의 번호를 전달인자로 받아서 자신의 students 프로퍼티의 인덱스에 맞는 Student 인스턴스를 반환

struct Student{
    var name: String
    var number: Int
}

class School {
    
    var number: Int = 0
    var students: [Student] = [Student]()
    
    func addStudent(name: String) {
        let student: Student = Student(name: name, number: self.number)
        self.students.append(student)
        self.number += 1
    }
    
    func addStudents(names: String...) {
        for name in names {
            self.addStudent(name: name)
        }
    }
    
    subscript(index: Int) -> Student? {
        if index < self.number {
            return self.students[index]
        }
        
        return nil
    }
}

let highSchool: School = School()
highSchool.addStudents(names: "Mijeong", "JuHyun", "JiYoung", "SeongUk", "MoonDuk")

let aStudent: Student? = highSchool[1]
print("\(aStudent?.number) \(aStudent?.name)") // Optional(1) Optional("JuHyun")

 

 

 

복수 서브스크립트

- 하나의 타입이 여러 개의 서브스크립트를 가질 수 있음

struct Student {
    var name: String
    var number: Int
}

class School {
    
    var number: Int = 0
    var students: [Student] = [Student]()
    
    func addStudent(name: String) {
        let student: Student = Student(name: name, number: self.number)
        self.students.append(student)
        self.number += 1
    }
    
    func addStudents(names: String...) {
        for name in names {
            self.addStudent(name: name)
        }
    }
    
    subscript(index: Int) -> Student? { // 첫 번째 서브스크립트 - 읽기 쓰기 가능
        get {
            if index < self.number {
                return self.students[index]
            }
            return nil
        }
        
        set {
            guard var newStudent: Student = newValue else {
                return
            }
            
            var number: Int = index
            
            if index > self.number {
                number = self.number
                self.number += 1
            }
            
            newStudent.number = number
            self.students[number] = newStudent
        }
    }
    
    subscript(name: String) -> Int? { // 두 번째 서브스크립트 - 읽기 쓰기 가능
        get {
            return self.students.filter({$0.name == name}).first?.number
        }
        
        set {
            guard var number: Int = newValue else {
                return
            }
            
            if number > self.number {
                number = self.number
                self.number += 1
            }
            
            let newStudent: Student = Student(name: name, number: number)
            self.students[number] = newStudent
        }
    }
    
    subscript(name: String, number: Int) -> Student? { // 세 번째 서브스크립트 - 읽기만 가능
        return self.students.filter{ $0.name == name && $0.number == number }.first
    }
}

let highSchool: School = School()
highSchool.addStudents(names: "MiJeong", "JuHyun", "JiYoung", "SeongUk", "MoonDuk")

let aStudent: Student? = highSchool[1]
print("\(aStudent?.number) \(aStudent?.name)") // Optional(1) Optional("JuHyun")

print(highSchool["MiJeong"]) // Optional(0)
print(highSchool["DongJin"]) // nil

highSchool[0] = Student(name: "HongEui", number: 0)
highSchool["MangGu"] = 1

print(highSchool["JuHyun"]) // nil
print(highSchool["MangGu"]) // Optional(1)
print(highSchool["SeongUl", 3]) // nil
print(highSchool["HeeJin", 3]) // nil

 

 

타입 서브스크립트

- 위의 서브스크립트들은 인스턴스에서 사용할 수 있는 서브스크립트

- 타입 서브스크립트는 인스턴스가 아니라 타입 자체에서 사용할 수 있음

- 타입 서브스크립트를 구현하려면 정의할 때 subscript 키워드 앞에 static  키워드를 붙여주면 됨

- 클래스의 경우 class 키워드를 사용할 수도 있음

 

enum School: Int {
    case elementary = 1, middle, high, university
    
    static subscript(level: Int) -> School? {
        return Self(rawValue: level)
    }
}

let school: School? = School[2]
print(school!) // middle