#919 - io.ebean package initial

src/main/java/io/ebeaninternal/server/loadcontext/DLoadBaseContext.java

@@ -0,0 +1,86 @@
+package io.ebeaninternal.server.loadcontext;
+
+import io.ebean.FetchConfig;
+import io.ebean.bean.ObjectGraphNode;
+import io.ebean.bean.PersistenceContext;
+import io.ebeaninternal.server.deploy.BeanDescriptor;
+import io.ebeaninternal.server.querydefn.OrmQueryProperties;
+
+/**
+ * Base class for Bean and BeanCollection loading (lazy loading and query join loading).
+ */
+public abstract class DLoadBaseContext {
+
+  protected final DLoadContext parent;
+
+  protected final BeanDescriptor<?> desc;
+
+  protected final String fullPath;
+
+  protected final OrmQueryProperties queryProps;
+
+  protected final boolean hitCache;
+
+  protected final String serverName;
+
+  protected final int firstBatchSize;
+
+  protected final int secondaryBatchSize;
+
+  protected final ObjectGraphNode objectGraphNode;
+
+  protected final boolean queryFetch;
+
+
+  public DLoadBaseContext(DLoadContext parent, BeanDescriptor<?> desc, String path, int defaultBatchSize, OrmQueryProperties queryProps) {
+
+    this.parent = parent;
+    this.serverName = parent.getEbeanServer().getName();
+    this.desc = desc;
+    this.queryProps = queryProps;
+    this.fullPath = parent.getFullPath(path);
+
+    this.hitCache = !parent.isExcludeBeanCache() && desc.isBeanCaching();
+
+    this.objectGraphNode = parent.getObjectGraphNode(path);
+
+    this.queryFetch = queryProps != null && queryProps.isQueryFetch();
+    this.firstBatchSize = initFirstBatchSize(defaultBatchSize, queryProps);
+    this.secondaryBatchSize = initSecondaryBatchSize(defaultBatchSize, firstBatchSize, queryProps);
+  }
+
+  private int initFirstBatchSize(int batchSize, OrmQueryProperties queryProps) {
+    if (queryProps == null) {
+      return batchSize;
+    }
+
+    int queryBatchSize = queryProps.getQueryFetchBatch();
+    if (queryBatchSize == -1) {
+      return batchSize;
+
+    } else if (queryBatchSize == 0) {
+      return 100;
+
+    } else {
+      return queryBatchSize;
+    }
+  }
+
+  private int initSecondaryBatchSize(int defaultBatchSize, int firstBatchSize, OrmQueryProperties queryProps) {
+    if (queryProps == null) {
+      return defaultBatchSize;
+    }
+    FetchConfig fetchConfig = queryProps.getFetchConfig();
+    if (fetchConfig.isQueryAll()) {
+      return firstBatchSize;
+    }
+
+    int lazyBatchSize = fetchConfig.getLazyBatchSize();
+    return (lazyBatchSize > 1) ? lazyBatchSize : defaultBatchSize;
+  }
+
+  protected PersistenceContext getPersistenceContext() {
+    return parent.getPersistenceContext();
+  }
+
+}