123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197 |
- /*
- Copyright (c) 2022 Sogou, Inc.
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
- http://www.apache.org/licenses/LICENSE-2.0
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- */
- #ifndef __SRPC_CONTROLLER__
- #define __SRPC_CONTROLLER__
- #include "srpc_config.h"
- class CommandController
- {
- public:
- bool parse_args(int argc, const char **argv);
- virtual bool dependencies_and_dir();
- virtual bool copy_files();
- virtual void print_success_info() const;
- virtual void print_usage(const char *name) const = 0;
- protected:
- virtual bool check_args();
- private:
- virtual bool get_opt(int argc, const char **argv) = 0;
- public:
- CommandController() { }
- virtual ~CommandController() { }
- using transform_function_t = bool (*)(const std::string&, FILE *,
- const struct srpc_config *);
- struct file_info
- {
- std::string in_file;
- std::string out_file;
- transform_function_t transform;
- };
- protected:
- std::vector<struct file_info> default_files;
- struct srpc_config config;
- protected:
- bool get_path(const char *file, char *path, int depth);
- bool copy_single_file(const std::string& in_file,
- const std::string& out_file,
- transform_function_t transform);
- void fill_rpc_default_files();
- };
- class HttpController : public CommandController
- {
- public:
- void print_usage(const char *name) const override;
- private:
- bool get_opt(int argc, const char **argv) override;
- public:
- HttpController();
- ~HttpController() { }
- };
- class RedisController : public CommandController
- {
- public:
- void print_usage(const char *name) const override;
- private:
- bool get_opt(int argc, const char **argv) override;
- public:
- RedisController();
- ~RedisController() { }
- };
- class RPCController : public CommandController
- {
- public:
- void print_usage(const char *name) const override;
- bool copy_files() override;
- protected:
- bool check_args() override;
- private:
- bool get_opt(int argc, const char **argv) override;
- public:
- RPCController();
- ~RPCController() { }
- };
- class APIController : public CommandController
- {
- public:
- bool dependencies_and_dir() override;
- void print_usage(const char *name) const override;
- void print_success_info() const override;
- protected:
- bool check_args() override { return true; }
- private:
- bool get_opt(int argc, const char **argv) override;
- public:
- APIController();
- ~APIController() { }
- };
- class ProxyController : public CommandController
- {
- public:
- void print_usage(const char *name) const override;
- void print_success_info() const override;
- bool copy_files() override;
- protected:
- bool check_args() override;
- private:
- bool get_opt(int argc, const char **argv) override;
- public:
- ProxyController();
- ~ProxyController() { }
- };
- class FileServiceController : public CommandController
- {
- public:
- void print_usage(const char *name) const override;
- void print_success_info() const override;
- private:
- bool get_opt(int argc, const char **argv) override;
- public:
- FileServiceController();
- ~FileServiceController() { }
- };
- class ComputeController : public CommandController
- {
- public:
- void print_usage(const char *name) const override;
- void print_success_info() const override;
- private:
- bool get_opt(int argc, const char **argv) override;
- public:
- ComputeController();
- ~ComputeController() { }
- };
- ////////// common functions //////////
- int mkdir_p(const char *name, mode_t mode);
- ////////// common transform functions //////////
- bool common_cmake_transform(const std::string& format, FILE *out,
- const struct srpc_config *config);
- bool basic_client_transform(const std::string& format, FILE *out,
- const struct srpc_config *config);
- bool basic_server_transform(const std::string& format, FILE *out,
- const struct srpc_config *config);
- bool basic_client_config_transform(const std::string& format, FILE *out,
- const struct srpc_config *config);
- bool basic_server_config_transform(const std::string& format, FILE *out,
- const struct srpc_config *config);
- bool rpc_idl_transform(const std::string& format, FILE *out,
- const struct srpc_config *config);
- bool rpc_client_transform(const std::string& format, FILE *out,
- const struct srpc_config *config);
- bool rpc_server_transform(const std::string& format, FILE *out,
- const struct srpc_config *config);
- bool rpc_cmake_transform(const std::string& format, FILE *out,
- const struct srpc_config *config);
- #endif
|