recycleview two columns and if list size is one, show row as a one column, othervise two columns

recycleview two columns

//documents -> list of recycleview

val layoutManager = GridLayoutManager(activity, 2)
        layoutManager.spanSizeLookup = object : GridLayoutManager.SpanSizeLookup() {
            override fun getSpanSize(position: Int): Int {
                return if (documents.size > 1) 1 else 2
            }
        }