one.xingyi.profile.Profile Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of profiling Show documentation
There is a newer version: 1.5.9
package one.xingyi.profile;

import lombok.RequiredArgsConstructor;
import one.xingyi.helpers.MapHelpers;
import one.xingyi.interfaces.INanoTime;
import one.xingyi.interfaces.RunnableWithExceptionE;
import one.xingyi.interfaces.SupplierWithExceptionE;
import one.xingyi.profile.pathmap.IPathMap;

import javax.management.MBeanServer;
import javax.management.ObjectName;
import java.lang.management.ManagementFactory;
import java.util.HashMap;
import java.util.Map;
import java.util.function.Consumer;
import java.util.function.Supplier;

@RequiredArgsConstructor
public class Profile implements IProfile, IProfileInfo, ProfileMBean {
    final String jmxName;
    final INanoTime nanoTime;
    final IPathMap> map;
    final Consumer errorHandler;
    public INanoTime timer() {return nanoTime;}
    @Override
    public  T profile(SupplierWithExceptionE fn) throws E {
        long start = nanoTime.nanoTime();
        try {
            return fn.get();
        } catch (RuntimeException e) {
            errorHandler.accept(e);
            throw e;
        } catch (Exception e) {
            errorHandler.accept(e);
            throw (E) e;
        } finally {
            long duration = nanoTime.nanoTime() - start;
            ProfileBuckets.add(map.get(), duration);
        }
    }
    @Override
    public  void run(RunnableWithExceptionE fn) throws E {
        long start = nanoTime.nanoTime();
        try {
            fn.run();
        } catch (RuntimeException e) {
            errorHandler.accept(e);
            throw e;
        } catch (Exception e) {
            errorHandler.accept(e);
            throw (E) e;
        } finally {
            long duration = nanoTime.nanoTime() - start;
            ProfileBuckets.add(map.get(), duration);
        }
    }
    public void record( long duration) {
        ProfileBuckets.add(map.get(), duration);
    }

    @Override
    public IProfile child(String name) {return new Profile(jmxName, nanoTime, map.child(name), errorHandler);}
    @Override
    public IProfile registerMBean() {
        try {
            MBeanServer mbs = ManagementFactory.getPlatformMBeanServer();
            ObjectName name = new ObjectName(jmxName + ":type=Profile,name=" + map.path());
            if (mbs.isRegistered(name))
                mbs.unregisterMBean(name);
            mbs.registerMBean(this, name);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }

        return this;
    }
    @Override
    public IProfile withErrorHandler(Consumer errorHandler) {
        return new Profile(jmxName, nanoTime, map, errorHandler);
    }
    final static int nanosToMs = 1000000;

    @Override
    public int getCount() {
        return map.get().mapAndAdd(ProfileBucket::getCount, Integer::sum);
    }
    @Override
    public long getTotalMs() {
        return map.get().mapAndAdd(ProfileBucket::getTotal, Long::sum) / nanosToMs;
    }

    @Override
    public String getPath() {
        return map.path();
    }
    @Override
    public long getSnapshotMs() {
        return map.get().snapshot() / nanosToMs;
    }
    public IProfileInfo info() {
        return new ProfileInfo(map.path(), getCount(), getTotalMs(), map.get().snapshot() / nanosToMs);
    }

    public String json() {
        Map map = new HashMap<>();
        this.map.fold(null, (acc, path, buckets) -> map.put(path, ProfileBuckets.json(buckets)));
        return MapHelpers.jsonPrint("\n", map, (key, value) -> value);
    }

    @Override
    public void clear() {
        map.fold(null, (acc, path, buckets) -> {
            ProfileBuckets.clear(buckets);
            return null;
        });
        ProfileBuckets.clear(map.get());
    }

}