[improvement][headless]Use LamdaQueryWrapper to avoid hard-coded column names.

auth/authentication/src/main/java/com/tencent/supersonic/auth/authentication/persistence/repository/impl/UserRepositoryImpl.java

@@ -51,7 +51,7 @@ public class UserRepositoryImpl implements UserRepository {
     @Override
     public List<UserTokenDO> getUserTokenListByName(String userName) {
         QueryWrapper<UserTokenDO> queryWrapper = new QueryWrapper<>();
-        queryWrapper.eq("user_name", userName);
+        queryWrapper.lambda().eq(UserTokenDO::getUserName, userName);
         return userTokenDOMapper.selectList(queryWrapper);
     }
 
@@ -68,7 +68,7 @@ public class UserRepositoryImpl implements UserRepository {
     @Override
     public void deleteUserTokenByName(String userName) {
         QueryWrapper<UserTokenDO> queryWrapper = new QueryWrapper<>();
-        queryWrapper.eq("user_name", userName);
+        queryWrapper.lambda().eq(UserTokenDO::getUserName, userName);
         userTokenDOMapper.delete(queryWrapper);
     }
 

chat/server/src/main/java/com/tencent/supersonic/chat/server/processor/parse/QueryRecommendProcessor.java

@@ -56,7 +56,7 @@ public class QueryRecommendProcessor implements ParseResultProcessor {
     private void updateChatQuery(ChatQueryDO chatQueryDO) {
         ChatQueryRepository chatQueryRepository = ContextUtils.getBean(ChatQueryRepository.class);
         UpdateWrapper<ChatQueryDO> updateWrapper = new UpdateWrapper<>();
-        updateWrapper.eq("question_id", chatQueryDO.getQuestionId());
+        updateWrapper.lambda().eq(ChatQueryDO::getQuestionId, chatQueryDO.getQuestionId());
         chatQueryRepository.updateChatQuery(chatQueryDO, updateWrapper);
     }
 }

headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/impl/DimensionRepositoryImpl.java

@@ -54,36 +54,39 @@ public class DimensionRepositoryImpl implements DimensionRepository {
     @Override
     public List<DimensionDO> getDimension(DimensionFilter dimensionFilter) {
         QueryWrapper<DimensionDO> queryWrapper = new QueryWrapper<>();
-        queryWrapper.ne("status", 3);
+        queryWrapper.lambda().ne(DimensionDO::getStatus, 3);
         if (Objects.nonNull(dimensionFilter.getIds()) && !dimensionFilter.getIds().isEmpty()) {
-            queryWrapper.in("id", dimensionFilter.getIds());
+            queryWrapper.lambda().in(DimensionDO::getId, dimensionFilter.getIds());
         }
         if (StringUtils.isNotBlank(dimensionFilter.getId())) {
-            queryWrapper.eq("id", dimensionFilter.getId());
+            queryWrapper.lambda().eq(DimensionDO::getId, dimensionFilter.getId());
         }
         if (Objects.nonNull(dimensionFilter.getModelIds())
                 && !dimensionFilter.getModelIds().isEmpty()) {
-            queryWrapper.in("model_id", dimensionFilter.getModelIds());
+            queryWrapper.lambda().in(DimensionDO::getModelId, dimensionFilter.getModelIds());
         }
         if (StringUtils.isNotBlank(dimensionFilter.getName())) {
-            queryWrapper.like("name", dimensionFilter.getName());
+            queryWrapper.lambda().like(DimensionDO::getName, dimensionFilter.getName());
         }
         if (StringUtils.isNotBlank(dimensionFilter.getId())) {
-            queryWrapper.like("biz_name", dimensionFilter.getBizName());
+            queryWrapper.lambda().like(DimensionDO::getBizName, dimensionFilter.getBizName());
         }
         if (Objects.nonNull(dimensionFilter.getStatus())) {
-            queryWrapper.eq("status", dimensionFilter.getStatus());
+            queryWrapper.lambda().eq(DimensionDO::getStatus, dimensionFilter.getStatus());
         }
         if (Objects.nonNull(dimensionFilter.getSensitiveLevel())) {
-            queryWrapper.eq("sensitive_level", dimensionFilter.getSensitiveLevel());
+            queryWrapper.lambda().eq(DimensionDO::getSensitiveLevel,
+                    dimensionFilter.getSensitiveLevel());
         }
         if (StringUtils.isNotBlank(dimensionFilter.getCreatedBy())) {
-            queryWrapper.eq("created_by", dimensionFilter.getCreatedBy());
+            queryWrapper.lambda().eq(DimensionDO::getCreatedBy, dimensionFilter.getCreatedBy());
         }
         if (StringUtils.isNotBlank(dimensionFilter.getKey())) {
             String key = dimensionFilter.getKey();
-            queryWrapper.like("name", key).or().like("biz_name", key).or().like("description", key)
+            queryWrapper.lambda().like(DimensionDO::getName, key).or()
-                    .or().like("alias", key).or().like("created_by", key);
+                    .like(DimensionDO::getBizName, key).or().like(DimensionDO::getDescription, key)
+                    .or().like(DimensionDO::getAlias, key).or()
+                    .like(DimensionDO::getCreatedBy, key);
         }
 
         return dimensionDOMapper.selectList(queryWrapper);

headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/impl/MetricRepositoryImpl.java

@@ -76,42 +76,42 @@ public class MetricRepositoryImpl implements MetricRepository {
     @Override
     public List<MetricDO> getMetric(MetricFilter metricFilter) {
         QueryWrapper<MetricDO> queryWrapper = new QueryWrapper<>();
-        queryWrapper.ne("status", 3);
+        queryWrapper.lambda().ne(MetricDO::getStatus, 3);
         if (Objects.nonNull(metricFilter.getIds()) && !metricFilter.getIds().isEmpty()) {
-            queryWrapper.in("id", metricFilter.getIds());
+            queryWrapper.lambda().in(MetricDO::getId, metricFilter.getIds());
         }
         if (StringUtils.isNotBlank(metricFilter.getId())) {
-            queryWrapper.eq("id", metricFilter.getId());
+            queryWrapper.lambda().eq(MetricDO::getId, metricFilter.getId());
         }
         if (Objects.nonNull(metricFilter.getModelIds()) && !metricFilter.getModelIds().isEmpty()) {
-            queryWrapper.in("model_id", metricFilter.getModelIds());
+            queryWrapper.lambda().in(MetricDO::getModelId, metricFilter.getModelIds());
         }
         if (StringUtils.isNotBlank(metricFilter.getType())) {
-            queryWrapper.eq("type", metricFilter.getType());
+            queryWrapper.lambda().eq(MetricDO::getType, metricFilter.getType());
         }
         if (StringUtils.isNotBlank(metricFilter.getName())) {
-            queryWrapper.like("name", metricFilter.getName());
+            queryWrapper.lambda().like(MetricDO::getName, metricFilter.getName());
         }
         if (StringUtils.isNotBlank(metricFilter.getId())) {
-            queryWrapper.like("biz_name", metricFilter.getBizName());
+            queryWrapper.lambda().like(MetricDO::getBizName, metricFilter.getBizName());
         }
         if (Objects.nonNull(metricFilter.getStatus())) {
-            queryWrapper.eq("status", metricFilter.getStatus());
+            queryWrapper.lambda().eq(MetricDO::getStatus, metricFilter.getStatus());
         }
         if (Objects.nonNull(metricFilter.getSensitiveLevel())) {
-            queryWrapper.eq("sensitive_level", metricFilter.getSensitiveLevel());
+            queryWrapper.lambda().eq(MetricDO::getSensitiveLevel, metricFilter.getSensitiveLevel());
         }
         if (StringUtils.isNotBlank(metricFilter.getCreatedBy())) {
-            queryWrapper.eq("created_by", metricFilter.getCreatedBy());
+            queryWrapper.lambda().eq(MetricDO::getCreatedBy, metricFilter.getCreatedBy());
         }
         if (Objects.nonNull(metricFilter.getIsPublish()) && metricFilter.getIsPublish() == 1) {
-            queryWrapper.eq("is_publish", metricFilter.getIsPublish());
+            queryWrapper.lambda().eq(MetricDO::getIsPublish, metricFilter.getIsPublish());
         }
         if (StringUtils.isNotBlank(metricFilter.getKey())) {
             String key = metricFilter.getKey();
-            queryWrapper.like("name", key).or().like("biz_name", key).or().like("description", key)
+            queryWrapper.lambda().like(MetricDO::getName, key).or().like(MetricDO::getBizName, key)
-                    .or().like("alias", key).or().like("classifications", key).or()
+                    .or().like(MetricDO::getDescription, key).or().like(MetricDO::getAlias, key)
-                    .like("created_by", key);
+                    .or().like(MetricDO::getCreatedBy, key);
         }
 
         return metricDOMapper.selectList(queryWrapper);

headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/impl/QueryRuleRepositoryImpl.java

@@ -51,7 +51,7 @@ public class QueryRuleRepositoryImpl implements QueryRuleRepository {
 
         QueryWrapper<QueryRuleDO> wrapperSys = new QueryWrapper<>();
         // 返回系统设置的规则
-        wrapperSys.or().eq("priority", 0L);
+        wrapperSys.lambda().or().eq(QueryRuleDO::getPriority, 0L);
         List<QueryRuleDO> queryRuleDOListSys = mapper.selectList(wrapperSys);
 
         queryRuleDOList.addAll(queryRuleDOListSys);

headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/impl/StatRepositoryImpl.java

@@ -74,13 +74,13 @@ public class StatRepositoryImpl implements StatRepository {
     private List<QueryStatDO> getQueryStats(ItemUseReq itemUseReq) {
         QueryWrapper<QueryStatDO> queryWrapper = new QueryWrapper<>();
         if (Objects.nonNull(itemUseReq.getModelId())) {
-            queryWrapper.eq("model_id", itemUseReq.getModelId());
+            queryWrapper.lambda().eq(QueryStatDO::getModelId, itemUseReq.getModelId());
         }
         if (Objects.nonNull(itemUseReq.getModelIds()) && !itemUseReq.getModelIds().isEmpty()) {
-            queryWrapper.in("model_id", itemUseReq.getModelIds());
+            queryWrapper.lambda().in(QueryStatDO::getModelId, itemUseReq.getModelIds());
         }
         if (Objects.nonNull(itemUseReq.getMetric())) {
-            queryWrapper.like("metrics", itemUseReq.getMetric());
+            queryWrapper.lambda().like(QueryStatDO::getMetrics, itemUseReq.getMetric());
         }
 
         return statMapper.selectList(queryWrapper);