一、Thrift 基操实现
1.1 thrift文件
namespace java com.zn.opit.thrift.helloworld service HelloWorldService { string sayHello(1:string username) }
1.2 执行命令生成Java文件
thrift -r --gen java helloworld.thrift
生成代码HelloWorldService接口如下
/** * Autogenerated by Thrift Compiler (0.11.0) * * DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING * @generated */ package com.zn.opit.thrift.helloworld; @SuppressWarnings({"cast", "rawtypes", "serial", "unchecked", "unused"}) @javax.annotation.Generated(value = "Autogenerated by Thrift Compiler (0.11.0)", date = "2023-10-29") public class HelloWorldService { public interface Iface { public java.lang.String sayHello(java.lang.String username) throws org.apache.thrift.TException; } public interface AsyncIface { public void sayHello(java.lang.String username, org.apache.thrift.async.AsyncMethodCallbackresultHandler) throws org.apache.thrift.TException; } public static class Client extends org.apache.thrift.TServiceClient implements Iface { public static class Factory implements org.apache.thrift.TServiceClientFactory { public Factory() {} public Client getClient(org.apache.thrift.protocol.TProtocol prot) { return new Client(prot); } public Client getClient(org.apache.thrift.protocol.TProtocol iprot, org.apache.thrift.protocol.TProtocol oprot) { return new Client(iprot, oprot); } } public Client(org.apache.thrift.protocol.TProtocol prot) { super(prot, prot); } public Client(org.apache.thrift.protocol.TProtocol iprot, org.apache.thrift.protocol.TProtocol oprot) { super(iprot, oprot); } public java.lang.String sayHello(java.lang.String username) throws org.apache.thrift.TException { send_sayHello(username); return recv_sayHello(); } public void send_sayHello(java.lang.String username) throws org.apache.thrift.TException { sayHello_args args = new sayHello_args(); args.setUsername(username); sendBase("sayHello", args); } public java.lang.String recv_sayHello() throws org.apache.thrift.TException { sayHello_result result = new sayHello_result(); receiveBase(result, "sayHello"); if (result.isSetSuccess()) { return result.success; } throw new org.apache.thrift.TApplicationException(org.apache.thrift.TApplicationException.MISSING_RESULT, "sayHello failed: unknown result"); } } public static class AsyncClient extends org.apache.thrift.async.TAsyncClient implements AsyncIface { public static class Factory implements org.apache.thrift.async.TAsyncClientFactory { private org.apache.thrift.async.TAsyncClientManager clientManager; private org.apache.thrift.protocol.TProtocolFactory protocolFactory; public Factory(org.apache.thrift.async.TAsyncClientManager clientManager, org.apache.thrift.protocol.TProtocolFactory protocolFactory) { this.clientManager = clientManager; this.protocolFactory = protocolFactory; } public AsyncClient getAsyncClient(org.apache.thrift.transport.TNonblockingTransport transport) { return new AsyncClient(protocolFactory, clientManager, transport); } } public AsyncClient(org.apache.thrift.protocol.TProtocolFactory protocolFactory, org.apache.thrift.async.TAsyncClientManager clientManager, org.apache.thrift.transport.TNonblockingTransport transport) { super(protocolFactory, clientManager, transport); } public void sayHello(java.lang.String username, org.apache.thrift.async.AsyncMethodCallback resultHandler) throws org.apache.thrift.TException { checkReady(); sayHello_call method_call = new sayHello_call(username, resultHandler, this, ___protocolFactory, ___transport); this.___currentMethod = method_call; ___manager.call(method_call); } public static class sayHello_call extends org.apache.thrift.async.TAsyncMethodCall { private java.lang.String username; public sayHello_call(java.lang.String username, org.apache.thrift.async.AsyncMethodCallback resultHandler, org.apache.thrift.async.TAsyncClient client, org.apache.thrift.protocol.TProtocolFactory protocolFactory, org.apache.thrift.transport.TNonblockingTransport transport) throws org.apache.thrift.TException { super(client, protocolFactory, transport, resultHandler, false); this.username = username; } public void write_args(org.apache.thrift.protocol.TProtocol prot) throws org.apache.thrift.TException { prot.writeMessageBegin(new org.apache.thrift.protocol.TMessage("sayHello", org.apache.thrift.protocol.TMessageType.CALL, 0)); sayHello_args args = new sayHello_args(); args.setUsername(username); args.write(prot); prot.writeMessageEnd(); } public java.lang.String getResult() throws org.apache.thrift.TException { if (getState() != org.apache.thrift.async.TAsyncMethodCall.State.RESPONSE_READ) { throw new java.lang.IllegalStateException("Method call not finished!"); } org.apache.thrift.transport.TMemoryInputTransport memoryTransport = new org.apache.thrift.transport.TMemoryInputTransport(getFrameBuffer().array()); org.apache.thrift.protocol.TProtocol prot = client.getProtocolFactory().getProtocol(memoryTransport); return (new Client(prot)).recv_sayHello(); } } } public static class Processor extends org.apache.thrift.TBaseProcessor implements org.apache.thrift.TProcessor { private static final org.slf4j.Logger _LOGGER = org.slf4j.LoggerFactory.getLogger(Processor.class.getName()); public Processor(I iface) { super(iface, getProcessMap(new java.util.HashMap >())); } protected Processor(I iface, java.util.Map > processMap) { super(iface, getProcessMap(processMap)); } private static java.util.Map > getProcessMap(java.util.Map > processMap) { processMap.put("sayHello", new sayHello()); return processMap; } public static class sayHello extends org.apache.thrift.ProcessFunction { public sayHello() { super("sayHello"); } public sayHello_args getEmptyArgsInstance() { return new sayHello_args(); } protected boolean isOneway() { return false; } @Override protected boolean handleRuntimeExceptions() { return false; } public sayHello_result getResult(I iface, sayHello_args args) throws org.apache.thrift.TException { sayHello_result result = new sayHello_result(); result.success = iface.sayHello(args.username); return result; } } } public static class AsyncProcessor extends org.apache.thrift.TBaseAsyncProcessor { private static final org.slf4j.Logger _LOGGER = org.slf4j.LoggerFactory.getLogger(AsyncProcessor.class.getName()); public AsyncProcessor(I iface) { super(iface, getProcessMap(new java.util.HashMap >())); } protected AsyncProcessor(I iface, java.util.Map > processMap) { super(iface, getProcessMap(processMap)); } private static java.util.Map > getProcessMap(java.util.Map > processMap) { processMap.put("sayHello", new sayHello()); return processMap; } public static class sayHello extends org.apache.thrift.AsyncProcessFunction { public sayHello() { super("sayHello"); } public sayHello_args getEmptyArgsInstance() { return new sayHello_args(); } public org.apache.thrift.async.AsyncMethodCallback getResultHandler(final org.apache.thrift.server.AbstractNonblockingServer.AsyncFrameBuffer fb, final int seqid) { final org.apache.thrift.AsyncProcessFunction fcall = this; return new org.apache.thrift.async.AsyncMethodCallback () { public void onComplete(java.lang.String o) { sayHello_result result = new sayHello_result(); result.success = o; try { fcall.sendResponse(fb, result, org.apache.thrift.protocol.TMessageType.REPLY,seqid); } catch (org.apache.thrift.transport.TTransportException e) { _LOGGER.error("TTransportException writing to internal frame buffer", e); fb.close(); } catch (java.lang.Exception e) { _LOGGER.error("Exception writing to internal frame buffer", e); onError(e); } } public void onError(java.lang.Exception e) { byte msgType = org.apache.thrift.protocol.TMessageType.REPLY; org.apache.thrift.TSerializable msg; sayHello_result result = new sayHello_result(); if (e instanceof org.apache.thrift.transport.TTransportException) { _LOGGER.error("TTransportException inside handler", e); fb.close(); return; } else if (e instanceof org.apache.thrift.TApplicationException) { _LOGGER.error("TApplicationException inside handler", e); msgType = org.apache.thrift.protocol.TMessageType.EXCEPTION; msg = (org.apache.thrift.TApplicationException)e; } else { _LOGGER.error("Exception inside handler", e); msgType = org.apache.thrift.protocol.TMessageType.EXCEPTION; msg = new org.apache.thrift.TApplicationException(org.apache.thrift.TApplicationException.INTERNAL_ERROR, e.getMessage()); } try { fcall.sendResponse(fb,msg,msgType,seqid); } catch (java.lang.Exception ex) { _LOGGER.error("Exception writing to internal frame buffer", ex); fb.close(); } } }; } protected boolean isOneway() { return false; } public void start(I iface, sayHello_args args, org.apache.thrift.async.AsyncMethodCallback resultHandler) throws org.apache.thrift.TException { iface.sayHello(args.username,resultHandler); } } } public static class sayHello_args implements org.apache.thrift.TBase , java.io.Serializable, Cloneable, Comparable { private static final org.apache.thrift.protocol.TStruct STRUCT_DESC = new org.apache.thrift.protocol.TStruct("sayHello_args"); private static final org.apache.thrift.protocol.TField USERNAME_FIELD_DESC = new org.apache.thrift.protocol.TField("username", org.apache.thrift.protocol.TType.STRING, (short)1); private static final org.apache.thrift.scheme.SchemeFactory STANDARD_SCHEME_FACTORY = new sayHello_argsStandardSchemeFactory(); private static final org.apache.thrift.scheme.SchemeFactory TUPLE_SCHEME_FACTORY = new sayHello_argsTupleSchemeFactory(); public java.lang.String username; // required /** The set of fields this struct contains, along with convenience methods for finding and manipulating them. */ public enum _Fields implements org.apache.thrift.TFieldIdEnum { USERNAME((short)1, "username"); private static final java.util.Map byName = new java.util.HashMap (); static { for (_Fields field : java.util.EnumSet.allOf(_Fields.class)) { byName.put(field.getFieldName(), field); } } /** * Find the _Fields constant that matches fieldId, or null if its not found. */ public static _Fields findByThriftId(int fieldId) { switch(fieldId) { case 1: // USERNAME return USERNAME; default: return null; } } /** * Find the _Fields constant that matches fieldId, throwing an exception * if it is not found. */ public static _Fields findByThriftIdOrThrow(int fieldId) { _Fields fields = findByThriftId(fieldId); if (fields == null) throw new java.lang.IllegalArgumentException("Field " + fieldId + " doesn't exist!"); return fields; } /** * Find the _Fields constant that matches name, or null if its not found. */ public static _Fields findByName(java.lang.String name) { return byName.get(name); } private final short _thriftId; private final java.lang.String _fieldName; _Fields(short thriftId, java.lang.String fieldName) { _thriftId = thriftId; _fieldName = fieldName; } public short getThriftFieldId() { return _thriftId; } public java.lang.String getFieldName() { return _fieldName; } } // isset id assignments public static final java.util.Map<_Fields, org.apache.thrift.meta_data.FieldMetaData> metaDataMap; static { java.util.Map<_Fields, org.apache.thrift.meta_data.FieldMetaData> tmpMap = new java.util.EnumMap<_Fields, org.apache.thrift.meta_data.FieldMetaData>(_Fields.class); tmpMap.put(_Fields.USERNAME, new org.apache.thrift.meta_data.FieldMetaData("username", org.apache.thrift.TFieldRequirementType.DEFAULT, new org.apache.thrift.meta_data.FieldValueMetaData(org.apache.thrift.protocol.TType.STRING))); metaDataMap = java.util.Collections.unmodifiableMap(tmpMap); org.apache.thrift.meta_data.FieldMetaData.addStructMetaDataMap(sayHello_args.class, metaDataMap); } public sayHello_args() { } public sayHello_args( java.lang.String username) { this(); this.username = username; } /** * Performs a deep copy on other. */ public sayHello_args(sayHello_args other) { if (other.isSetUsername()) { this.username = other.username; } } public sayHello_args deepCopy() { return new sayHello_args(this); } @Override public void clear() { this.username = null; } public java.lang.String getUsername() { return this.username; } public sayHello_args setUsername(java.lang.String username) { this.username = username; return this; } public void unsetUsername() { this.username = null; } /** Returns true if field username is set (has been assigned a value) and false otherwise */ public boolean isSetUsername() { return this.username != null; } public void setUsernameIsSet(boolean value) { if (!value) { this.username = null; } } public void setFieldValue(_Fields field, java.lang.Object value) { switch (field) { case USERNAME: if (value == null) { unsetUsername(); } else { setUsername((java.lang.String)value); } break; } } public java.lang.Object getFieldValue(_Fields field) { switch (field) { case USERNAME: return getUsername(); } throw new java.lang.IllegalStateException(); } /** Returns true if field corresponding to fieldID is set (has been assigned a value) and false otherwise */ public boolean isSet(_Fields field) { if (field == null) { throw new java.lang.IllegalArgumentException(); } switch (field) { case USERNAME: return isSetUsername(); } throw new java.lang.IllegalStateException(); } @Override public boolean equals(java.lang.Object that) { if (that == null) return false; if (that instanceof sayHello_args) return this.equals((sayHello_args)that); return false; } public boolean equals(sayHello_args that) { if (that == null) return false; if (this == that) return true; boolean this_present_username = true && this.isSetUsername(); boolean that_present_username = true && that.isSetUsername(); if (this_present_username || that_present_username) { if (!(this_present_username && that_present_username)) return false; if (!this.username.equals(that.username)) return false; } return true; } @Override public int hashCode() { int hashCode = 1; hashCode = hashCode * 8191 + ((isSetUsername()) ? 131071 : 524287); if (isSetUsername()) hashCode = hashCode * 8191 + username.hashCode(); return hashCode; } @Override public int compareTo(sayHello_args other) { if (!getClass().equals(other.getClass())) { return getClass().getName().compareTo(other.getClass().getName()); } int lastComparison = 0; lastComparison = java.lang.Boolean.valueOf(isSetUsername()).compareTo(other.isSetUsername()); if (lastComparison != 0) { return lastComparison; } if (isSetUsername()) { lastComparison = org.apache.thrift.TBaseHelper.compareTo(this.username, other.username); if (lastComparison != 0) { return lastComparison; } } return 0; } public _Fields fieldForId(int fieldId) { return _Fields.findByThriftId(fieldId); } public void read(org.apache.thrift.protocol.TProtocol iprot) throws org.apache.thrift.TException { scheme(iprot).read(iprot, this); } public void write(org.apache.thrift.protocol.TProtocol oprot) throws org.apache.thrift.TException { scheme(oprot).write(oprot, this); } @Override public java.lang.String toString() { java.lang.StringBuilder sb = new java.lang.StringBuilder("sayHello_args("); boolean first = true; sb.append("username:"); if (this.username == null) { sb.append("null"); } else { sb.append(this.username); } first = false; sb.append(")"); return sb.toString(); } public void validate() throws org.apache.thrift.TException { // check for required fields // check for sub-struct validity } private void writeObject(java.io.ObjectOutputStream out) throws java.io.IOException { try { write(new org.apache.thrift.protocol.TCompactProtocol(new org.apache.thrift.transport.TIOStreamTransport(out))); } catch (org.apache.thrift.TException te) { throw new java.io.IOException(te); } } private void readObject(java.io.ObjectInputStream in) throws java.io.IOException, java.lang.ClassNotFoundException { try { read(new org.apache.thrift.protocol.TCompactProtocol(new org.apache.thrift.transport.TIOStreamTransport(in))); } catch (org.apache.thrift.TException te) { throw new java.io.IOException(te); } } private static class sayHello_argsStandardSchemeFactory implements org.apache.thrift.scheme.SchemeFactory { public sayHello_argsStandardScheme getScheme() { return new sayHello_argsStandardScheme(); } } private static class sayHello_argsStandardScheme extends org.apache.thrift.scheme.StandardScheme { public void read(org.apache.thrift.protocol.TProtocol iprot, sayHello_args struct) throws org.apache.thrift.TException { org.apache.thrift.protocol.TField schemeField; iprot.readStructBegin(); while (true) { schemeField = iprot.readFieldBegin(); if (schemeField.type == org.apache.thrift.protocol.TType.STOP) { break; } switch (schemeField.id) { case 1: // USERNAME if (schemeField.type == org.apache.thrift.protocol.TType.STRING) { struct.username = iprot.readString(); struct.setUsernameIsSet(true); } else { org.apache.thrift.protocol.TProtocolUtil.skip(iprot, schemeField.type); } break; default: org.apache.thrift.protocol.TProtocolUtil.skip(iprot, schemeField.type); } iprot.readFieldEnd(); } iprot.readStructEnd(); // check for required fields of primitive type, which can't be checked in the validate method struct.validate(); } public void write(org.apache.thrift.protocol.TProtocol oprot, sayHello_args struct) throws org.apache.thrift.TException { struct.validate(); oprot.writeStructBegin(STRUCT_DESC); if (struct.username != null) { oprot.writeFieldBegin(USERNAME_FIELD_DESC); oprot.writeString(struct.username); oprot.writeFieldEnd(); } oprot.writeFieldStop(); oprot.writeStructEnd(); } } private static class sayHello_argsTupleSchemeFactory implements org.apache.thrift.scheme.SchemeFactory { public sayHello_argsTupleScheme getScheme() { return new sayHello_argsTupleScheme(); } } private static class sayHello_argsTupleScheme extends org.apache.thrift.scheme.TupleScheme { @Override public void write(org.apache.thrift.protocol.TProtocol prot, sayHello_args struct) throws org.apache.thrift.TException { org.apache.thrift.protocol.TTupleProtocol oprot = (org.apache.thrift.protocol.TTupleProtocol) prot; java.util.BitSet optionals = new java.util.BitSet(); if (struct.isSetUsername()) { optionals.set(0); } oprot.writeBitSet(optionals, 1); if (struct.isSetUsername()) { oprot.writeString(struct.username); } } @Override public void read(org.apache.thrift.protocol.TProtocol prot, sayHello_args struct) throws org.apache.thrift.TException { org.apache.thrift.protocol.TTupleProtocol iprot = (org.apache.thrift.protocol.TTupleProtocol) prot; java.util.BitSet incoming = iprot.readBitSet(1); if (incoming.get(0)) { struct.username = iprot.readString(); struct.setUsernameIsSet(true); } } } private static S scheme(org.apache.thrift.protocol.TProtocol proto) { return (org.apache.thrift.scheme.StandardScheme.class.equals(proto.getScheme()) ? STANDARD_SCHEME_FACTORY : TUPLE_SCHEME_FACTORY).getScheme(); } } public static class sayHello_result implements org.apache.thrift.TBase, java.io.Serializable, Cloneable, Comparable { private static final org.apache.thrift.protocol.TStruct STRUCT_DESC = new org.apache.thrift.protocol.TStruct("sayHello_result"); private static final org.apache.thrift.protocol.TField SUCCESS_FIELD_DESC = new org.apache.thrift.protocol.TField("success", org.apache.thrift.protocol.TType.STRING, (short)0); private static final org.apache.thrift.scheme.SchemeFactory STANDARD_SCHEME_FACTORY = new sayHello_resultStandardSchemeFactory(); private static final org.apache.thrift.scheme.SchemeFactory TUPLE_SCHEME_FACTORY = new sayHello_resultTupleSchemeFactory(); public java.lang.String success; // required /** The set of fields this struct contains, along with convenience methods for finding and manipulating them. */ public enum _Fields implements org.apache.thrift.TFieldIdEnum { SUCCESS((short)0, "success"); private static final java.util.Map byName = new java.util.HashMap (); static { for (_Fields field : java.util.EnumSet.allOf(_Fields.class)) { byName.put(field.getFieldName(), field); } } /** * Find the _Fields constant that matches fieldId, or null if its not found. */ public static _Fields findByThriftId(int fieldId) { switch(fieldId) { case 0: // SUCCESS return SUCCESS; default: return null; } } /** * Find the _Fields constant that matches fieldId, throwing an exception * if it is not found. */ public static _Fields findByThriftIdOrThrow(int fieldId) { _Fields fields = findByThriftId(fieldId); if (fields == null) throw new java.lang.IllegalArgumentException("Field " + fieldId + " doesn't exist!"); return fields; } /** * Find the _Fields constant that matches name, or null if its not found. */ public static _Fields findByName(java.lang.String name) { return byName.get(name); } private final short _thriftId; private final java.lang.String _fieldName; _Fields(short thriftId, java.lang.String fieldName) { _thriftId = thriftId; _fieldName = fieldName; } public short getThriftFieldId() { return _thriftId; } public java.lang.String getFieldName() { return _fieldName; } } // isset id assignments public static final java.util.Map<_Fields, org.apache.thrift.meta_data.FieldMetaData> metaDataMap; static { java.util.Map<_Fields, org.apache.thrift.meta_data.FieldMetaData> tmpMap = new java.util.EnumMap<_Fields, org.apache.thrift.meta_data.FieldMetaData>(_Fields.class); tmpMap.put(_Fields.SUCCESS, new org.apache.thrift.meta_data.FieldMetaData("success", org.apache.thrift.TFieldRequirementType.DEFAULT, new org.apache.thrift.meta_data.FieldValueMetaData(org.apache.thrift.protocol.TType.STRING))); metaDataMap = java.util.Collections.unmodifiableMap(tmpMap); org.apache.thrift.meta_data.FieldMetaData.addStructMetaDataMap(sayHello_result.class, metaDataMap); } public sayHello_result() { } public sayHello_result( java.lang.String success) { this(); this.success = success; } /** * Performs a deep copy on other. */ public sayHello_result(sayHello_result other) { if (other.isSetSuccess()) { this.success = other.success; } } public sayHello_result deepCopy() { return new sayHello_result(this); } @Override public void clear() { this.success = null; } public java.lang.String getSuccess() { return this.success; } public sayHello_result setSuccess(java.lang.String success) { this.success = success; return this; } public void unsetSuccess() { this.success = null; } /** Returns true if field success is set (has been assigned a value) and false otherwise */ public boolean isSetSuccess() { return this.success != null; } public void setSuccessIsSet(boolean value) { if (!value) { this.success = null; } } public void setFieldValue(_Fields field, java.lang.Object value) { switch (field) { case SUCCESS: if (value == null) { unsetSuccess(); } else { setSuccess((java.lang.String)value); } break; } } public java.lang.Object getFieldValue(_Fields field) { switch (field) { case SUCCESS: return getSuccess(); } throw new java.lang.IllegalStateException(); } /** Returns true if field corresponding to fieldID is set (has been assigned a value) and false otherwise */ public boolean isSet(_Fields field) { if (field == null) { throw new java.lang.IllegalArgumentException(); } switch (field) { case SUCCESS: return isSetSuccess(); } throw new java.lang.IllegalStateException(); } @Override public boolean equals(java.lang.Object that) { if (that == null) return false; if (that instanceof sayHello_result) return this.equals((sayHello_result)that); return false; } public boolean equals(sayHello_result that) { if (that == null) return false; if (this == that) return true; boolean this_present_success = true && this.isSetSuccess(); boolean that_present_success = true && that.isSetSuccess(); if (this_present_success || that_present_success) { if (!(this_present_success && that_present_success)) return false; if (!this.success.equals(that.success)) return false; } return true; } @Override public int hashCode() { int hashCode = 1; hashCode = hashCode * 8191 + ((isSetSuccess()) ? 131071 : 524287); if (isSetSuccess()) hashCode = hashCode * 8191 + success.hashCode(); return hashCode; } @Override public int compareTo(sayHello_result other) { if (!getClass().equals(other.getClass())) { return getClass().getName().compareTo(other.getClass().getName()); } int lastComparison = 0; lastComparison = java.lang.Boolean.valueOf(isSetSuccess()).compareTo(other.isSetSuccess()); if (lastComparison != 0) { return lastComparison; } if (isSetSuccess()) { lastComparison = org.apache.thrift.TBaseHelper.compareTo(this.success, other.success); if (lastComparison != 0) { return lastComparison; } } return 0; } public _Fields fieldForId(int fieldId) { return _Fields.findByThriftId(fieldId); } public void read(org.apache.thrift.protocol.TProtocol iprot) throws org.apache.thrift.TException { scheme(iprot).read(iprot, this); } public void write(org.apache.thrift.protocol.TProtocol oprot) throws org.apache.thrift.TException { scheme(oprot).write(oprot, this); } @Override public java.lang.String toString() { java.lang.StringBuilder sb = new java.lang.StringBuilder("sayHello_result("); boolean first = true; sb.append("success:"); if (this.success == null) { sb.append("null"); } else { sb.append(this.success); } first = false; sb.append(")"); return sb.toString(); } public void validate() throws org.apache.thrift.TException { // check for required fields // check for sub-struct validity } private void writeObject(java.io.ObjectOutputStream out) throws java.io.IOException { try { write(new org.apache.thrift.protocol.TCompactProtocol(new org.apache.thrift.transport.TIOStreamTransport(out))); } catch (org.apache.thrift.TException te) { throw new java.io.IOException(te); } } private void readObject(java.io.ObjectInputStream in) throws java.io.IOException, java.lang.ClassNotFoundException { try { read(new org.apache.thrift.protocol.TCompactProtocol(new org.apache.thrift.transport.TIOStreamTransport(in))); } catch (org.apache.thrift.TException te) { throw new java.io.IOException(te); } } private static class sayHello_resultStandardSchemeFactory implements org.apache.thrift.scheme.SchemeFactory { public sayHello_resultStandardScheme getScheme() { return new sayHello_resultStandardScheme(); } } private static class sayHello_resultStandardScheme extends org.apache.thrift.scheme.StandardScheme { public void read(org.apache.thrift.protocol.TProtocol iprot, sayHello_result struct) throws org.apache.thrift.TException { org.apache.thrift.protocol.TField schemeField; iprot.readStructBegin(); while (true) { schemeField = iprot.readFieldBegin(); if (schemeField.type == org.apache.thrift.protocol.TType.STOP) { break; } switch (schemeField.id) { case 0: // SUCCESS if (schemeField.type == org.apache.thrift.protocol.TType.STRING) { struct.success = iprot.readString(); struct.setSuccessIsSet(true); } else { org.apache.thrift.protocol.TProtocolUtil.skip(iprot, schemeField.type); } break; default: org.apache.thrift.protocol.TProtocolUtil.skip(iprot, schemeField.type); } iprot.readFieldEnd(); } iprot.readStructEnd(); // check for required fields of primitive type, which can't be checked in the validate method struct.validate(); } public void write(org.apache.thrift.protocol.TProtocol oprot, sayHello_result struct) throws org.apache.thrift.TException { struct.validate(); oprot.writeStructBegin(STRUCT_DESC); if (struct.success != null) { oprot.writeFieldBegin(SUCCESS_FIELD_DESC); oprot.writeString(struct.success); oprot.writeFieldEnd(); } oprot.writeFieldStop(); oprot.writeStructEnd(); } } private static class sayHello_resultTupleSchemeFactory implements org.apache.thrift.scheme.SchemeFactory { public sayHello_resultTupleScheme getScheme() { return new sayHello_resultTupleScheme(); } } private static class sayHello_resultTupleScheme extends org.apache.thrift.scheme.TupleScheme { @Override public void write(org.apache.thrift.protocol.TProtocol prot, sayHello_result struct) throws org.apache.thrift.TException { org.apache.thrift.protocol.TTupleProtocol oprot = (org.apache.thrift.protocol.TTupleProtocol) prot; java.util.BitSet optionals = new java.util.BitSet(); if (struct.isSetSuccess()) { optionals.set(0); } oprot.writeBitSet(optionals, 1); if (struct.isSetSuccess()) { oprot.writeString(struct.success); } } @Override public void read(org.apache.thrift.protocol.TProtocol prot, sayHello_result struct) throws org.apache.thrift.TException { org.apache.thrift.protocol.TTupleProtocol iprot = (org.apache.thrift.protocol.TTupleProtocol) prot; java.util.BitSet incoming = iprot.readBitSet(1); if (incoming.get(0)) { struct.success = iprot.readString(); struct.setSuccessIsSet(true); } } } private static S scheme(org.apache.thrift.protocol.TProtocol proto) { return (org.apache.thrift.scheme.StandardScheme.class.equals(proto.getScheme()) ? STANDARD_SCHEME_FACTORY : TUPLE_SCHEME_FACTORY).getScheme(); } } }
1.3 实现HelloWorldService
public class HelloWordServiceImpl implements HelloWorldService.Iface { @Override public String sayHello(String username) throws TException { return "hello, " + username; } }
1.4 服务端
@Slf4j public class ThriftServer { public static final int SERVER_PORT = 8010; public static void main(String[] args) throws Exception { log.info("服务端开启...."); // 启动服务 new TSimpleServer(new TServer.Args(new TServerSocket(new ServerSocket(SERVER_PORT))) .processor(new HelloWorldService.Processor(new HelloWordServiceImpl())) .protocolFactory(new TBinaryProtocol.Factory())).serve(); } }
1.5 客户端
@Slf4j public class ThriftClient { // 服务端IP地址 public static final String SERVER_IP = "127.0.0.1"; // 服务端端口号 public static final int SERVER_PORT = 8010; public static void main(String[] args) throws Exception { log.info("客户端启动...."); try (TTransport transport = new TSocket(SERVER_IP, SERVER_PORT)) { TProtocol protocol = new TBinaryProtocol(transport); HelloWorldService.Client client = new HelloWorldService.Client(protocol); transport.open(); System.out.println("result = [remote: " + client.sayHello("wangxiaohu") + "]"); } catch (Exception e) { log.error("客户端启动失败", e); } } }
1.6 启动服务端、客户端进行测试
- 先启动服务端
- 在启动客户端
二、微服务间相互调用
2.1 通过thrift生成HelloWorldService
往前调转 1.2 执行命令生成Java文件
2.2 thrift客户端代理
/** * Thrift客户端代理 */ @SuppressWarnings({"unchecked" }) public class ThriftClientProxy { // 服务端端口号 public static final int SERVER_PORT = 50000; public static
T getRemoteProxyObj(final Class> serviceInterface, String host) { return (T) Proxy.newProxyInstance(serviceInterface.getClassLoader(), new Class>[]{serviceInterface}, (proxy, method, args) -> { TTransport transport = new TSocket(host, SERVER_PORT); TProtocol protocol = new TBinaryProtocol(transport); Class>[] argsClass = new Class[]{TProtocol.class}; Constructor> cons = Class.forName(serviceInterface.getName().replace("Iface", "Client")).getConstructor(argsClass); T client = (T) cons.newInstance(protocol); transport.open(); return method.invoke(client, args); }); } } 2.3 Thrift服务端代理
/** * Thrift服务端代理 */ public class ThriftServerProxy { private final Logger logger = LoggerFactory.getLogger(getClass()); private static final int THRIFT_SERVER_PORT = 50000;// 端口 private String serviceInterface;// 实现类接口 private Object serviceImplObject;// 实现类 @SuppressWarnings({ "unchecked", "rawtypes" }) public void start() { new Thread(() -> { try { TServerSocket serverTransport = new TServerSocket(THRIFT_SERVER_PORT); // 实现类处理类class Class Processor = Class.forName(getServiceInterface() + "$Processor"); // 接口 Class Iface = Class.forName(getServiceInterface() + "$Iface"); // 接口构造方法类 Constructor con = Processor.getConstructor(Iface); // 实现类处理类 TProcessor processor = (TProcessor) con.newInstance(serviceImplObject); TBinaryProtocol.Factory protocolFactory = new TBinaryProtocol.Factory(true, true); TThreadPoolServer.Args args = new TThreadPoolServer.Args(serverTransport); args.protocolFactory(protocolFactory); args.processor(processor); TServer server = new TThreadPoolServer(args); logger.info("Starting thrift server on port " + THRIFT_SERVER_PORT + " ..." + getServiceInterface()); server.serve(); } catch (Exception e) { e.printStackTrace(); } }).start(); } public ThriftServerProxy() {} public ThriftServerProxy(String serviceInterface, Object serviceImplObject) { this.serviceInterface = serviceInterface; this.serviceImplObject = serviceImplObject; } public String getServiceInterface() { return serviceInterface; } public void setServiceInterface(String serviceInterface) { this.serviceInterface = serviceInterface; } public Object getServiceImplObject() { return serviceImplObject; } public void setServiceImplObject(Object serviceImplObject) { this.serviceImplObject = serviceImplObject; } }
2.4 实现注解Rpc Service导入
@Slf4j @Component public class ThriftClientInterceptor implements BeanPostProcessor { @Resource private DiscoveryClient discoveryClient; @Override public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException { Class> beanClass = bean.getClass(); Field[] fields = beanClass.getDeclaredFields(); final Random random = new Random(); for (Field field : fields) { ImpThrift impThrift = field.getAnnotation(ImpThrift.class); if (Objects.isNull(impThrift)) continue; try { String serviceId = impThrift.def(); List
instances = discoveryClient.getInstances(serviceId); if (CollectionUtils.isEmpty(instances)) { // throw new ExtThriftException("instance [" + serviceId + "] not exists"); log.warn("instance [{}] not exists", serviceId); } ServiceInstance serviceInstance = instances.get(random.nextInt(instances.size())); field.setAccessible(true); field.set(bean, ThriftClientProxy.getRemoteProxyObj(field.getType(), serviceInstance.getHost())); } catch (IllegalAccessException e) { e.printStackTrace(); } } return BeanPostProcessor.super.postProcessAfterInitialization(bean, beanName); } } 2.5 实现注解Rpc Service导出
@Component public class ThriftServerInterceptor implements BeanPostProcessor { @Override public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException { ExtThrift extThrift = bean.getClass().getAnnotation(ExtThrift.class); if (!Objects.isNull(extThrift)) { // String serviceName = extThrift.serviceName(); Class>[] interfaces = bean.getClass().getInterfaces(); ThriftServerProxy thriftServerProxy = new ThriftServerProxy(interfaces[0].getName().replace("$Iface", ""), bean); thriftServerProxy.start(); } return BeanPostProcessor.super.postProcessAfterInitialization(bean, beanName); } }
2.6 A服务引入含有thrift生成的service的SDK对Service进行实现
@ExtThrift(serviceName = "HelloWorldService") @Service public class HelloWorldServiceImpl implements HelloWorldService.Iface { @Override public String sayHello(String username) throws TException { return username; } }
@RestController @Slf4j public class PaymentController { @Value("${server.port}") private String serverPort; @Autowired private HelloWorldService.Iface helloWorldService; @RequestMapping(value = "/payment/zk") public String paymentZk(@RequestParam("username") String username) throws TException { return "SpringCloud with zookeeper: " + serverPort + "\t" + UUID.randomUUID() + "-" + helloWorldService.sayHello(username); } }
2.7 B服务引入含有thrift生成的service的SDK和A服务进行调用
@RestController @Slf4j public class OrderController { @Value("${server.port}") private String serverPort; @ImpThrift(def = "CLOUD-PAYMENT-SERVICE", serviceName = "helloWorldService") private HelloWorldService.Iface helloWorldService; @RequestMapping(value = "/payment/zk") public String paymentZk(@RequestParam("username") String username) throws TException { return "SpringCloud with zookeeper: " + serverPort + "\t" + UUID.randomUUID() + "-" + helloWorldService.sayHello(username); } }
2.8 启动zk
zhinian@192 bin % ./zkServer.sh start ZooKeeper JMX enabled by default Using config: /Users/zhinian/Desktop/tools/zookeeper-3.4.11/bin/../conf/zoo.cfg Starting zookeeper ... STARTED
2.9 启动AB服务进行测试
调用成功 !
猜你喜欢
- 3小时前Android Proguard混淆
- 3小时前从一到无穷大 #21 从基于多数据模型分析负载的Benchmark讨论多模数据库的发展方向
- 3小时前【开源】基于JAVA的河南软件客服系统
- 3小时前【大数据处理技术实践】期末考查题目:集群搭建、合并文件与数据统计可视化
- 3小时前基于Ray和vLLM构建70B+模型的开源RLHF全量训练框架
- 3小时前JAVA RPC Thrift基操实现与微服务间调用
- 3小时前python的pandas中如何在dataframe中插入一行或一列数据?
- 3小时前【华为OD机考 统一考试机试C卷】城市聚集度找城市(C++ Java JavaScript Python)
- 3小时前UDP和TCP
- 3小时前Vue 封装的 axios 类的使用(小bug 改进)
网友评论
- 搜索
- 最新文章
- 热门文章