Exemple #1
0
 void send(Object[] args, TProtocol protocol, int seqid) throws TException {
   protocol.writeMessageBegin(new TMessage(methodName, TMessageType.CALL, seqid));
   protocol.writeStructBegin(argsStruct);
   // method with no parameters
   if (null != args) {
     for (int i = 0, n = args.length; i < n; i++) {
       FieldSpec fieldSpec = requestTypeList.get(i);
       Object value = args[i];
       if (value == null) {
         if (fieldSpec.required) {
           throw new TProtocolException(
               "Required field '"
                   + fieldSpec.name
                   + "' was not present! Struct: "
                   + argsStruct.name);
         }
       } else {
         protocol.writeFieldBegin(fieldSpec.tField);
         fieldSpec.typeAdapter.write(value, protocol);
         protocol.writeFieldEnd();
       }
     }
   }
   protocol.writeFieldStop();
   protocol.writeStructEnd();
   protocol.writeMessageEnd();
   protocol.getTransport().flush();
 }
  @Override
  public boolean process(TProtocol in, TProtocol out) throws TException {
    TMessage msg = in.readMessageBegin();
    String name = msg.name;
    int idx = name.indexOf('.');
    if (idx != -1) {
      name = name.substring(idx + 1);
    }
    TDynamicFunction f = descManager.getFunction(name);
    if (f == null) throw new TException("unknow function '" + name + "'");
    Reader reader = new Reader(f.getManager());
    Xdom param = reader.readStruct(f.getParamStruct(), in);
    in.readMessageEnd();

    if (!f.isOneway()) {

      XdomMap rep;
      TDynamicField retf = f.getReturnStruct().getField("success");
      if (retf.getTypeValue() != TType.VOID) {
        Xdom ret = handle(name, param);
        rep = new XdomMap();
        rep.put("success", ret);
      } else {
        rep = new XdomMap();
      }

      TMessage repmsg = new TMessage(name, TMessageType.REPLY, msg.seqid);
      out.writeMessageBegin(repmsg);
      Writer writer = new Writer(f.getManager());
      writer.writeStruct(f.getReturnStruct(), rep, out);
      out.writeMessageEnd();
      out.getTransport().flush();
    }
    return true;
  }
Exemple #3
0
 public void write_args(org.apache.thrift.protocol.TProtocol prot)
     throws org.apache.thrift.TException {
   prot.writeMessageBegin(
       new org.apache.thrift.protocol.TMessage(
           "hi", org.apache.thrift.protocol.TMessageType.CALL, 0));
   hi_args args = new hi_args();
   args.write(prot);
   prot.writeMessageEnd();
 }
 public void write_args(org.apache.thrift.protocol.TProtocol prot)
     throws org.apache.thrift.TException {
   prot.writeMessageBegin(
       new org.apache.thrift.protocol.TMessage(
           "getStruct", org.apache.thrift.protocol.TMessageType.CALL, 0));
   getStruct_args args = new getStruct_args();
   args.setKey(key);
   args.write(prot);
   prot.writeMessageEnd();
 }
 public void write_args(org.apache.thrift.protocol.TProtocol prot)
     throws org.apache.thrift.TException {
   prot.writeMessageBegin(
       new org.apache.thrift.protocol.TMessage(
           "show", org.apache.thrift.protocol.TMessageType.CALL, 0));
   show_args args = new show_args();
   args.setUser(user);
   args.write(prot);
   prot.writeMessageEnd();
 }
Exemple #6
0
 public void write_args(org.apache.thrift.protocol.TProtocol prot)
     throws org.apache.thrift.TException {
   prot.writeMessageBegin(
       new org.apache.thrift.protocol.TMessage(
           "call", org.apache.thrift.protocol.TMessageType.CALL, 0));
   call_args args = new call_args();
   args.setRequest(request);
   args.write(prot);
   prot.writeMessageEnd();
 }
 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.setName(name);
   args.write(prot);
   prot.writeMessageEnd();
 }
 public void write_args(org.apache.thrift.protocol.TProtocol prot)
     throws org.apache.thrift.TException {
   prot.writeMessageBegin(
       new org.apache.thrift.protocol.TMessage(
           "getDecision", org.apache.thrift.protocol.TMessageType.CALL, 0));
   getDecision_args args = new getDecision_args();
   args.setRequest(request);
   args.setSessionId(sessionId);
   args.write(prot);
   prot.writeMessageEnd();
 }
Exemple #9
0
 public void write_args(org.apache.thrift.protocol.TProtocol prot)
     throws org.apache.thrift.TException {
   prot.writeMessageBegin(
       new org.apache.thrift.protocol.TMessage(
           "tokenize", org.apache.thrift.protocol.TMessageType.CALL, 0));
   tokenize_args args = new tokenize_args();
   args.setDocument(document);
   args.setWordGran(wordGran);
   args.write(prot);
   prot.writeMessageEnd();
 }
Exemple #10
0
 public void write_args(org.apache.thrift.protocol.TProtocol prot)
     throws org.apache.thrift.TException {
   prot.writeMessageBegin(
       new org.apache.thrift.protocol.TMessage(
           "execute", org.apache.thrift.protocol.TMessageType.CALL, 0));
   execute_args args = new execute_args();
   args.set_functionName(functionName);
   args.set_funcArgs(funcArgs);
   args.write(prot);
   prot.writeMessageEnd();
 }
  public boolean process(TProtocol iprot, TProtocol oprot) throws TException {
    String ip = getClientIp(iprot);
    if (!whiteListHelper.check(ip)) {
      System.out.println(
          "------------ customized secure check, client ip (" + ip + ") not allowed");
      TMessage msg = iprot.readMessageBegin();
      TProtocolUtil.skip(iprot, TType.STRUCT);
      iprot.readMessageEnd();
      TApplicationException appException =
          new TApplicationException(TApplicationException.UNKNOWN, "Invalid ip: " + ip);
      oprot.writeMessageBegin(new TMessage(msg.name, TMessageType.EXCEPTION, msg.seqid));
      appException.write(oprot);
      oprot.writeMessageEnd();
      oprot.getTransport().flush();
      return false;
    }

    System.out.println("------------ customized secure check, client ip (" + ip + ") allowed");
    boolean result = super.process(iprot, oprot);
    return result;
  }
 public void writeMessageEnd() throws TException {
   delegate.writeMessageEnd();
 }
 protected void sendBase(String methodName, TBase args) throws TException {
   oprot_.writeMessageBegin(new TMessage(methodName, TMessageType.CALL, ++seqid_));
   args.write(oprot_);
   oprot_.writeMessageEnd();
   oprot_.getTransport().flush();
 }