Allow passing DataOutput instead of FSTWriter

Author: dungba88

lucene/core/src/java/org/apache/lucene/util/fst/FSTCompiler.java

@@ -249,23 +249,25 @@ public Builder<T> allowFixedLengthArcs(boolean allowFixedLengthArcs) {
      *
      * <p>Default = 15.
      *
-     * @deprecated use {@link #fstWriter(FSTWriter)} instead
+     * @deprecated use {@link #dataOutput(DataOutput)} instead
      */
     @Deprecated
     public Builder<T> bytesPageBits(int bytesPageBits) {
-      return fstWriter(new BytesStore(bytesPageBits));
+      this.fstWriter = new BytesStore(bytesPageBits);
+      return this;
     }
 
     /**
-     * Set the {@link FSTWriter} which is used for low-level writing of FST.
+     * Set the {@link DataOutput} which is used for low-level writing of FST.
      *
-     * <p>Note: Setting this will override the settings in {@link #bytesPageBits(int)}
+     * <p>Note: Setting this will override the settings in {@link #bytesPageBits(int)} as it will
+     * use {@link DataOutputFSTWriter}
      *
-     * @param fstWriter the FSTWriter
+     * @param dataOutput the DataOutput
      * @return this builder
      */
-    public Builder<T> fstWriter(FSTWriter fstWriter) {
-      this.fstWriter = fstWriter;
+    public Builder<T> dataOutput(DataOutput dataOutput) {
+      this.fstWriter = new DataOutputFSTWriter(dataOutput);
       return this;
     }
 

lucene/core/src/test/org/apache/lucene/util/fst/TestDataOutputFSTWriter.java

@@ -204,8 +204,7 @@ protected FSTCompiler.Builder<T> getFSTBuilder() {
       // as the byte array could already contain another FST bytes, we should get the current offset
       // to know where to start reading from
       this.previousOffset = baos.size();
-      return super.getFSTBuilder()
-          .fstWriter(new DataOutputFSTWriter(new OutputStreamDataOutput(baos)));
+      return super.getFSTBuilder().dataOutput(new OutputStreamDataOutput(baos));
     }
 
     @Override