package io.dataease.dds.interceptor;

import io.dataease.dds.DynamicContextHolder;
import io.dataease.dds.constant.DataSourceConstant;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import org.springframework.web.servlet.HandlerInterceptor;

/**
 * 不同租户不同二级域名
 * nginx下把租户ID以及租户创建时间设置在请求头
 * 获取请求头部租户ID和创建时间
 * 根据租户ID和创建时间获取租户数据源名称
 * 切换数据源
 */
public class DSInterceptor implements HandlerInterceptor {


    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        String tenant = request.getHeader("tenant");
        String time = request.getHeader("time");
        DynamicContextHolder.push(getTenantDsKey(tenant, time));
        return true;
    }

    private String getTenantDsKey(String tenant, String time) {
        if (null == tenant || null == time) return "official-ds";
        return String.format(DataSourceConstant.DS_NAME_PREFIX, tenant, time);
    }
}