[Swift] Custom Calendar 달의 마지막 날 구하기

 

 

이 글은 UI를 제외하고 날짜 계산을 하는 로직만 담고 있습니다

 

 

Custom Calendar를 만들다보니 해당하는 달의 마지막 날이 언제인지 계산해야 했다.

알고리즘 처럼 접근하면 윤 달도 계산해줘야 하고 속도 측면에서도 비효율적이라고 생각해서

foundation상에 있는 calendar라는 구조체를 사용해서 만들었다!

 

 

먼저 날짜 계산에 필요한 Calendar와 DateFormatter를 만들어주자

let calendar = Calendar(identifier: .gregorian)
let dateFormatter = DateFormatter()

dateFormatter.dateFormat = "yyyy-MM-dd"

 

첫 날은 무조건 1일이니까 직접 입력해준다

let st = dateFormatter.date(from: "\(year)-\(month)-01")!

마지막 날을 구하는 방법은 구하고자 하는 달의 다음 달 1일을 구해서 빼 줄 것이다

(ex. 2월1일 - 1월1일 = 31일)

// 다음달 구하기
let end = calendar.date(byAdding: .month, value: +1, to: st)!

let component = calendar.dateComponents([.day], from: st, to: end)
component.day // end-st

 

 

full code

let formatter = DateFormatter()
let calendar = Calendar(identifier: .gregorian)

func lastDay(year: String, month: String) -> Int?{
    formatter.dateFormat = "yyyy-MM-dd"
    let st = formatter.date(from: "\(year)-\(month)-01")!
    let end = calendar.date(byAdding: .month, value: +1, to: st)!
    let result = calendar.dateComponents([.day], from: st, to: end)

    guard let day = result.day else {
        return nil
    }
    return day
}

 

 

 

 

 

더 좋은 방법 있으면 공유해주세요 ~~ 😉