Take another approach for the dict trainer min samples

Author: luben

src/main/java/com/github/luben/zstd/Zstd.java

@@ -708,7 +708,13 @@ public static long decompressedDirectByteBufferSize(ByteBuffer src, int srcPosit
      * @return the number of bytes into buffer 'dictBuffer' or an error code if
      *          it fails (which can be tested using ZSTD_isError())
      */
-    public static native long trainFromBuffer(byte[][] samples, byte[] dictBuffer, boolean legacy);
+    public static long trainFromBuffer(byte[][] samples, byte[] dictBuffer, boolean legacy) {
+        if (samples.length <= 10) {
+            throw new ZstdException(Zstd.errGeneric(), "nb of samples too low");
+        }
+        return trainFromBuffer0(samples, dictBuffer, legacy);
+    }
+    private static native long trainFromBuffer0(byte[][] samples, byte[] dictBuffer, boolean legacy);
 
     /**
      * Creates a new dictionary to tune a kind of samples
@@ -720,7 +726,13 @@ public static long decompressedDirectByteBufferSize(ByteBuffer src, int srcPosit
      * @return the number of bytes into buffer 'dictBuffer' or an error code if
      *          it fails (which can be tested using ZSTD_isError())
      */
-    public static native long trainFromBufferDirect(ByteBuffer samples, int[] sampleSizes, ByteBuffer dictBuffer, boolean legacy);
+    public static long trainFromBufferDirect(ByteBuffer samples, int[] sampleSizes, ByteBuffer dictBuffer, boolean legacy) {
+        if (sampleSizes.length <= 10) {
+            throw new ZstdException(Zstd.errGeneric(), "nb of samples too low");
+        }
+        return trainFromBufferDirect0(samples, sampleSizes, dictBuffer, legacy);
+    }
+    private static native long trainFromBufferDirect0(ByteBuffer samples, int[] sampleSizes, ByteBuffer dictBuffer, boolean legacy);
 
     /**
      * Get DictId from a compressed frame

src/main/native/jni_zdict.c

@@ -7,15 +7,10 @@
 #include <stdlib.h>
 #include <string.h>
 
-JNIEXPORT jlong Java_com_github_luben_zstd_Zstd_trainFromBuffer
+JNIEXPORT jlong Java_com_github_luben_zstd_Zstd_trainFromBuffer0
   (JNIEnv *env, jclass obj, jobjectArray samples, jbyteArray dictBuffer, jboolean legacy) {
     size_t size = 0;
     jsize num_samples = (*env)->GetArrayLength(env, samples);
-    if (num_samples <= 10) {
-        jclass eClass = (*env)->FindClass(env, "Ljava/lang/RuntimeException;");
-        (*env)->ThrowNew(env, eClass, "nb of samples too low");
-        goto E1;
-    }
     size_t *samples_sizes = malloc(sizeof(size_t) * num_samples);
     if (!samples_sizes) {
         jclass eClass = (*env)->FindClass(env, "Ljava/lang/OutOfMemoryError;");
@@ -60,7 +55,7 @@ E2: free(samples_sizes);
 E1: return size;
 }
 
-JNIEXPORT jlong Java_com_github_luben_zstd_Zstd_trainFromBufferDirect
+JNIEXPORT jlong Java_com_github_luben_zstd_Zstd_trainFromBufferDirect0
   (JNIEnv *env, jclass obj, jobject samples, jintArray sampleSizes, jobject dictBuffer, jboolean legacy) {
 
     size_t size = 0;
@@ -70,11 +65,6 @@ JNIEXPORT jlong Java_com_github_luben_zstd_Zstd_trainFromBufferDirect
 
     /* convert sized from int to size_t */
     jsize num_samples = (*env)->GetArrayLength(env, sampleSizes);
-    if (num_samples <= 10) {
-        jclass eClass = (*env)->FindClass(env, "Ljava/lang/RuntimeException;");
-        (*env)->ThrowNew(env, eClass, "nb of samples too low");
-        goto E1;
-    }
     size_t *samples_sizes = malloc(sizeof(size_t) * num_samples);
     if (!samples_sizes) {
         jclass eClass = (*env)->FindClass(env, "Ljava/lang/OutOfMemoryError;");