-
Notifications
You must be signed in to change notification settings - Fork 34
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #2234 from subutai-io/base-2226
Base 2226
- Loading branch information
Showing
5 changed files
with
134 additions
and
7 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
110 changes: 110 additions & 0 deletions
110
...pl/src/main/java/io/subutai/core/hubmanager/impl/requestor/ContainerDiskUsageChecker.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,110 @@ | ||
| package io.subutai.core.hubmanager.impl.requestor; | ||
|
|
||
|
|
||
| import org.slf4j.Logger; | ||
| import org.slf4j.LoggerFactory; | ||
|
|
||
| import io.subutai.common.command.CommandResult; | ||
| import io.subutai.common.command.CommandUtil; | ||
| import io.subutai.common.command.RequestBuilder; | ||
| import io.subutai.common.environment.ContainerDto; | ||
| import io.subutai.common.environment.EnvironmentDto; | ||
| import io.subutai.common.peer.ContainerHost; | ||
| import io.subutai.common.peer.LocalPeer; | ||
| import io.subutai.common.settings.Common; | ||
| import io.subutai.core.environment.api.EnvironmentManager; | ||
| import io.subutai.core.hubmanager.api.HubManager; | ||
| import io.subutai.core.hubmanager.api.HubRequester; | ||
| import io.subutai.core.hubmanager.api.RestClient; | ||
| import io.subutai.core.hubmanager.api.RestResult; | ||
|
|
||
|
|
||
| //https://github.com/subutai-io/agent/wiki/Switch-to-Soft-Quota | ||
| public class ContainerDiskUsageChecker extends HubRequester | ||
| { | ||
| private final static Logger LOG = LoggerFactory.getLogger( ContainerDiskUsageChecker.class ); | ||
|
|
||
| private final EnvironmentManager environmentManager; | ||
| private final LocalPeer localPeer; | ||
| private final CommandUtil commandUtil = new CommandUtil(); | ||
|
|
||
|
|
||
| public ContainerDiskUsageChecker( final HubManager hubManager, final RestClient restClient, | ||
| final EnvironmentManager environmentManager, final LocalPeer localPeer ) | ||
| { | ||
| super( hubManager, restClient ); | ||
| this.environmentManager = environmentManager; | ||
| this.localPeer = localPeer; | ||
| } | ||
|
|
||
|
|
||
| @Override | ||
| public void request() throws Exception | ||
| { | ||
| for ( EnvironmentDto environment : environmentManager.getTenantEnvironments() ) | ||
| { | ||
| //iterate over hub containers | ||
| if ( Common.HUB_ID.equals( environment.getDataSource() ) ) | ||
| { | ||
| for ( ContainerDto container : environment.getContainers() ) | ||
| { | ||
| checkDiskUsage( container ); | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
|
|
||
| private void checkDiskUsage( ContainerDto containerDto ) | ||
| { | ||
| // a. get its disk usage -> "subutai info du foo" | ||
| // b. compare with container size disk quota limit | ||
| // b.a if du is >= 90 % of quota -> notify Hub | ||
| // b.b if du is >= 150 % of quota -> stop container, notify Hub | ||
| try | ||
| { | ||
| ContainerHost containerHost = localPeer.getContainerHostById( containerDto.getId() ); | ||
|
|
||
| CommandResult result = commandUtil | ||
| .execute( new RequestBuilder( "subutai info du " + containerDto.getContainerName() ), | ||
| containerHost ); | ||
|
|
||
| long diskUsed = Long.parseLong( result.getStdOut() ); | ||
|
|
||
| long diskLimit = containerHost.getContainerSize().getDiskQuota().longValue(); | ||
|
|
||
| if ( diskUsed >= diskLimit * 0.9 ) | ||
| { | ||
| boolean stop = diskUsed >= diskLimit * 1.5; | ||
|
|
||
| if ( stop ) | ||
| { | ||
| //stop container | ||
| containerHost.stop(); | ||
| } | ||
|
|
||
| //notify Hub | ||
| notifyHub( containerDto.getPeerId(), containerDto.getEnvironmentId(), containerDto.getId(), diskUsed, | ||
| stop ); | ||
| } | ||
| } | ||
| catch ( Exception e ) | ||
| { | ||
| LOG.error( "Error checking disk usage of container " + containerDto.getContainerName(), e.getMessage() ); | ||
| } | ||
| } | ||
|
|
||
|
|
||
| private void notifyHub( String peerId, String envId, String contId, long diskUsage, boolean containerWasStopped ) | ||
| { | ||
| RestResult result = restClient.post( String | ||
| .format( "/rest/v1/peers/%s/environments/%s/containers/%s/disk_usage/%d/%s", peerId, envId, contId, | ||
| diskUsage, containerWasStopped ), null ); | ||
|
|
||
| if ( !result.isSuccess() ) | ||
| { | ||
| LOG.error( "Error notifying Hub about container disk usage excess: HTTP {} - {}", result.getStatus(), | ||
| result.getReasonPhrase() ); | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters