Python 调用 Hprose接口、Dubbo接口、Java方法

#!/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()