source: java/main/src/main/java/com/framsticks/communication/queries/CallRequest.java @ 193

package com.framsticks.communication.queries;

import com.framsticks.util.Misc;
import com.framsticks.util.lang.Pair;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

/**
 * @author Piotr Sniegowski
 */
public class CallRequest extends ApplicationRequest {

        protected final List<Object> arguments = new ArrayList<>();
        protected String procedure;

        public CallRequest addArguments(String arguments) {
                return this;
        }

        /**
         * @return the arguments
         */
        public List<Object> getArguments() {
                return arguments;
        }

        /**
         * @return the procedure
         */
        public String getProcedure() {
                return procedure;
        }

        /**
         * @param arguments the arguments to set
         */
        public CallRequest arguments(Collection<Object> arguments) {
                this.arguments.clear();
                this.arguments.addAll(arguments);
                return this;
        }

        public CallRequest argument(Object argument) {
                arguments.add(argument);
                return this;
        }

        public CallRequest procedure(String procedure) {
                this.procedure = procedure;
                return this;
        }

        @Override
        protected StringBuilder construct(StringBuilder buffer) {
                super.construct(buffer);
                Misc.throwIfNull(procedure);
                buffer.append(' ').append(procedure);
                for (Object arg : arguments) {
                        buffer.append(' ');
                        quoteArgumentIfNeeded(buffer, arg);
                }
                return buffer;
        }

        @Override
        public String getCommand() {
                return "call";
        }

        @Override
        public CharSequence parseRest(CharSequence rest) {
                rest = super.parseRest(rest);
                Pair<CharSequence, CharSequence> p = Misc.throwIfNull(takeIdentifier(rest));
                this.procedure = p.first.toString();
                while ((p = takeString(p.second)) != null) {
                        arguments.add(p.first);
                }
                return null;
        }

}