(improvement)(headless)Remove unnecessary TranslateSqlReq, use SemanticQueryReq instead.

This commit is contained in:
jerryjzhang
2024-07-09 10:48:48 +08:00
parent 7a376bd9a3
commit f0b4eb46cf
32 changed files with 138 additions and 176 deletions

View File

@@ -5,8 +5,13 @@ import lombok.Data;
@Data
public class SqlInfo {
private String s2SQL;
private String correctS2SQL;
// S2SQL generated by semantic parsers
private String parsedS2SQL;
// S2SQL corrected by semantic correctors
private String correctedS2SQL;
// SQL to be executed finally
private String querySQL;
private String sourceId;
}

View File

@@ -164,7 +164,7 @@ public class QueryStructReq extends SemanticQueryReq {
result.setDataSetId(this.getDataSetId());
result.setModelIds(this.getModelIdSet());
result.setParams(new ArrayList<>());
result.getSqlInfo().setCorrectS2SQL(sql);
result.getSqlInfo().setCorrectedS2SQL(sql);
return result;
}

View File

@@ -1,20 +0,0 @@
package com.tencent.supersonic.headless.api.pojo.request;
import com.tencent.supersonic.headless.api.pojo.enums.QueryMethod;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.ToString;
@Data
@ToString
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class TranslateSqlReq<T> {
private QueryMethod queryTypeEnum;
private T queryReq;
}

View File

@@ -13,10 +13,12 @@ import java.io.Serializable;
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class TranslateResp implements Serializable {
public class SemanticTranslateResp implements Serializable {
private String querySQL;
private String sql;
private boolean isOk;
private String errMsg;
}