scouter.server.netio.service.handle.AlertScriptingService.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
/*
 *  Copyright 2015 the original author or authors.
 *  @https://github.com/scouter-project/scouter
 *
 *  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 scouter.server.netio.service.handle

import scouter.io.{DataInputX, DataOutputX}
import scouter.lang.pack.MapPack
import scouter.lang.value.{DecimalValue, ListValue, MapValue}
import scouter.net.{RequestCmd, TcpFlag}
import scouter.server.core.cache.{AlertScriptLoadMessageCache, XLogCache}
import scouter.server.netio.service.anotation.ServiceHandler
import scouter.server.plugin.PluginHelper
import scouter.server.plugin.alert.RealCounter.Desc
import scouter.server.plugin.alert.{AlertRuleLoader, RealCounter}
import scouter.server.util.EnumerScala

/**
  * @author Gun Lee ([email protected]) on 2017. 9. 24.
  *
  */
class AlertScriptingService {

    @ServiceHandler(RequestCmd.GET_ALERT_SCRIPTING_CONTETNS)
    def getAlertScriptingContents(din: DataInputX, dout: DataOutputX, login: Boolean) {
        val param = din.readPack().asInstanceOf[MapPack]
        val contents = AlertRuleLoader.getInstance().getRuleContents(param.getText("counterName"))

        val result = new MapPack()
        result.put("contents", contents)
        dout.writeByte(TcpFlag.HasNEXT)
        dout.writePack(result)
    }

    @ServiceHandler(RequestCmd.GET_ALERT_SCRIPTING_CONFIG_CONTETNS)
    def getAlertScriptingConfigContents(din: DataInputX, dout: DataOutputX, login: Boolean) {
        val param = din.readPack().asInstanceOf[MapPack]
        val contents = AlertRuleLoader.getInstance().getRuleConfigContents(param.getText("counterName"))

        val result = new MapPack()
        result.put("contents", contents)
        dout.writeByte(TcpFlag.HasNEXT)
        dout.writePack(result)
    }

    @ServiceHandler(RequestCmd.SAVE_ALERT_SCRIPTING_CONTETNS)
    def saveAlertScriptingContents(din: DataInputX, dout: DataOutputX, login: Boolean) {
        val param = din.readPack().asInstanceOf[MapPack]
        val success = AlertRuleLoader.getInstance().saveRuleContents(param.getText("counterName"), param.getText("contents"))

        val result = new MapPack()
        result.put("success", success)
        dout.writeByte(TcpFlag.HasNEXT)
        dout.writePack(result)
    }

    @ServiceHandler(RequestCmd.SAVE_ALERT_SCRIPTING_CONFIG_CONTETNS)
    def saveAlertScriptingConfigContents(din: DataInputX, dout: DataOutputX, login: Boolean) {
        val param = din.readPack().asInstanceOf[MapPack]
        val success = AlertRuleLoader.getInstance().saveConfigContents(param.getText("counterName"), param.getText("contents"))

        val result = new MapPack()
        result.put("success", success)
        dout.writeByte(TcpFlag.HasNEXT)
        dout.writePack(result)
    }

    @ServiceHandler(RequestCmd.GET_ALERT_SCRIPT_LOAD_MESSAGE)
    def getAlertScriptLoadMessage(din: DataInputX, dout: DataOutputX, login: Boolean) {
        val param = din.readMapPack()
        val index = param.getInt("index")
        val loop = param.getLong("loop")

        val cacheOut = AlertScriptLoadMessageCache.get(loop, index)
        if (cacheOut != null) {
            val metaPack = new MapPack()
            metaPack.put("loop", new DecimalValue(cacheOut.loop))
            metaPack.put("index", new DecimalValue(cacheOut.index))
            val messageLv = new ListValue()
            metaPack.put("messages", messageLv)

            EnumerScala.forward(cacheOut.data, (message: String) => {
                messageLv.add(message)
            })

            dout.writeByte(TcpFlag.HasNEXT)
            dout.writePack(metaPack)
        }
    }

    @ServiceHandler(RequestCmd.GET_ALERT_REAL_COUNTER_DESC)
    def getRealCounterDesc(din: DataInputX, dout: DataOutputX, login: Boolean) {
        val descList = RealCounter.getRealCounterDescription

        EnumerScala.forward(descList, (desc: RealCounter.Desc) => {
            val mapPack = new MapPack()
            mapPack.put("desc", desc.desc)
            mapPack.put("methodName", desc.methodName)
            mapPack.put("returnTypeName", desc.returnTypeName)
            val listValue = new ListValue()
            EnumerScala.forward(desc.parameterTypeNames, (name: String) => {
                listValue.add(name);
            })
            mapPack.put("parameterTypeNames", listValue)
            dout.writeByte(TcpFlag.HasNEXT)
            dout.writePack(mapPack)
        })
    }

    @ServiceHandler(RequestCmd.GET_PLUGIN_HELPER_DESC)
    def getPluginHelperDesc(din: DataInputX, dout: DataOutputX, login: Boolean) {
        val descList = PluginHelper.getPluginHelperDesc

        EnumerScala.forward(descList, (desc: PluginHelper.Desc) => {
            val mapPack = new MapPack()
            mapPack.put("desc", desc.desc)
            mapPack.put("methodName", desc.methodName)
            mapPack.put("returnTypeName", desc.returnTypeName)
            val listValue = new ListValue()
            EnumerScala.forward(desc.parameterTypeNames, (name: String) => {
                listValue.add(name);
            })
            mapPack.put("parameterTypeNames", listValue)
            dout.writeByte(TcpFlag.HasNEXT)
            dout.writePack(mapPack)
        })
    }

}