Как добиться многопользовательской аренды на весеннем потоке с kotlin сопрограммой? - PullRequest
Новая
       36

Как добиться многопользовательской аренды на весеннем потоке с kotlin сопрограммой?

0 голосов
/ 13 января 2020

Извините за мой английский sh.

Я новичок весной и kotlin.
Проблема, которую я пытался решить, заключается в получении значения арендатора внутри kotlin сопрограммы. Я сделал простой пример https://github.com/cardid-zz/spring-multitenant-test

У меня есть класс TenantContext, в котором содержится переданное значение арендатора 

@Component
object TenantContext {
    const val DEFAULT: String = "default"

    private val logger = LoggerFactory.getLogger(javaClass)

    private val currentTenant = InheritableThreadLocal<String?>()

    fun getTenant() : String {
        return currentTenant.get() ?: DEFAULT
    }
    fun set(tenantId : String) {
        currentTenant.set(tenantId)
    }
    fun remove() {
        currentTenant.remove()
    }

    fun asContextElement(): ThreadContextElement<String?> {
        logger.debug("[d] asContextElement ${getTenant()}")
        return currentTenant.asContextElement(getTenant())
    }
}

Наборы значений арендатора в фильтре 

@Component
class TenantFilter (
    private val tenantContext: TenantContext
) : WebFilter {
    private val logger = LoggerFactory.getLogger(javaClass)

    private val tenantHeader = "tenant"

    override fun filter(
        serverWebExchange: ServerWebExchange,
        webFilterChain: WebFilterChain
    ): Mono<Void> {
        val tenant = serverWebExchange.request.headers[tenantHeader]

        if (tenant.isNullOrEmpty()) {
            setTenant(TenantContext.DEFAULT)
        } else {
            setTenant(tenant.first())
        }
        logger.debug("[d] currentThread =  ${Thread.currentThread()}")
        return webFilterChain.filter(serverWebExchange)
    }

    private fun setTenant(tenant: String) {
        try {
            tenantContext.set(tenant)
        } catch (e: Exception) {
            throw RuntimeException()
        }
    }
}

В контроллере у меня есть две конечные точки 

@RestController
class RestController(
    private val service: SomeService
) {
    private val logger = LoggerFactory.getLogger(javaClass)

    @PostMapping("/working")
    suspend fun working(@RequestParam("param") param : Int) : ResponseEntity<*> = coroutineScope(){
        val res = async{ service.doSomething(param)}
        return@coroutineScope ResponseEntity.ok(res)
    }

    @PostMapping("/failed")
    suspend fun failed(@RequestBody body: BodyParam) : ResponseEntity<*> = coroutineScope(){
        logger.debug("[d] ${body.toString()}")
        val res = async { service.doSomething(body.value) }

        return@coroutineScope ResponseEntity.ok(res)
    }
}

Одна из них работает нормально, другая нет. Разница в параметрах, метод работы имеет requestparam, неудачный получает param через тело. 

printf "\nworking go"

curl -i -X POST \
   -H "tenant:properTenant" \
   -H "Content-Type:application/json" \
 'http://localhost:8080/working?param=123'

printf "\nfailed go"

curl -i -X POST \
   -H "tenant:properTenant" \
   -H "Content-Type:application/json" \
   -d \
'{"value":21312}' \
 'http://localhost:8080/failed'


working returns
"param = 123 and tenant was = properTenant"

failed returns
"param = 21312 and tenant was = default"

Как получить значение арендатора из / с ошибочным методом?

1 Ответ

0 голосов
/ 26 апреля 2020

Я недостаточно знаком с Kotlin, чтобы привести конкретный пример, но вы не должны использовать ThreadLocal или InheritableThreadLocal в приложении Flux, поскольку у вас нет выделенного потока на запрос, как в традиционном веб-приложении. , Вероятно, поэтому вы получаете неожиданные результаты.

Рассмотрите возможность использования контекста, как описано здесь .

...