yosql-codegen/src/main/java/wtf/metio/yosql/codegen/logging/DelegatingLoggingGenerator.java

   1 /*
   2  * This file is part of yosql. It is subject to the license terms in the LICENSE file found in the top-level
   3  * directory of this distribution and at https://creativecommons.org/publicdomain/zero/1.0/. No part of yosql,
   4  * including this file, may be copied, modified, propagated, or distributed except according to the terms contained
   5  * in the LICENSE file.
   6  */
   7 package wtf.metio.yosql.codegen.logging;
   8
   9 import com.squareup.javapoet.CodeBlock;
  10 import com.squareup.javapoet.FieldSpec;
  11 import com.squareup.javapoet.TypeName;
  12 import wtf.metio.yosql.codegen.exceptions.CodeGenerationException;
  13 import wtf.metio.yosql.models.configuration.LoggingApis;
  14 import wtf.metio.yosql.models.immutables.LoggingConfiguration;
  15
  16 import java.util.Optional;
  17 import java.util.Set;
  18
  19 /**
  20  * Delegates logging statement generation to a set of {@link LoggingGenerator}s.
  21  */
  22 public final class DelegatingLoggingGenerator implements LoggingGenerator {
  23
  24     private final LoggingConfiguration loggingConfiguration;
  25     private final Set<LoggingGenerator> generators;
  26
  27     public DelegatingLoggingGenerator(
  28             final LoggingConfiguration loggingConfiguration,
  29             final Set<LoggingGenerator> generators) {
  30         this.loggingConfiguration = loggingConfiguration;
  31         this.generators = generators;
  32     }
  33
  34     @Override
  35     public boolean supports(final LoggingApis api) {
  36         return generators.stream().anyMatch(generator -> generator.supports(api));
  37     }
  38
  39     @Override
  40     public CodeBlock queryPicked(final String fieldName) {
  41         return log().queryPicked(fieldName);
  42     }
  43
  44     @Override
  45     public CodeBlock indexPicked(final String fieldName) {
  46         return log().indexPicked(fieldName);
  47     }
  48
  49     @Override
  50     public CodeBlock vendorQueryPicked(final String fieldName) {
  51         return log().vendorQueryPicked(fieldName);
  52     }
  53
  54     @Override
  55     public CodeBlock vendorIndexPicked(final String fieldName) {
  56         return log().vendorIndexPicked(fieldName);
  57     }
  58
  59     @Override
  60     public CodeBlock vendorDetected() {
  61         return log().vendorDetected();
  62     }
  63
  64     @Override
  65     public CodeBlock executingQuery() {
  66         return log().executingQuery();
  67     }
  68
  69     @Override
  70     public CodeBlock shouldLog() {
  71         return log().shouldLog();
  72     }
  73
  74     @Override
  75     public boolean isEnabled() {
  76         return log().isEnabled();
  77     }
  78
  79     @Override
  80     public Optional<FieldSpec> logger(final TypeName repoClass) {
  81         return log().logger(repoClass);
  82     }
  83
  84     @Override
  85     public CodeBlock entering(final String repository, final String method) {
  86         return log().entering(repository, method);
  87     }
  88
  89     private LoggingGenerator log() {
  90         return generators.stream()
  91                 .filter(generator -> generator.supports(loggingConfiguration.api()))
  92                 .findFirst()
  93                 .orElseThrow(() -> new CodeGenerationException("TODO: add error message for missing support for " + loggingConfiguration.api()));
  94     }
  95
  96 }