functions_framework 0.0.0 → 0.1.0

data/lib/functions_framework/registry.rb

@@ -0,0 +1,137 @@
+# Copyright 2020 Google LLC
+#
+# 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
+#
+#     https://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.
+
+require "monitor"
+
+module FunctionsFramework
+  ##
+  # Registry providing lookup of functions by name.
+  #
+  class Registry
+    include ::MonitorMixin
+
+    ##
+    # Create a new empty registry.
+    #
+    def initialize
+      super()
+      @functions = {}
+    end
+
+    ##
+    # Look up a function definition by name.
+    #
+    # @param name [String] The function name
+    # @return [FunctionsFramework::Function] if the function is found
+    # @return [nil] if the function is not found
+    #
+    def [] name
+      @functions[name.to_s]
+    end
+
+    ##
+    # Returns the list of defined names
+    #
+    # @return [Array<String>]
+    #
+    def names
+      @functions.keys.sort
+    end
+
+    ##
+    # Add an HTTP function to the registry.
+    #
+    # You must provide a name for the function, and a block that implemets the
+    # function. The block should take a single `Rack::Request` argument. It
+    # should return one of the following:
+    #  *  A standard 3-element Rack response array. See
+    #     https://github.com/rack/rack/blob/master/SPEC
+    #  *  A `Rack::Response` object.
+    #  *  A simple String that will be sent as the response body.
+    #  *  A Hash object that will be encoded as JSON and sent as the response
+    #     body.
+    #
+    # @param name [String] The function name
+    # @param block [Proc] The function code as a proc
+    # @return [self]
+    #
+    def add_http name, &block
+      name = name.to_s
+      synchronize do
+        raise ::ArgumentError, "Function already defined: #{name}" if @functions.key? name
+        @functions[name] = Function.new name, :http, &block
+      end
+      self
+    end
+
+    ##
+    # Add a CloudEvent function to the registry.
+    #
+    # You must provide a name for the function, and a block that implemets the
+    # function. The block should take two arguments: the event _data_ and the
+    # event _context_. Any return value is ignored.
+    #
+    # The event data argument will be one of the following types:
+    #  *  A `String` (with encoding `ASCII-8BIT`) if the data is in the form of
+    #     binary data. You may choose to perform additional interpretation of
+    #     the binary data using information in the content type provided by the
+    #     context argument.
+    #  *  Any data type that can be represented in JSON (i.e. `String`,
+    #     `Integer`, `Array`, `Hash`, `true`, `false`, or `nil`) if the event
+    #     came with a JSON payload. The content type may also be set in the
+    #     context if the data is a String.
+    #
+    # The context argument will be of type {FunctionsFramework::CloudEvents::Event},
+    # and will contain CloudEvents context attributes such as `id` and `type`.
+    #
+    # See also {#add_cloud_event} which creates a function that takes a single
+    # argument of type {FunctionsFramework::CloudEvents::Event}.
+    #
+    # @param name [String] The function name
+    # @param block [Proc] The function code as a proc
+    # @return [self]
+    #
+    def add_event name, &block
+      name = name.to_s
+      synchronize do
+        raise ::ArgumentError, "Function already defined: #{name}" if @functions.key? name
+        @functions[name] = Function.new name, :event, &block
+      end
+      self
+    end
+
+    ##
+    # Add a CloudEvent function to the registry.
+    #
+    # You must provide a name for the function, and a block that implemets the
+    # function. The block should take _one_ argument: the event object of type
+    # {FunctionsFramework::CloudEvents::Event}. Any return value is ignored.
+    #
+    # See also {#add_event} which creates a function that takes data and
+    # context as separate arguments.
+    #
+    # @param name [String] The function name
+    # @param block [Proc] The function code as a proc
+    # @return [self]
+    #
+    def add_cloud_event name, &block
+      name = name.to_s
+      synchronize do
+        raise ::ArgumentError, "Function already defined: #{name}" if @functions.key? name
+        @functions[name] = Function.new name, :cloud_event, &block
+      end
+      self
+    end
+  end
+end

data/lib/functions_framework/server.rb

