scouter.server.db.RealtimeCounterRD.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scouter-server Show documentation
There is a newer version: 2.20.0
package scouter.server.db;

import scouter.io.DataInputX
import scouter.lang.value.MapValue
import scouter.server.db.counter.RealtimeCounterDBHelper
import scouter.util.DateUtil
import scouter.util.FileUtil
import scouter.util.HashUtil
import scouter.util.StringEnumer

import scala.collection.mutable.ArrayBuffer

object RealtimeCounterRD {
    def read(objName: String, date: String, handler: (Long, MapValue) => Any) {
        val stime = DateUtil.yyyymmdd(date);
        val etime = stime + DateUtil.MILLIS_PER_DAY;
        read(objName, date, stime, etime, handler);
    }

    def read(objName: String, date: String, stime: Long, etime: Long, handler: (Long, MapValue) => Any) {
        if (objName == null)
            return ;
        var perfdb: RealtimeCounterDBHelper = null;
        try {
            perfdb = new RealtimeCounterDBHelper().open(date, true);
            if (perfdb == null)
                return ;
            perfdb.counterIndex.read(HashUtil.hash(objName), stime, etime, handler, perfdb.counterDbHeader.getTagIntStr(), perfdb.counterData.read);
        } catch {
            case e: Exception => e.printStackTrace();
        } finally {
            FileUtil.close(perfdb);
        }
    }

    def readBulk(date: String, _stime: Long, etime: Long, handler: (MapValue) => Any) {
        var perfdb: RealtimeCounterDBHelper = null;
        try {
            perfdb = new RealtimeCounterDBHelper().open(date, true);
            if (perfdb == null) {
                return
            }
            val unit = 1 * 60 * 1000
            var stime = _stime
            while(stime <= etime) {
                val pos = perfdb.counterIndex.getStartEndDataPos(stime, math.min(stime+unit-1, etime))
                stime += unit
                if(pos._1 != null && pos._2 != null) {
                    val posStart = DataInputX.toLong5(pos._1, 0)
                    val posEnd = DataInputX.toLong5(pos._2, 0)
                    if(posStart > 0 && posEnd > 0) {
                        val buffer = perfdb.counterData.readBulk(posStart, posEnd)
                        val keyMap = perfdb.counterDbHeader.getTagIntStr()
                        val items = RealtimeCounterDBHelper.setTagBytesMulti(keyMap, buffer);
                        items.foreach(mapValue => handler(mapValue))
                    }
                }
            }
        } catch {
            case e: Exception => e.printStackTrace();
        } finally {
            FileUtil.close(perfdb);
        }
    }

    def getCounterSet(date: String): StringEnumer = {
        var logdb: RealtimeCounterDBHelper = null;
        try {
            logdb = new RealtimeCounterDBHelper().open(date, true);
            if (logdb == null)
                return null;
            return logdb.counterDbHeader.getTagStrInt().keys();
        } catch {
            case e: Exception => e.printStackTrace();
        } finally {
            FileUtil.close(logdb);
        }
        return null;
    }

    def readFromEnd(objName: String, date: String, handler: (Long, MapValue) => Boolean) {
        val stime = DateUtil.yyyymmdd(date);
        val etime = stime + DateUtil.MILLIS_PER_DAY;
        readFromEnd(objName, date, stime, etime, handler);
    }

    def readFromEnd(objName: String, date: String, stime: Long, etime: Long, handler: (Long, MapValue) => Any) {
        var logdb: RealtimeCounterDBHelper = null;
        try {
            logdb = new RealtimeCounterDBHelper().open(date, true);
            if (logdb == null)
                return ;

            logdb.counterIndex.readFromEnd(HashUtil.hash(objName), stime, etime, handler, logdb.counterDbHeader.getTagIntStr(),
                logdb.counterData.read);
        } catch {
            case e: Exception => e.printStackTrace();
        } finally {
            FileUtil.close(logdb);
        }
    }

    def getDataNames(objName: String, date: String): StringEnumer = {
        var logdb: RealtimeCounterDBHelper = null;
        try {
            logdb = new RealtimeCounterDBHelper().open(date, true);
            if (logdb == null)
                return null;
            return logdb.counterDbHeader.getTagStrInt().keys();
        } catch {
            case e: Exception => e.printStackTrace();
        } finally {
            FileUtil.close(logdb);
        }
        return null;
    }
}