Chức năng ghi nhớ Trong Kotlin

Question

Tôi có một lớp hiện có với một phương thức thể hiện buildHierarchyUncached có chữ ký có thể được tìm thấy bên dưới.

private fun buildHierarchyUncached(date: LocalDate): Node { ... } 

tôi muốn cung cấp một công chức năng buildHiearchy đó là một phiên bản memoized của buildHierarchyUncached. Tôi có thể lấy gần với những gì tôi muốn:

val buildHiearchy = Memoize<LocalDate, Node>({buildHierarchy(it)}) 

nào có thể được gọi như:

hierarchyService.buildHiearchy(businessDate) 

Sử dụng:

class Memoize<I, O>(val func: (I) -> O): (I) -> O{ 
    val cache = hashMapOf<I, O>(); 
    override fun invoke(p1: I): O { 
    return cache.getOrPut(p1, { func(p1) }) 
    } 
} 

Tôi muốn để có thể khai báo hàm memoized như một chức năng thay vì một tài sản, mà không phải là một vấn đề lớn, mặc dù tôi nghĩ rằng nó giúp dễ đọc. Như thế này:

fun buildHierarchy(date: LocalDate): Node = Memoize<LocalDate, Node>({ buildHierarchyUncached(it)}) 

nhưng không biên dịch: "Loại không phù hợp. Yêu cầu nút. Tìm thấy ghi nhớ".

Ngoài ra, tại sao bản dịch này không biên dịch?

val buildHiearchy = Memoize<LocalDate, Node>({(date) -> buildHierarchy(date)}) 

Answer 1

Do bản chất của sự cố, bạn cần trường lớp để lưu bộ nhớ cache của bạn (giá trị được lưu trong bộ nhớ cache hoặc đối tượng bộ nhớ cache hoặc đại biểu). Vì vậy, bạn phải khai báo một lớp val ở một nơi nào đó, vì các hàm không thể làm điều đó.

Lưu ý rằng khi bạn khai báo giá trị buildHiearchy, bạn có hai thứ trong một: bạn lưu trữ đối tượng Memoize<..>(..) trong trường lớp và bạn nhận được hàm invoke() (được khai báo ở một nơi khác, nhưng vẫn ..). Tôi biết không có cách nào bạn có thể tuyên bố một chức năng và nhận được các lĩnh vực lưu trữ mà không có cú pháp bổ sung.

Đoạn mã sử dụng cú pháp lỗi thời. Khắc phục như thế này (không có dấu ngoặc đơn):

val buildHiearchy = Memoize<LocalDate, Node>({date -> buildHierarchy(date)}) 

Answer 2

Giải pháp sau đây hoạt động cho các hàm đối số đơn. Nếu bạn muốn tạo ra một phiên bản cache của hàm bar bạn chỉ cần khai báo nó như thế này:

val cachedBar = makeFunctionCache({ bar(it) }) 

Các cửa hàng thực hiện bộ nhớ cache trong một đóng cửa để bạn không cần phải đặt nó trong một lớp học chuyên dụng:

fun <X, R> makeFunctionCache(fn: (X) -> R): (X) -> R { 
    val cache: MutableMap<X, R> = HashMap() 
    return { 
     cache.getOrPut(it, { fn(it) }) 
    } 
}