Simple Swift Function to memoize generic functions of the form T -> U

MemoizationWithGenerics

func memoize<T: Hashable,U>(function: @escaping (T) -> U) ->  (T) -> U{
    var cache : [T:U] = [:]
    func memoWrapper(input: T) -> U {
        if let cacheValue = cache[input] {
            print("used cached result")
            return cacheValue
        }
        let newVal = function(input)
        cache[input] = newVal
        print("new result")
        return newVal
    }
    return memoWrapper
}

func square(_ input: Int) -> Int {
    return input * input
}
let memoSquare = memoize(function: square)



memoSquare(2)
memoSquare(2)
memoSquare(3)
memoSquare(3)