Examples with DrServerList io.datarouter.client.hbase.cluster.DrServerList used on opensource projects

Example 1 with DrServerList

use of io.datarouter.client.hbase.cluster.DrServerList in project datarouter by hotpads.

the class HBaseRegionBalancerJob method balanceClient.

// return true if completed balancing
private boolean balanceClient(TaskTracker tracker, ClientId clientId) {
    logger.warn("calculating region movements for client {}", clientId.getName());
    Admin admin = hBaseClientManager.getAdmin(clientId);
    List<HBaseRegionMovement> movements = new ArrayList<>();
    DrServerList serverList = new DrServerList(admin);
    List<String> tableNames = nodes.getTableNamesForClient(clientId.getName());
    Collections.sort(tableNames);
    int tableCounter = 0;
    for (String tableName : tableNames) {
        if (tracker.shouldStop()) {
            return false;
        }
        ++tableCounter;
        PhaseTimer timer = new PhaseTimer("generating movements for table " + tableName + " #" + tableCounter + "/" + tableNames.size());
        List<HBaseRegionMovement> tableMovements = new ArrayList<>();
        PhysicalNode<?, ?, ?> physicalNodeForTable = nodes.getPhysicalNodeForClientAndTable(clientId.getName(), tableName);
        BaseHBaseRegionBalancer balancer = balancerFactory.getBalancerForTable(clientId, tableName);
        DrRegionList regionList;
        try {
            regionList = drRegionListFactory.make(clientId, serverList, tableName, physicalNodeForTable, balancer);
        } catch (Exception e) {
            logger.error("skipping table " + tableName, e);
            continue;
        }
        for (DrRegionInfo<?> region : regionList.getRegions()) {
            if (region.isNotOnAnyServer()) {
                logger.warn("region {} is not currently hosted, so not attempting to move it", region.getRegion().getRegionNameAsString());
                continue;
            }
            if (region.isOnCorrectServer()) {
                continue;
            }
            HBaseRegionMovement movement = new HBaseRegionMovement(tableName, region.getRegion().getEncodedName(), region.getHBaseServerName(), region.getBalancerDestinationHBaseServerName());
            tableMovements.add(movement);
        }
        timer.add("generated " + tableMovements.size() + " movements");
        logger.warn(timer.toString());
        movements.addAll(tableMovements);
    }
    // we calculated all the movements.  now move them
    long iterationStartTimeMs = System.currentTimeMillis();
    Map<ServerName, List<HBaseRegionMovement>> movementsByCurrentServer = HBaseRegionMovement.getByCurrentServer(movements);
    int clusterMovementCounter = 0;
    logger.warn("processing {} total movements", movements.size());
    for (DrServerInfo serverInfo : serverList.getServersSortedByDescendingLoad()) {
        List<HBaseRegionMovement> movementsForServer = movementsByCurrentServer.getOrDefault(serverInfo.getServerName(), List.of());
        logger.warn("expecting {} movements for server {}", movementsForServer.size(), serverInfo.getServerName());
    }
    for (DrServerInfo serverInfo : serverList.getServersSortedByDescendingLoad()) {
        List<HBaseRegionMovement> movementsForServer = movementsByCurrentServer.getOrDefault(serverInfo.getServerName(), List.of());
        logger.warn("processing {} movements for server {}", movementsForServer.size(), serverInfo.getServerName());
        int serverMovementCounter = 0;
        for (HBaseRegionMovement movement : movementsForServer) {
            ++clusterMovementCounter;
            ++serverMovementCounter;
            logger.warn("moving region {}/{} ({}/{}): {}", serverMovementCounter, movementsForServer.size(), clusterMovementCounter, movements.size(), movement);
            try {
                admin.move(movement.getRegionNameBytes(), movement.getDestinationServerNameBytes());
            } catch (Exception ex) {
                // in 0.94, this is UndeclaredThrowableException wrapping other Exceptions
                logger.error("exception moving region, skipping", ex);
            }
            ThreadTool.sleepUnchecked(hbaseSettings.getSleepBetweenRegionMovementMs());
            if (tracker.increment().shouldStop()) {
                return false;
            }
            if (System.currentTimeMillis() - iterationStartTimeMs > RECALCULATE_AFTER_MS) {
                logger.warn("suspending to check for new servers", RECALCULATE_AFTER_MS);
                return false;
            }
        }
        logger.warn("processed {} movements for server {}", serverMovementCounter, serverInfo.getServerName());
    }
    return true;
}

Example 2 with DrServerList

use of io.datarouter.client.hbase.cluster.DrServerList in project datarouter by hotpads.

the class HBaseHandler method initialize.

/*---------------------------- useful methods ----------------------------*/
private void initialize() {
    mav = new Mav();
    encodedRegionNameStrings = RequestTool.getCheckedBoxes(request, PARAM_PREFIX_encodedRegionName_);
    numRegions = encodedRegionNameStrings.size();
    datarouterWebRequestParams = datarouterWebRequestParamsFactory.new DatarouterWebRequestParams<>(params, getClientType());
    mav.put("clientType", datarouterWebRequestParams.getClientType().getName());
    drServerList = SingletonSupplier.of(() -> new DrServerList(hBaseClientManager.getAdmin(datarouterWebRequestParams.getClientId())));
    regionList = SingletonSupplier.of(() -> drRegionListFactory.make(datarouterWebRequestParams.getClientId(), drServerList.get(), datarouterWebRequestParams.getTableName(), datarouterWebRequestParams.getPhysicalNode(), balancerFactory.getBalancerForTable(datarouterWebRequestParams.getClientId(), datarouterWebRequestParams.getTableName())));
}

Example 3 with DrServerList

use of io.datarouter.client.hbase.cluster.DrServerList in project datarouter by hotpads.

the class HBaseCompactionJob method run.

@Override
public void run(TaskTracker tracker) {
    for (ClientId clientId : compactionInfo.getManagedClientIds()) {
        Admin admin = hBaseClientManager.getAdmin(clientId);
        DrServerList servers = new DrServerList(admin);
        List<String> tableNames = nodes.getTableNamesForClient(clientId.getName());
        for (String tableName : tableNames) {
            if (tracker.heartbeat(numRegions.sum()).shouldStop()) {
                return;
            }
            try {
                compactTable(tracker, admin, servers, clientId, tableName);
            } catch (Exception e) {
                logger.warn("error compacting {}.{}", clientId, tableName, e);
            }
        }
    }
    logger.warn("numTables={}, numRegions={}, numMissingLoad={}, numAlreadyLocal={}, numTriggered={}", numTables, numRegions, numMissingLoad, numAlreadyLocal, numTriggered);
}