package org.ruoyi.system.cofing;
|
|
import jakarta.annotation.PostConstruct;
|
import lombok.Getter;
|
import lombok.RequiredArgsConstructor;
|
import org.ruoyi.common.core.utils.OkHttpUtil;
|
import org.ruoyi.system.domain.bo.SysModelBo;
|
import org.ruoyi.system.domain.vo.SysModelVo;
|
import org.ruoyi.system.service.ISysModelService;
|
import org.springframework.stereotype.Component;
|
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
@Component
|
@RequiredArgsConstructor
|
public class OkHttpConfig {
|
|
private final ISysModelService sysModelService;
|
private final Map<String, OkHttpUtil> okHttpUtilMap = new HashMap<>();
|
@Getter
|
private String generate;
|
|
@PostConstruct
|
public void init() {
|
initializeOkHttpUtil("suno");
|
initializeOkHttpUtil("luma");
|
initializeOkHttpUtil("ppt");
|
}
|
|
private void initializeOkHttpUtil(String modelName) {
|
SysModelBo sysModelBo = new SysModelBo();
|
sysModelBo.setModelName(modelName);
|
List<SysModelVo> sysModelList = sysModelService.queryList(sysModelBo);
|
if (!sysModelList.isEmpty()) {
|
SysModelVo model = sysModelList.get(0);
|
OkHttpUtil okHttpUtil = new OkHttpUtil();
|
okHttpUtil.setApiHost(model.getApiHost());
|
okHttpUtil.setApiKey(model.getApiKey());
|
generate = String.valueOf(model.getModelPrice());
|
okHttpUtilMap.put(modelName, okHttpUtil);
|
}
|
}
|
|
public OkHttpUtil getOkHttpUtil(String modelName) {
|
return okHttpUtilMap.get(modelName);
|
}
|
}
|
