Python register_flavor Examples

Example #1

File: gorpc_protocols_flavor.py Project: tjyang/vitess

  def tabletconn_protocol(self):
    # GoRPC tabletconn doesn't work for the vtgate->vttablet interface,
    # since the go/bson package no longer encodes the non-standard
    # uint64 type.
    return 'grpc'

  def vtgate_protocol(self):
    return 'gorpc'

  def vtgate_python_protocol(self):
    return 'gorpc'

  def rpc_timeout_message(self):
    return 'context deadline exceeded'

  def service_map(self):
    return [
        'bsonrpc-vt-updatestream',
        'bsonrpc-vt-vtctl',
        'bsonrpc-vt-vtgateservice',
        'grpc-vtworker',
        'grpc-queryservice',
        'grpc-tabletmanager',
        ]

  def vttest_protocol(self):
    return 'gorpc'


protocols_flavor.register_flavor('gorpc', GoRpcProtocolsFlavor)

Example #2

    def tabletconn_protocol(self):
        # GoRPC tabletconn doesn't work for the vtgate->vttablet interface,
        # since the go/bson package no longer encodes the non-standard
        # uint64 type.
        return 'grpc'

    def vtgate_protocol(self):
        return 'gorpc'

    def vtgate_python_protocol(self):
        return 'gorpc'

    def rpc_timeout_message(self):
        return 'timeout waiting for'

    def service_map(self):
        return [
            'bsonrpc-vt-queryservice',
            'bsonrpc-vt-tabletmanager',
            'bsonrpc-vt-toporeader',
            'bsonrpc-vt-updatestream',
            'bsonrpc-vt-vtctl',
            'bsonrpc-vt-vtgateservice',
            'grpc-vtworker',
            'grpc-queryservice',
        ]


protocols_flavor.register_flavor('gorpc', GoRpcProtocolsFlavor)