#!/usr/bin/env python

 # -*- coding:utf-8 -*-

 # *************************************

 # @Time    : 2019/7/1

 # @Author  : Zhang Fan

 # @Desc    : Library

 # @File    : MyTools.py

 # @Update  : 2019/8/23

 # *************************************

 import telnetlib

 import hprose

 import jpype

 import json

 import os

 class MyHprose(object):

     """

     ===================================================================

     =====================       MyHprose        =======================

     ===================================================================

     """

     def __init__(self):

         self.client = None

         self.base_url = None

     def create_http_client(self, address, url):

         """

         创建HTTP客户端连接

         """

         self.base_url = 'http://' + address + url

         print('Creating Session : %s' % self.base_url)

         self.client = hprose.HproseHttpClient(self.base_url)

     def set_header(self, key, value):

         """

         设置连接头

         """

         print('Set Header : { "%s": "%s" }' % (key, value))

         self.client.setHeader(key, value)

     def set_parameter(self, *args):

         """

         设置参数值

         """

         arg_list = list()

         for arg in args:

             if isinstance(arg, str):

                 try:

                     arg = eval(arg)

                 except Exception as e:

                     logger.error(e)

             arg_list.append(arg)

         print('Set Parameter : %s' % str(arg_list))

         return arg_list

     def invoke_method(self, method, args):

         """

         调用方法

         """

         result = {

             'type': None,

             'msg': None,

             'value': None

         }

         print('Invoke Method : method={0}, args={1}'.format(method, args))

         ret = self.client.invoke(method, args)

         result['type'] = ret.types

         result['msg'] = ret.msg

         result['value'] = ret.value

         return json.dumps(result, ensure_ascii=False)

 class MyJpype(object):

     """

     ===================================================================

     =====================       MyJpype        ========================

     ===================================================================

     """

     def __init__(self):

         pass

     def start_jvm(self, jvmpath, jarpath):

         """

         开启java虚拟机.

         """

         print('Executing : Start JVM | java -ea -Djava.class.path=JedisSerialize.jar')

         jarpath = os.path.join(os.path.abspath(jarpath), "JedisSerialize.jar")

         jpype.startJVM(jvmpath, "-ea", "-Djava.class.path=%s" % jarpath, convertStrings=False)

     def save_token_jvm(self, db, host, port, auth):

         """

         调用java保存token到内存库.

         """

         JDClass = jpype.JClass("com.util.SearchRedis")

         jd = JDClass()

         jd.saveToken(int(db), str(host), int(port), str(auth))

     def get_redis_jvm(self, key, db, host, port, auth):

         """

         调用java查询String类型内存库.

         """

         print('Executing : Query Key[%s] From Select[%s]' % (key, db))

         JDClass = jpype.JClass("com.util.SearchRedis")

         jd = JDClass()

         result = jd.getbit(str(key), int(db), str(host), int(port), str(auth))

         try:

             return eval(str(result))

         except:

             return json.loads(str(result))

     def get_hash_redis_jvm(self, key, db, host, port, auth):

         """

         调用java查询Hash类型内存库.

         """

         print('Executing : Query Key[%s] From Select[%s]' % (key, db))

         JDClass = jpype.JClass("com.util.SearchRedis")

         jd = JDClass()

         result = jd.gethashbit(str(key), int(db), str(host), int(port), str(auth))

         try:

             return eval(str(result))

         except:

             return json.loads(str(result))

     def shutdown_jvm(self):

         """

         关闭java虚拟机.

         """

         print('Executing : Shutdown JVM')

         jpype.shutdownJVM()

 class MyDubbo(telnetlib.Telnet):

     """

     ===================================================================

     =====================       MyDubbo        =========================

     ===================================================================

     """

     def __init__(self, host=None, port=0):

         # super().__init__(host, port)

         super(MyDubbo, self).__init__(host, port)

         self.write(b"\n")

     def command(self, flag, str_=""):

         data = self.read_until(flag.encode())

         self.write(str_.encode() + b"\n")

         return data

     def invoke(self, service_name, method_name, arg):

         command_str = "invoke {0}.{1}({2})".format(service_name, method_name, arg)

         print('dubbo>:%s' % command_str)

         self.command('dubbo>', command_str)

         data = self.command('dubbo>', "\n")

         new_data = data.decode('utf-8', errors='ignore').split('\n')[0].strip()

         return json.loads(new_data)

 if __name__ == '__main__':

     print('This is test.')

     mh = MyHprose()

     mj = MyJpype()

     md = MyDubbo()