Start line:  
End line:  

Snippet Preview

Snippet HTML Code

Stack Overflow Questions
 /*
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *     http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.facebook.presto.memory;
import static io.airlift.units.DataSize.Unit.BYTE;
import static java.util.Objects.requireNonNull;
public final class LocalMemoryManager
    public static final MemoryPoolId GENERAL_POOL = new MemoryPoolId("general");
    public static final MemoryPoolId RESERVED_POOL = new MemoryPoolId("reserved");
    private final DataSize maxMemory;
    private final MBeanExporter exporter;
    private final Map<MemoryPoolIdMemoryPoolpools;
    @Inject
    public LocalMemoryManager(MemoryManagerConfig configReservedSystemMemoryConfig systemMemoryConfigMBeanExporter exporter)
    {
        this. = requireNonNull(exporter"exporter is null");
        requireNonNull(config"config is null");
        requireNonNull(systemMemoryConfig"systemMemoryConfig is null");
         = new DataSize(Runtime.getRuntime().maxMemory() - systemMemoryConfig.getReservedSystemMemory().toBytes(), );
        ImmutableMap.Builder<MemoryPoolIdMemoryPoolbuilder = ImmutableMap.builder();
        builder.put(createPool(config.getMaxQueryMemoryPerNode(), exporter));
        DataSize generalPoolSize = new DataSize(.toBytes() - config.getMaxQueryMemoryPerNode().toBytes(), );
        builder.put(createPool(generalPoolSizeexporter));
        this. = builder.build();
    }
    public MemoryInfo getInfo()
    {
        ImmutableMap.Builder<MemoryPoolIdMemoryPoolInfobuilder = ImmutableMap.builder();
        for (Map.Entry<MemoryPoolIdMemoryPoolentry : .entrySet()) {
            builder.put(entry.getKey(), entry.getValue().getInfo());
        }
        return new MemoryInfo(builder.build());
    }
    public MemoryPool getPool(MemoryPoolId id)
    {
        return .get(id);
    }
    private static MemoryPool createPool(MemoryPoolId idDataSize sizeMBeanExporter exporter)
    {
        // TODO: Once these pools are integrated with the coordinator, we should enable blocking
        MemoryPool pool = new MemoryPool(idsizefalse);
        String objectName = ObjectNames.builder(MemoryPool.classpool.getId().toString()).build();
        exporter.export(objectNamepool);
        return pool;
    }
    @PreDestroy
    public void destroy()
    {
        for (MemoryPool pool : .values()) {
            String objectName = ObjectNames.builder(MemoryPool.classpool.getId().toString()).build();
            .unexport(objectName);
        }
    }
New to GrepCode? Check out our FAQ X