| % 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. |
| |
| -module(couch_external_server). |
| -behaviour(gen_server). |
| |
| -export([start_link/2, stop/1, execute/2]). |
| -export([init/1, terminate/2, handle_call/3, handle_cast/2, handle_info/2, code_change/3]). |
| |
| -include("couch_db.hrl"). |
| |
| % External API |
| |
| start_link(Name, Command) -> |
| gen_server:start_link(couch_external_server, [Name, Command], []). |
| |
| stop(Pid) -> |
| gen_server:cast(Pid, stop). |
| |
| execute(Pid, JsonReq) -> |
| gen_server:call(Pid, {execute, JsonReq}, infinity). |
| |
| % Gen Server Handlers |
| |
| init([Name, Command]) -> |
| ?LOG_INFO("EXTERNAL: Starting process for: ~s", [Name]), |
| ?LOG_INFO("COMMAND: ~s", [Command]), |
| Timeout = list_to_integer(couch_config:get("couchdb", "os_process_timeout", |
| "5000")), |
| {ok, Pid} = couch_os_process:start_link(Command, [{timeout, Timeout}]), |
| couch_config:register(fun("couchdb", "os_process_timeout", NewTimeout) -> |
| couch_os_process:set_timeout(Pid, list_to_integer(NewTimeout)) |
| end), |
| {ok, {Name, Command, Pid}}. |
| |
| terminate(_Reason, {_Name, _Command, Pid}) -> |
| couch_os_process:stop(Pid), |
| ok. |
| |
| handle_call({execute, JsonReq}, _From, {Name, Command, Pid}) -> |
| {reply, couch_os_process:prompt(Pid, JsonReq), {Name, Command, Pid}}. |
| |
| handle_info({'EXIT', Pid, Reason}, {Name, Command, Pid}) -> |
| ?LOG_INFO("EXTERNAL: Process for ~s exiting. (reason: ~w)", [Name, Reason]), |
| {stop, normal, {Name, Command, Pid}}. |
| |
| handle_cast(stop, {Name, Command, Pid}) -> |
| ?LOG_INFO("EXTERNAL: Shutting down ~s", [Name]), |
| exit(Pid, normal), |
| {stop, normal, {Name, Command, Pid}}; |
| handle_cast(_Whatever, State) -> |
| {noreply, State}. |
| |
| code_change(_OldVsn, State, _Extra) -> |
| {ok, State}. |
| |
| |