Easy-Es 框架深度集成-ORM 方式操作 Easysearch 的最佳实践

<dependencies>
  <dependency>
    <groupId>org.dromara.easy-es</groupId>
    <artifactId>easy-es-boot-starter</artifactId>
    <version>2.1.0-easysearch</version>
  </dependency>

  <dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
  </dependency>

  <dependency>
    <groupId>org.projectlombok</groupId>
    <artifactId>lombok</artifactId>
    <optional>true</optional>
  </dependency>
</dependencies>

easy-es:
  enable: true
  address: localhost:9200
  schema: http

  global-config:
    process-index-mode: smoothly
    print-dsl: false

    db-config:
      index-prefix: dev_
      map-underscore-to-camel-case: true

easy-es:
  username: admin
  password: your_password_here
  keep-alive-millis: 18000

  global-config:
    db-config:
      refresh-policy: wait_until
      field-strategy: not_empty
      enable-track-total-hits: true

@SpringBootApplication
@EsMapperScan("com.example.demo.mapper")
public class DemoApplication {
  public static void main(String[] args) {
    SpringApplication.run(DemoApplication.class, args);
  }
}

@Data
@Settings(shardsNum = 3, replicasNum = 1)
@IndexName(value = "alarm_log", keepGlobalPrefix = true)
public class AlarmLog {

  @IndexId(type = IdType.CUSTOMIZE)
  private String id;

  @IndexField(fieldType = FieldType.KEYWORD)
  private String service;

  @IndexField(fieldType = FieldType.KEYWORD)
  private String level; // INFO/WARN/ERROR

  @IndexField(fieldType = FieldType.DATE, dateFormat = "epoch_millis")
  private Long timestamp;

  @HighLight(mappingField = "highlightMessage", preTag = "<em>", postTag = "</em>")
  @IndexField(fieldType = FieldType.TEXT, analyzer = Analyzer.IK_MAX_WORD, searchAnalyzer = Analyzer.IK_SMART)
  private String message;

  @IndexField(fieldType = FieldType.KEYWORD)
  private List<String> tags;

  private String highlightMessage;
}

public interface AlarmLogMapper extends BaseEsMapper<AlarmLog> {
}

@Service
@RequiredArgsConstructor
public class AlarmLogService {

  private final AlarmLogMapper alarmLogMapper;

  public List<AlarmLog> search(String service, String keyword, Long start, Long end, int page, int size) {
    int safePage = Math.max(page, 1);
    int safeSize = Math.min(Math.max(size, 1), 500);
    int from = (safePage - 1) * safeSize;

    LambdaEsQueryWrapper<AlarmLog> wrapper = new LambdaEsQueryWrapper<>();

    if (StringUtils.hasText(service)) {
      wrapper.eq(AlarmLog::getService, service);
    }
    if (start != null) {
      wrapper.ge(AlarmLog::getTimestamp, start);
    }
    if (end != null) {
      wrapper.le(AlarmLog::getTimestamp, end);
    }
    if (StringUtils.hasText(keyword)) {
      wrapper.match(AlarmLog::getMessage, keyword);
    }

    wrapper.orderByDesc(AlarmLog::getTimestamp)
        .from(from)
        .size(safeSize);

    return alarmLogMapper.selectList(wrapper);
  }
}

@Service
@RequiredArgsConstructor
public class AlarmLogAggService {

  private final AlarmLogMapper alarmLogMapper;

  public Aggregations aggByLevelAndService(Long start, Long end) {
    LambdaEsQueryWrapper<AlarmLog> wrapper = new LambdaEsQueryWrapper<>();

    if (start != null) {
      wrapper.ge(AlarmLog::getTimestamp, start);
    }
    if (end != null) {
      wrapper.le(AlarmLog::getTimestamp, end);
    }

    wrapper.groupBy(false, AlarmLog::getLevel, AlarmLog::getService);

    SearchResponse response = alarmLogMapper.search(wrapper);
    return response.getAggregations();
  }
}

@Service
@RequiredArgsConstructor
public class AlarmLogSqlService {

  private final AlarmLogMapper alarmLogMapper;
  private static final Set<String> LEVEL_WHITELIST = Set.of("INFO", "WARN", "ERROR");

  public String queryBySql(String level) {
    String sql = "SELECT * FROM dev_alarm_log LIMIT 10";

    if (StringUtils.hasText(level)) {
      String upper = level.trim().toUpperCase();
      if (!LEVEL_WHITELIST.contains(upper)) {
        throw new IllegalArgumentException("invalid level");
      }
      sql = "SELECT * FROM dev_alarm_log WHERE level = '" + upper + "' LIMIT 10";
    }

    return alarmLogMapper.executeSQL(sql);
  }
}