@@ -0,0 +1,423 @@
+# Copyright 2020 Google LLC
+#
+# 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
+#
+#     https://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.
+
+require "json"
+require "monitor"
+
+require "puma"
+require "puma/server"
+require "rack"
+
+module FunctionsFramework
+  ##
+  # A web server that wraps a function.
+  #
+  class Server
+    include ::MonitorMixin
+
+    ##
+    # Create a new web server given a function. Yields a
+    # {FunctionsFramework::Server::Config} object that you can use to set
+    # server configuration parameters. This block is the only opportunity to
+    # set configuration; once the server is initialized, configuration is
+    # frozen.
+    #
+    # @param function [FunctionsFramework::Function] The function to execute.
+    # @yield [FunctionsFramework::Server::Config] A config object that can be
+    #     manipulated to configure this server.
+    #
+    def initialize function
+      super()
+      @config = Config.new
+      yield @config if block_given?
+      @config.freeze
+      @function = function
+      @app =
+        case function.type
+        when :http
+          HttpApp.new function, @config
+        when :event, :cloud_event
+          EventApp.new function, @config
+        else
+          raise "Unrecognized function type: #{function.type}"
+        end
+      @server = nil
+      @signals_installed = false
+    end
+
+    ##
+    # The function to execute.
+    # @return [FunctionsFramework::Function]
+    #
+    attr_reader :function
+
+    ##
+    # The final configuration. This is a frozen object that cannot be modified.
+    # @return [FunctionsFramework::Server::Config]
+    #
+    attr_reader :config
+
+    ##
+    # Start the web server in the background. Does nothing if the web server
+    # is already running.
+    #
+    # @return [self]
+    #
+    def start
+      synchronize do
+        unless running?
+          @server = ::Puma::Server.new @app
+          @server.min_threads = @config.min_threads
+          @server.max_threads = @config.max_threads
+          @server.leak_stack_on_error = @config.show_error_details?
+          @server.binder.add_tcp_listener @config.bind_addr, @config.port
+          @server.run true
+          @config.logger.info "FunctionsFramework: Serving function #{@function.name.inspect}" \
+                              " on port #{@config.port}..."
+        end
+      end
+      self
+    end
+
+    ##
+    # Stop the web server in the background. Does nothing if the web server
+    # is not running.
+    #
+    # @param force [Boolean] Use a forced halt instead of a graceful shutdown
+    # @param wait [Boolean] Block until shutdown is complete
+    # @return [self]
+    #
+    def stop force: false, wait: false
+      synchronize do
+        if running?
+          @config.logger.info "FunctionsFramework: Shutting down server..."
+          if force
+            @server.halt wait
+          else
+            @server.stop wait
+          end
+        end
+      end
+      self
+    end
+
+    ##
+    # Wait for the server to stop. Returns immediately if the server is not
+    # running.
+    #
+    # @param timeout [nil,Numeric] The timeout. If `nil` (the default), waits
+    #     indefinitely, otherwise times out after the given number of seconds.
+    # @return [self]
+    #
+    def wait_until_stopped timeout: nil
+      @server&.thread&.join timeout
+      self
+    end
+
+    ##
+    # Determine if the web server is currently running
+    #
+    # @return [Boolean]
+    #
+    def running?
+      @server&.thread&.alive?
+    end
+
+    ##
+    # Cause this server to respond to SIGTERM, SIGINT, and SIGHUP by shutting
+    # down gracefully.
+    #
+    # @return [self]
+    #
+    def respond_to_signals
+      synchronize do
+        return self if @signals_installed
+        ::Signal.trap "SIGTERM" do
+          Server.signal_enqueue "SIGTERM", @config.logger, @server
+        end
+        ::Signal.trap "SIGINT" do
+          Server.signal_enqueue "SIGINT", @config.logger, @server
+        end
+        ::Signal.trap "SIGHUP" do
+          Server.signal_enqueue "SIGHUP", @config.logger, @server
+        end
+        @signals_installed = true
+      end
+      self
+    end
+
+    class << self
+      ## @private
+      def start_signal_queue
+        @signal_queue = ::Queue.new
+        ::Thread.start do
+          loop do
+            signal, logger, server = @signal_queue.pop
+            logger.info "FunctionsFramework: Caught #{signal}; shutting down server..."
+            server&.stop
+          end
+        end
+      end
+
+      ## @private
+      def signal_enqueue signal, logger, server
+        @signal_queue << [signal, logger, server]
+      end
+    end
+
+    start_signal_queue
+
+    ##
+    # The web server configuration. This object is yielded from the
+    # {FunctionsFramework::Server} constructor and can be modified at that
+    # point. Afterward, it is available from {FunctionsFramework::Server#config}
+    # but it is frozen.
+    #
+    class Config
+      ##
+      # Create a new config object with the default settings
+      #
+      def initialize
+        self.rack_env = nil
+        self.bind_addr = nil
+        self.port = nil
+        self.min_threads = nil
+        self.max_threads = nil
+        self.show_error_details = nil
+        self.logger = nil
+      end
+
+      ##
+      # Set the Rack environment, or `nil` to use the default.
+      # @param rack_env [String,nil]
+      #
+      def rack_env= rack_env
+        @rack_env = rack_env || ::ENV["RACK_ENV"] ||
+                    (::ENV["K_REVISION"] ? "production" : "development")
+      end
+
+      ##
+      # Set the bind address, or `nil` to use the default.
+      # @param bind_addr [String,nil]
+      #
+      def bind_addr= bind_addr
+        @bind_addr = bind_addr || ::ENV["BIND_ADDR"] || "0.0.0.0"
+      end
+
+      ##
+      # Set the port number, or `nil` to use the default.
+      # @param port [Integer,nil]
+      #
+      def port= port
+        @port = (port || ::ENV["PORT"] || 8080).to_i
+      end
+
+      ##
+      # Set the minimum number of worker threads, or `nil` to use the default.
+      # @param min_threads [Integer,nil]
+      #
+      def min_threads= min_threads
+        @min_threads = (min_threads || ::ENV["MIN_THREADS"])&.to_i
+      end
+
+      ##
+      # Set the maximum number of worker threads, or `nil` to use the default.
+      # @param max_threads [Integer,nil]
+      #
+      def max_threads= max_threads
+        @max_threads = (max_threads || ::ENV["MAX_THREADS"])&.to_i
+      end
+
+      ##
+      # Set whether to show detailed error messages, or `nil` to use the default.
+      # @param show_error_details [Boolean,nil]
+      #
+      def show_error_details= show_error_details
+        val = show_error_details.nil? ? ::ENV["DETAILED_ERRORS"] : show_error_details
+        @show_error_details = val ? true : false
+      end
+
+      ##
+      # Set the logger for server messages, or `nil` to use the global default.
+      # @param logger [Logger]
+      #
+      def logger= logger
+        @logger = logger || ::FunctionsFramework.logger
+      end
+
+      ##
+      # Returns the current Rack environment.
+      # @return [String]
+      #
+      def rack_env
+        @rack_env
+      end
+
+      ##
+      # Returns the current bind address.
+      # @return [String]
+      #
+      def bind_addr
+        @bind_addr
+      end
+
+      ##
+      # Returns the current port number.
+      # @return [Integer]
+      #
+      def port
+        @port
+      end
+
+      ##
+      # Returns the minimum number of worker threads in the thread pool.
+      # @return [Integer]
+      #
+      def min_threads
+        @min_threads || 1
+      end
+
+      ##
+      # Returns the maximum number of worker threads in the thread pool.
+      # @return [Integer]
+      #
+      def max_threads
+        @max_threads || (@rack_env == "development" ? 1 : 16)
+      end
+
+      ##
+      # Returns whether to show detailed error messages.
+      # @return [Boolean]
+      #
+      def show_error_details?
+        @show_error_details.nil? ? (@rack_env == "development") : @show_error_details
+      end
+
+      ##
+      # Returns the logger.
+      # @return [Logger]
+      #
+      def logger
+        @logger
+      end
+    end
+
+    ## @private
+    class AppBase
+      def initialize config
+        @config = config
+      end
+
+      def interpret_response response
+        case response
+        when ::Array
+          response
+        when ::Rack::Response
+          response.finish
+        when ::String
+          string_response response, "text/plain", 200
+        when ::Hash
+          json = ::JSON.dump response
+          string_response json, "application/json", 200
+        when ::StandardError
+          error = error_message response
+          string_response error, "text/plain", 500
+        else
+          e = ::StandardError.new "Unexpected response type: #{response.class}"
+          error = error_message e
+          string_response error, "text/plain", 500
+        end
+      end
+
+      def string_response string, content_type, status
+        headers = {
+          "Content-Type"   => content_type,
+          "Content-Length" => string.bytesize
+        }
+        [status, headers, [string]]
+      end
+
+      def error_message error
+        if @config.show_error_details?
+          "#{error.class}: #{error.message}\n#{error.backtrace}\n"
+        else
+          "Unexpected internal error"
+        end
+      end
+
+      def usage_message error
+        if @config.show_error_details?
+          "Failed to decode CloudEvent: #{error.inspect}"
+        else
+          "Failed to decode CloudEvent"
+        end
+      end
+    end
+
+    ## @private
+    class HttpApp < AppBase
+      def initialize function, config
+        super config
+        @function = function
+      end
+
+      def call env
+        response =
+          begin
+            logger = env["rack.logger"] = @config.logger
+            request = ::Rack::Request.new env
+            logger.info "FunctionsFramework: Handling HTTP #{request.request_method} request"
+            @function.call request
+          rescue ::StandardError => e
+            logger.warn e
+            e
+          end
+        interpret_response response
+      end
+    end
+
+    ## @private
+    class EventApp < AppBase
+      def initialize function, config
+        super config
+        @function = function
+      end
+
+      def call env
+        logger = env["rack.logger"] = @config.logger
+        event =
+          begin
+            CloudEvents.decode_rack_env env
+          rescue ::StandardError => e
+            e
+          end
+        response =
+          if event.is_a? CloudEvents::Event
+            logger.info "FunctionsFramework: Handling CloudEvent"
+            begin
+              @function.call event
+              "ok"
+            rescue ::StandardError => e
+              logger.warn e
+              e
+            end
+          else
+            logger.warn e.inspect
+            string_response usage_message(e), "text/plain", 400
+          end
+        interpret_response response
+      end
+    end
+  end
+end