Examples with Results org.jooq.Results used on opensource projects

Example 6 with Results

use of org.jooq.Results in project Almura by AlmuraDev.

the class ServerExchangeManager method loadExchanges.

/**
 * Exchange
 */
private void loadExchanges() {
    final Map<String, Exchange> exchanges = new HashMap<>();
    try (final DSLContext context = this.databaseManager.createContext(true)) {
        final Results results = ExchangeQueries.createFetchAllExchanges().build(context).keepStatement(false).fetchMany();
        results.forEach(result -> {
            for (Record record : result) {
                final String id = record.getValue(Axs.AXS.ID);
                final Timestamp created = record.getValue(Axs.AXS.CREATED);
                final UUID creator = SerializationUtil.uniqueIdFromBytes(record.getValue(Axs.AXS.CREATOR));
                final String name = record.getValue(Axs.AXS.NAME);
                final String permission = record.getValue(Axs.AXS.PERMISSION);
                final boolean isHidden = record.getValue(Axs.AXS.IS_HIDDEN);
                this.logger.info("Loaded exchange '{}' ({})", name, id);
                exchanges.put(id, new BasicExchange(id, created.toInstant(), creator, name, permission, isHidden));
            }
        });
    } catch (SQLException e) {
        e.printStackTrace();
    }
    this.scheduler.createTaskBuilder().execute(() -> {
        this.exchanges.clear();
        this.exchanges.putAll(exchanges);
        if (this.exchanges.isEmpty()) {
            final BasicExchange exchange = new BasicExchange("almura.exchange.global", Instant.now(), FeatureConstants.UNKNOWN_OWNER, "Global " + "Exchange", "almura.exchange.global", false);
            this.exchanges.put("almura.exchange.global", exchange);
            // Yes, I am purposely running this sync
            try (final DSLContext context1 = this.databaseManager.createContext(true)) {
                ExchangeQueries.createInsertExchange(exchange.getCreated(), exchange.getCreator(), exchange.getId(), exchange.getName(), exchange.getPermission(), exchange.isHidden()).build(context1).keepStatement(false).execute();
                this.logger.info("Loaded exchange '{}' ({})", exchange.getName(), exchange.getId());
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
        this.logger.info("Loaded [{}] exchange(s).", this.exchanges.size());
        this.exchanges.values().forEach(IngameFeature::syncCreatorNameToUniqueId);
        Sponge.getServer().getOnlinePlayers().forEach(this::syncExchangeRegistryTo);
    }).submit(this.container);
}

Example 7 with Results

use of org.jooq.Results in project Almura by AlmuraDev.

the class ServerTitleManager method loadTitles.

public boolean loadTitles() {
    this.availableTitles.clear();
    this.logger.info("Querying database for titles, please wait...");
    this.scheduler.createTaskBuilder().async().execute(() -> {
        try (final DSLContext context = this.databaseManager.createContext(true)) {
            final Results results = TitleQueries.createFetchAllTitles().build(context).keepStatement(false).fetchMany();
            final Map<String, Title> titles = new HashMap<>();
            results.forEach(result -> {
                for (Record record : result) {
                    final String id = record.getValue(com.almuradev.generated.title.tables.Title.TITLE.ID);
                    final Timestamp created = record.getValue(com.almuradev.generated.title.tables.Title.TITLE.CREATED);
                    final UUID creator = SerializationUtil.uniqueIdFromBytes(record.getValue(com.almuradev.generated.title.tables.Title.TITLE.CREATOR));
                    final String permission = record.getValue(com.almuradev.generated.title.tables.Title.TITLE.PERMISSION);
                    final boolean isHidden = record.getValue(com.almuradev.generated.title.tables.Title.TITLE.IS_HIDDEN);
                    final String content = record.getValue(com.almuradev.generated.title.tables.Title.TITLE.CONTENT);
                    titles.put(id, new Title(created, creator, id, permission, isHidden, content));
                }
            });
            this.scheduler.createTaskBuilder().execute(() -> {
                this.titles.clear();
                if (!titles.isEmpty()) {
                    this.titles.putAll(titles);
                }
                this.logger.info("Loaded [{}] title(s).", this.titles.size());
                // Re-send titles to everyone
                this.network.sendToAll(new ClientboundTitlesRegistryPacket(this.titles.isEmpty() ? null : new HashSet<>(this.titles.values())));
                // Re-calculate available titles
                Sponge.getServer().getOnlinePlayers().forEach(player -> {
                    this.calculateAvailableTitlesFor(player);
                    this.verifySelectedTitle(player, null);
                    final Set<Title> availableTitles = this.getAvailableTitlesFor(player).orElse(null);
                    this.network.sendTo(player, new ClientboundAvailableTitlesResponsePacket(availableTitles));
                });
                // Send all selected titles out again as we've verified and corrected them in-case load changed
                this.network.sendToAll(new ClientboundSelectedTitleBulkPacket(this.selectedTitles.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, v -> v.getValue().getId()))));
            }).submit(this.container);
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }).submit(this.container);
    return true;
}

Example 8 with Results

use of org.jooq.Results in project Almura by AlmuraDev.

the class ServerExchangeManager method handleModifyListItems.

public void handleModifyListItems(final Player player, final String id, final List<InventoryAction> actions) {
    checkNotNull(player);
    checkNotNull(id);
    checkNotNull(actions);
    checkState(!actions.isEmpty());
    final Exchange axs = this.getExchange(id).orElse(null);
    if (axs == null) {
        this.logger.error("Player '{}' attempted to list items for exchange '{}' but the server has no knowledge of it. Syncing exchange " + "registry...", player.getName(), id);
        this.syncExchangeRegistryTo(player);
        return;
    }
    final UUID seller = player.getUniqueId();
    final EntityPlayerMP serverPlayer = (EntityPlayerMP) player;
    final IItemHandler simulatedInventory = serverPlayer.getCapability(CapabilityItemHandler.ITEM_HANDLER_CAPABILITY, EnumFacing.UP);
    // Inventory -> Listing
    final List<InventoryAction> toListingActions = actions.stream().filter(a -> a.getDirection() == InventoryAction.Direction.TO_LISTING).collect(Collectors.toList());
    final List<VanillaStack> toListingStacks = new ArrayList<>();
    final List<VanillaStack> unknownInventoryStacks = new ArrayList<>();
    final List<VanillaStack> leftoverToListingStacks = new ArrayList<>();
    for (InventoryAction toListingAction : toListingActions) {
        final VanillaStack stack = toListingAction.getStack();
        int amountLeft = stack.getQuantity();
        boolean matched = false;
        for (int j = 0; j < simulatedInventory.getSlots(); j++) {
            final ItemStack slotStack = simulatedInventory.getStackInSlot(j);
            if (ItemHandlerHelper.canItemStacksStack(slotStack, stack.asRealStack())) {
                amountLeft -= simulatedInventory.extractItem(j, amountLeft, false).getCount();
                matched = true;
            }
            if (amountLeft <= 0) {
                break;
            }
        }
        if (!matched) {
            unknownInventoryStacks.add(stack);
        } else {
            if (amountLeft > 0) {
                final VanillaStack copyStack = stack.copy();
                copyStack.setQuantity(amountLeft);
                leftoverToListingStacks.add(copyStack);
            }
            final VanillaStack copyStack = stack.copy();
            copyStack.setQuantity(stack.getQuantity() - amountLeft);
            if (copyStack.getQuantity() != 0) {
                toListingStacks.add(copyStack);
            }
        }
    }
    // Listing -> Inventory
    final List<InventoryAction> toInventoryActions = actions.stream().filter(a -> a.getDirection() == InventoryAction.Direction.TO_INVENTORY).collect(Collectors.toList());
    final List<VanillaStack> listingNotFoundStacks = new ArrayList<>();
    final List<ListItem> desyncToInventoryStacks = new ArrayList<>();
    final List<ListItem> toInventoryStacks = new ArrayList<>();
    final List<ListItem> partialToInventoryStacks = new ArrayList<>();
    final List<ListItem> currentListItems = axs.getListItemsFor(player.getUniqueId()).orElse(null);
    if (!toInventoryActions.isEmpty()) {
        if (currentListItems == null || currentListItems.isEmpty()) {
            this.logger.error("Player '{}' attempted to move listings back to the inventory for exchange '{}' but the server knows of no " + "listings for them. This could be a de-sync or an exploit. Printing stacks...", player.getName(), axs.getId());
            this.network.sendTo(player, new ClientboundListItemsResponsePacket(axs.getId(), null));
            this.printStacksToConsole(toInventoryActions.stream().map(InventoryAction::getStack).collect(Collectors.toList()));
        } else {
            for (final InventoryAction action : toInventoryActions) {
                final VanillaStack stack = action.getStack();
                ListItem found = null;
                for (final ListItem listItem : currentListItems) {
                    if (ItemHandlerHelper.canItemStacksStack(stack.asRealStack(), listItem.asRealStack())) {
                        found = listItem;
                        break;
                    }
                }
                // Unknown listing
                if (found == null) {
                    listingNotFoundStacks.add(stack);
                    continue;
                }
                ListItem toRemove = found.copy();
                // Listing quantity mismatch (tracking this to let the user know)
                if (found.getQuantity() < stack.getQuantity()) {
                    desyncToInventoryStacks.add(found);
                    toRemove.setQuantity(found.getQuantity());
                } else {
                    toRemove.setQuantity(stack.getQuantity());
                }
                final ItemStack resultStack = ItemHandlerHelper.insertItemStacked(simulatedInventory, toRemove.asRealStack(), true);
                // Simulated a partial stack insertion
                if (!resultStack.isEmpty()) {
                    final ListItem copyStack = toRemove.copy();
                    copyStack.setQuantity(resultStack.getCount());
                    partialToInventoryStacks.add(copyStack);
                }
                final ListItem copyStack = toRemove.copy();
                copyStack.setQuantity(toRemove.getQuantity() - resultStack.getCount());
                toInventoryStacks.add(copyStack);
            }
        }
    }
    // This may seem quite weird but we need to clear out the listing items reference for this player across the board to await what the results
    // are from the database to ensure we're 1:1 in sync. Otherwise, the Exchange would keep selling..
    axs.putListItemsFor(seller, null);
    axs.putForSaleItemsFor(seller, null);
    Sponge.getServer().getOnlinePlayers().stream().filter(p -> p.getUniqueId() != seller).forEach(p -> this.network.sendTo(p, new ClientboundForSaleFilterRequestPacket(axs.getId())));
    this.scheduler.createTaskBuilder().async().execute(() -> {
        try (final DSLContext context = this.databaseManager.createContext(true)) {
            final Iterator<VanillaStack> listingIter = toListingStacks.iterator();
            int index = 0;
            // New listing
            while (listingIter.hasNext()) {
                final VanillaStack stack = listingIter.next();
                final ItemStack realStack = stack.asRealStack();
                ListItem found = null;
                if (currentListItems != null) {
                    found = currentListItems.stream().filter(item -> ItemHandlerHelper.canItemStacksStack(realStack, item.asRealStack())).findAny().orElse(null);
                }
                if (found == null) {
                    final AxsListItemRecord itemRecord = ExchangeQueries.createInsertListItem(axs.getId(), Instant.now(), seller, realStack.getItem(), stack.getQuantity(), realStack.getMetadata(), index).build(context).fetchOne();
                    if (itemRecord == null) {
                        this.notificationManager.sendWindowMessage(player, Text.of("Exchange"), Text.of("Critical error encountered, check the server console for more details!"));
                        this.logger.error("Player '{}' submitted a new list item for exchange '{}' to the database but it failed. " + "Discarding changes and printing stack...", player.getName(), id);
                        this.printStacksToConsole(Lists.newArrayList(stack));
                        continue;
                    }
                    final NBTTagCompound compound = stack.getCompound();
                    if (compound == null) {
                        continue;
                    }
                    final AxsListItemDataRecord dataRecord = ExchangeQueries.createInsertItemData(itemRecord.getRecNo(), compound).build(context).fetchOne();
                    if (dataRecord == null) {
                        this.notificationManager.sendWindowMessage(player, Text.of("Exchange"), Text.of("Critical error encountered, check the server console for more details!"));
                        this.logger.error("Player '{}' submitted data for item record '{}' for exchange '{}' but it failed. " + "Discarding changes...", player.getName(), itemRecord.getRecNo(), id);
                        ExchangeQueries.createDeleteListItem(itemRecord.getRecNo()).build(context).execute();
                    }
                } else {
                    final int result = ExchangeQueries.createUpdateListItem(found.getRecord(), stack.getQuantity() + found.getQuantity(), index).build(context).execute();
                    if (result == 0) {
                        this.notificationManager.sendWindowMessage(player, Text.of("Exchange"), Text.of("Critical error encountered, check the server console for more details!"));
                        this.logger.error("Player '{}' attempted to add quantity to list item '{}' for exchange '{}' but it failed. " + "Discarding changes...", player.getName(), found.getRecord(), id);
                        listingIter.remove();
                    }
                }
                index++;
            }
            for (final ListItem next : toInventoryStacks) {
                ListItem existingStack = null;
                for (final ListItem stack : currentListItems) {
                    if (next.getRecord() == stack.getRecord()) {
                        existingStack = stack;
                        break;
                    }
                }
                final int diff = existingStack.getQuantity() - next.getQuantity();
                if (diff == 0) {
                    final int result = ExchangeQueries.createUpdateListItemIsHidden(next.getRecord(), true).build(context).execute();
                    if (result == 0) {
                        this.notificationManager.sendWindowMessage(player, Text.of("Exchange"), Text.of("Critical error encountered, check the server console for more details!"));
                        this.logger.error("Player '{}' attempted to remove list item '{}' for exchange '{}' but it failed. Discarding " + "changes...", player.getName(), next.getRecord(), id);
                    }
                // Update partial listings
                } else {
                    final int result = ExchangeQueries.createUpdateListItem(next.getRecord(), diff, next.getIndex()).build(context).execute();
                    if (result == 0) {
                        this.notificationManager.sendWindowMessage(player, Text.of("Exchange"), Text.of("Critical error encountered, check the server console for more details!"));
                        this.logger.error("Player '{}' removed quantity from list item '{}' for exchange '{}' to the database but it " + "failed. Discarding changes...", player.getName(), next.getRecord(), id);
                    }
                }
            }
            final Results listItemResults = ExchangeQueries.createFetchListItemsAndDataFor(seller, false).build(context).keepStatement(false).fetchMany();
            final Results forSaleItemResults = ExchangeQueries.createFetchForSaleItemsFor(seller, false).build(context).keepStatement(false).fetchMany();
            this.scheduler.createTaskBuilder().execute(() -> {
                final Player sellerPlayer = Sponge.getServer().getPlayer(seller).orElse(null);
                if (sellerPlayer != null) {
                    final IItemHandler inventory = ((EntityPlayerMP) sellerPlayer).getCapability(CapabilityItemHandler.ITEM_HANDLER_CAPABILITY, EnumFacing.UP);
                    // Add stacks from listings
                    for (final ListItem stack : toInventoryStacks) {
                        final ItemStack resultStack = ItemHandlerHelper.insertItemStacked(inventory, stack.asRealStack(), false);
                        if (!resultStack.isEmpty()) {
                        // TODO Their inventory changed since simulation. Best case scenario we toss it on the ground
                        }
                    }
                } else {
                // TODO They went offline on us. It is a very rare off-case. Half tempted to print what they should have got and let
                // TODO an admin deal with it
                }
                final List<ListItem> listItems = new ArrayList<>();
                listItemResults.forEach(result -> listItems.addAll(this.parseListItemsFrom(result)));
                final List<ForSaleItem> forSaleItems = new ArrayList<>();
                forSaleItemResults.forEach(result -> forSaleItems.addAll(this.parseForSaleItemsFrom(listItems, result)));
                // TODO Build a notification that says...
                // TODO  - Stacks requested to go to a listing but inventory can't fulfill it
                // TODO  - Stacks requested to go to the inventory but listings aren't found
                // TODO  - Stacks requested to go to the inventory but the listing couldn't fulfill it so we took what we could
                axs.putListItemsFor(seller, listItems);
                axs.putForSaleItemsFor(seller, forSaleItems);
                if (sellerPlayer != null) {
                    this.network.sendTo(sellerPlayer, new ClientboundListItemsResponsePacket(axs.getId(), listItems));
                    this.network.sendTo(sellerPlayer, new ClientboundListItemsSaleStatusPacket(axs.getId(), forSaleItems, null));
                }
                Sponge.getServer().getOnlinePlayers().forEach(p -> this.network.sendTo(p, new ClientboundForSaleFilterRequestPacket(axs.getId())));
            }).submit(this.container);
        } catch (SQLException | IOException e) {
            e.printStackTrace();
        }
    }).submit(this.container);
}

Example 9 with Results

use of org.jooq.Results in project Almura by AlmuraDev.

the class ServerExchangeManager method loadListItems.

/**
 * ListItem
 */
private void loadListItems(final Exchange axs) {
    checkNotNull(axs);
    this.logger.info("Querying list items for exchange '{}' ({}), please wait...", axs.getName(), axs.getId());
    final List<ListItem> items = new ArrayList<>();
    try (final DSLContext context = this.databaseManager.createContext(true)) {
        final Results results = ExchangeQueries.createFetchListItemsAndDataFor(axs.getId(), false).build(context).keepStatement(false).fetchMany();
        results.forEach(result -> items.addAll(this.parseListItemsFrom(result)));
        axs.clearListItems();
        axs.clearForSaleItems();
        final Map<UUID, List<ListItem>> itemsByOwner = items.stream().collect(Collectors.groupingBy(ListItem::getSeller));
        itemsByOwner.forEach((key, value) -> value.sort(Comparator.comparingInt(ListItem::getIndex)));
        axs.putListItems(itemsByOwner.isEmpty() ? null : itemsByOwner);
        this.logger.info("Loaded [{}] list item(s) for exchange '{}' ({}).", items.size(), axs.getName(), axs.getId());
    } catch (SQLException e) {
        e.printStackTrace();
    }
}

Example 10 with Results

use of org.jooq.Results in project Almura by AlmuraDev.

the class ServerExchangeManager method loadForSaleItems.

/**
 * ForSaleItem
 */
private void loadForSaleItems(final Exchange axs) {
    checkNotNull(axs);
    this.logger.info("Querying for sale items for exchange '{}' ({}), please wait...", axs.getName(), axs.getId());
    final List<ForSaleItem> forSaleItems = new ArrayList<>();
    final List<ListItem> listItems = axs.getListItems().entrySet().stream().map(Map.Entry::getValue).flatMap(List::stream).collect(Collectors.toList());
    try (final DSLContext context = this.databaseManager.createContext(true)) {
        final Results results = ExchangeQueries.createFetchForSaleItemsFor(axs.getId(), false).build(context).keepStatement(false).fetchMany();
        results.forEach(result -> forSaleItems.addAll(this.parseForSaleItemsFrom(listItems, result)));
        axs.clearForSaleItems();
        axs.putForSaleItems(forSaleItems.isEmpty() ? null : forSaleItems.stream().collect(Collectors.groupingBy(forSaleItem -> forSaleItem.getListItem().getSeller())));
        this.logger.info("Loaded [{}] for sale item(s) for exchange '{}' ({}).", forSaleItems.size(), axs.getName(), axs.getId());
    } catch (SQLException e) {
        e.printStackTrace();
    }
}