pgbouncerhero 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 054de6f1d8d3846419f355c4512129e06385c8af
+  data.tar.gz: 1254bddb23ead79c688f283b07e179b41a328fd9
+SHA512:
+  metadata.gz: cf35ed3c426b20262a6be30a555eba7547c8cd1c89db7fe200a76babe5f4842298df9b0b7eb4ce0a9b3efb555d46c392c67627f4aed826417997ff4830fe9a71
+  data.tar.gz: fb73c61c6dda241124a59db1820083cfcf24d4cf25ae16f3c524feb7f05485d1054ec74bed889a2faad6847773603aec8d565003f4d81f8468e92de8c9374204

data/.gitignore

@@ -0,0 +1,18 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+*.lock
+InstalledFiles
+_yardoc
+coverage
+lib/bundler/man
+pkg
+rdoc
+tmp
+*.bundle
+*.so
+*.o
+*.a
+mkmf.log

data/CHANGELOG.md

@@ -0,0 +1,3 @@
+## 0.1.0
+
+- First major release

data/Gemfile

@@ -0,0 +1,4 @@
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in pgbouncerhero.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,24 @@
+Copyright (c) 2017 Quentin Rousseau <contact@quent.in>
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person
+obtaining a copy of this software and associated documentation
+files (the "Software"), to deal in the Software without
+restriction, including without limitation the rights to use,
+copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the
+Software is furnished to do so, subject to the following
+conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
+OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
+HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,125 @@
+# PgBouncerHero
+
+A graphical user interface for your PGBouncers.
+
+[See it in action](https://pgbouncerhero-demo.herokuapp.com/). [Source Code](https://github.com/kwent/pgbouncerhero-demo).
+
+[![Screenshot1](https://github.com/kwent/pgbouncerhero/blob/master/doc/screenshot-1.png?raw=true)](https://pgbouncerhero-demo.herokuapp.com/)
+[![Screenshot2](https://github.com/kwent/pgbouncerhero/blob/master/doc/screenshot-2.png?raw=true)](https://pgbouncerhero-demo.herokuapp.com/)
+[![Screenshot2](https://github.com/kwent/pgbouncerhero/blob/master/doc/screenshot-3.png?raw=true)](https://pgbouncerhero-demo.herokuapp.com/)
+
+## Installation
+
+PgBouncerHero is available as a Rails engine.
+
+Add those dependencies to your application’s Gemfile:
+
+```ruby
+gem 'pgbouncerhero'
+gem 'jquery-rails'
+gem 'sass-rails'
+gem 'semantic-ui-sass'
+gem 'haml-rails'
+gem 'pg'
+```
+
+And mount the engine in your `config/routes.rb`:
+
+```ruby
+mount PgBouncerHero::Engine, at: "pgbouncerhero"
+```
+
+Add the following line in your `application.css`:
+
+```
+*= require pgbouncerhero/application
+```
+
+Add the following lines in your `application.js`:
+
+```
+//= require jquery
+//= require jquery_ujs
+//= require pgbouncerhero/application
+```
+
+### Basic Authentication
+
+Set the following variables in your environment or an initializer.
+
+```ruby
+ENV["PGBOUNCERHERO_USERNAME"] = "zelda"
+ENV["PGBOUNCERHERO_PASSWORD"] = "triforce"
+```
+
+### Devise
+
+```ruby
+authenticate :user, -> (user) { user.admin? } do
+  mount PgBouncerHero::Engine, at: "pgbouncerhero"
+end
+```
+
+## One PgBouncer
+
+```bash
+export PGBOUNCERHERO_DATABASE_URL=postgres://user:password@host:port/pgbouncer
+```
+
+## Multiple PgBouncers
+
+Create `config/pgbouncerhero.yml` with:
+
+```yml
+default: &default
+  pgbouncers:
+    production:
+      master:
+        url: <%= ENV["PGBOUNCER_PRODUCTION_MASTER_DATABASE_URL"] %>
+      slave:
+        url: <%= ENV["PGBOUNCER_PRODUCTION_SLAVE_DATABASE_URL"] %>
+    staging:
+      master:
+        url: <%= ENV["PGBOUNCER_STAGING_MASTER_DATABASE_URL"] %>
+      slave:
+        url: <%= ENV["PGBOUNCER_STAGING_SLAVE_DATABASE_URL"] %>
+
+development:
+  <<: *default
+
+production:
+  <<: *default
+```
+
+# Authors
+
+- [Quentin Rousseau](https://github.com/kwent)
+
+# License
+
+```plain
+Copyright (c) 2017 Quentin Rousseau <contact@quent.in>
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person
+obtaining a copy of this software and associated documentation
+files (the "Software"), to deal in the Software without
+restriction, including without limitation the rights to use,
+copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the
+Software is furnished to do so, subject to the following
+conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
+OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
+HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+OTHER DEALINGS IN THE SOFTWARE.
+```

data/app/assets/javascripts/pgbouncerhero/application.js

@@ -0,0 +1,2 @@
+//= require jquery_ujs
+//= require semantic-ui

data/app/assets/stylesheets/pgbouncerhero/application.css.scss

@@ -0,0 +1,15 @@
+@import "semantic-ui";
+
+body {
+  margin: 0;
+  padding-top: 50px;
+  background-color: #eee;
+}
+
+.ui.main.container {
+  padding: 10px;
+}
+
+.wireframe.image {
+  opacity: 0.5;
+}

data/app/controllers/pg_bouncer_hero/application_controller.rb

@@ -0,0 +1,27 @@
+module PgBouncerHero
+  class ApplicationController < ActionController::Base
+    layout "pg_bouncer_hero/application"
+
+    protect_from_forgery
+
+    http_basic_authenticate_with name: ENV["PGBOUNCERHERO_USERNAME"], password: ENV["PGBOUNCERHERO_PASSWORD"] if ENV["PGBOUNCERHERO_PASSWORD"]
+
+    if respond_to?(:before_action)
+      before_action :set_database
+    else
+      before_filter :set_database
+    end
+
+    protected
+
+    def set_database
+      @groups = PgBouncerHero.groups
+      if params[:group] && params[:database]
+        @database = PgBouncerHero.groups[params[:group]].databases.find { |db| db.id.to_s == params[:database].to_s }
+      else
+        @group = @groups.first
+        @database = @groups.first.last.databases.first
+      end
+    end
+  end
+end

data/app/controllers/pg_bouncer_hero/database_controller.rb

@@ -0,0 +1,70 @@
+module PgBouncerHero
+  class DatabaseController < ApplicationController
+    def summary
+      if @database.connection
+        @dbs = @database.summary
+      else
+        flash[:error] = "#{@database.name} does not look online."
+      end
+    end
+    def databases
+      if @database.connection
+        @dbs = @database.databases
+      else
+        flash[:error] = "#{@database.name} does not look online."
+      end
+    end
+    def stats
+      if @database.connection
+        @stats = @database.stats
+      else
+        flash[:error] = "#{@database.name} does not look online."
+      end
+    end
+    def pools
+      if @database.connection
+        @pools = @database.pools
+      else
+        flash[:error] = "#{@database.name} does not look online."
+      end
+    end
+    def clients
+      if @database.connection
+        @clients = @database.clients
+      else
+        flash[:error] = "#{@database.name} does not look online."
+      end
+    end
+    def conf
+      if @database.connection
+        @conf = @database.conf
+      else
+        flash[:error] = "#{@database.name} does not look online."
+      end
+    end
+    def reload
+      if @database.connection
+        @database.reload
+        flash[:success] = "#{@database.name} has been reloaded."
+      else
+        flash[:error] = "#{@database.name} does not look online."
+      end
+    end
+    def suspend
+      if @database.connection
+        @database.suspend
+        flash[:success] = "#{@database.name} has been suspended."
+      else
+        flash[:error] = "#{@database.name} does not look online."
+      end
+    end
+    def shutdown
+      if @database.connection
+        @database.shutdown
+        flash[:success] = "#{@database.name} has been shutdown."
+      else
+        flash[:error] = "#{@database.name} does not look online."
+      end
+    end
+  end
+end

data/app/controllers/pg_bouncer_hero/home_controller.rb

@@ -0,0 +1,6 @@
+module PgBouncerHero
+  class HomeController < ApplicationController
+    def index
+    end
+  end
+end

data/app/helpers/pg_bouncer_hero/application_helper.rb

@@ -0,0 +1,29 @@
+module PgBouncerHero
+  module ApplicationHelper
+    def is_active(action_name)
+      params[:action] == action_name ? 'active' : nil
+    end
+    def alert_class_for(flash_type)
+      case flash_type
+      when 'success'
+        'success'
+      when 'error'
+        'error'
+      when 'notice'
+        'info'
+      when 'warning'
+        'warning'
+      else
+        nil
+      end
+    end
+    def humanize_ms(millis)
+      [[1000, :ms], [60, :s], [60, :min], [24, :h], [1000, :d]].map{ |count, name|
+        if millis > 0
+          millis, n = millis.divmod(count)
+          "#{n.to_i} #{name}"
+        end
+      }.compact.reverse.join(' ')
+    end
+  end
+end

data/app/views/layouts/pg_bouncer_hero/application.html.haml

@@ -0,0 +1,20 @@
+!!!
+%html
+  %head
+    %meta{charset: "utf-8"}
+    %meta{content: "IE=edge,chrome=1", "http-equiv" => "X-UA-Compatible"}
+    %meta{content: "width=device-width, initial-scale=1.0, maximum-scale=1.0", name: "viewport"}
+    %title= [@groups.size > 1 ? "#{@database.group.name} | #{@database.name}" : "PgBouncerHero", @title].compact.join(" / ")
+    = stylesheet_link_tag 'application', media: 'all', 'data-turbolinks-track': true
+    = csrf_meta_tags
+  %body
+    .ui.fixed.menu
+      .ui.container
+        = link_to'PGBouncerHero', root_path, class: 'header item'
+    .pusher
+      .ui.main.container
+        #flash.row= render partial: 'shared/flash_messages', flash: flash
+        = yield
+      = javascript_include_tag 'application'
+      - if content_for? :inline_script
+        = yield :inline_script

data/app/views/pg_bouncer_hero/database/_actions.html.haml

@@ -0,0 +1,10 @@
+.ui.active.segment
+
+  %h4.ui.header
+    .content Actions
+
+  %button.ui.negative.basic.button Pause
+  %button.ui.positive.basic.button Resume
+  %button.ui.negative.basic.button Disable
+  %button.ui.positive.basic.button Enable
+  %button.ui.negative.basic.button Kill

data/app/views/pg_bouncer_hero/database/_clients.html.haml

@@ -0,0 +1,15 @@
+.ui.active.segment
+
+  %h4.ui.header
+    .content Clients
+
+  %table.ui.compact.table
+    %thead
+      %tr
+        - clients.first.keys.each do |key|
+          %th= key.titleize
+    %tbody
+      - clients.each do |row|
+        %tr
+          - row.each do |k,v|
+            %td= v

data/app/views/pg_bouncer_hero/database/_conf.html.haml

@@ -0,0 +1,15 @@
+.ui.active.segment
+
+  %h4.ui.header
+    .content Confs
+
+  %table.ui.compact.table
+    %thead
+      %tr
+        - conf.first.keys.each do |key|
+          %th= key.titleize
+    %tbody
+      - conf.each do |row|
+        %tr
+          - row.each do |k,v|
+            %td= v

data/app/views/pg_bouncer_hero/database/_databases.html.haml

@@ -0,0 +1,15 @@
+.ui.active.segment
+
+  %h4.ui.header
+    .content Databases
+
+  %table.ui.compact.table
+    %thead
+      %tr
+        - databases.first.keys.each do |key|
+          %th= key.titleize
+    %tbody
+      - databases.each do |row|
+        %tr
+          - row.each do |k,v|
+            %td= v

data/app/views/pg_bouncer_hero/database/_menu.html.haml

@@ -0,0 +1,15 @@
+.ui.pointing.menu
+  .header.item
+    = "#{database.group.name} > #{database.name}"
+  = link_to "Databases", databases_path(database: database.name.parameterize), class: "item #{is_active('databases')}"
+  = link_to "Stats", stats_path(database: database.name.parameterize), class: "item #{is_active('stats')}"
+  = link_to "Pools", pools_path(database: database.name.parameterize), class: "item #{is_active('pools')}"
+  = link_to "Clients", clients_path(database: database.name.parameterize), class: "item #{is_active('clients')}"
+  = link_to "Configuration", conf_path(database: database.name.parameterize), class: "item #{is_active('conf')}"
+  .right.menu
+    .item
+      = link_to 'Reload', reload_path(database: database.name.parameterize), remote: :true, method: :post, class: 'ui button basic positive button', data: { confirm: 'Are you sure?' }
+    .item
+      = link_to 'Suspend', suspend_path(database: database.name.parameterize), remote: :true, method: :post, class: 'ui button basic primary button', data: { confirm: 'Are you sure?' }
+    .item
+      = link_to 'Shutdown', shutdown_path(database: database.name.parameterize), remote: :true, method: :post, class: 'ui button basic negative button', data: { confirm: 'Are you sure?' }

data/app/views/pg_bouncer_hero/database/_pools.html.haml

@@ -0,0 +1,15 @@
+.ui.active.segment
+
+  %h4.ui.header
+    .content Pools
+
+  %table.ui.compact.table
+    %thead
+      %tr
+        - pools.first.keys.each do |key|
+          %th= key.titleize
+    %tbody
+      - pools.each do |row|
+        %tr
+          - row.each do |k,v|
+            %td= v

data/app/views/pg_bouncer_hero/database/_stats.html.haml

@@ -0,0 +1,20 @@
+.ui.active.segment
+
+  %h4.ui.header
+    .content Stats
+
+  %table.ui.compact.table
+    %thead
+      %tr
+        - stats.first.keys.each do |key|
+          %th= key.titleize
+    %tbody
+      - stats.each do |row|
+        %tr
+          - row.each do |k,v|
+            - if ['total_received', 'total_sent', 'avg_recv', 'avg_sent'].include? k
+              %td= number_to_human_size(v)
+            - elsif ['total_query_time', 'avg_query'].include? k
+              %td= humanize_ms(v.to_i)
+            - else
+              %td= v

data/app/views/pg_bouncer_hero/database/clients.html.haml

@@ -0,0 +1,4 @@
+= render partial: "menu", locals: {database: @database}
+-# = render partial: "actions", locals: {database: @database}
+- if @clients
+  = render partial: "clients", locals: {database: @database, clients: @clients}

data/app/views/pg_bouncer_hero/database/conf.html.haml

@@ -0,0 +1,4 @@
+= render partial: "menu", locals: {database: @database}
+-# = render partial: "actions", locals: {database: @database}
+- if @conf
+  = render partial: "conf", locals: {database: @database, conf: @conf}

data/app/views/pg_bouncer_hero/database/databases.html.haml

@@ -0,0 +1,5 @@
+= render partial: "menu", locals: {database: @database}
+-# = render partial: "actions", locals: {database: @database}
+
+- if @dbs
+  = render partial: "databases", locals: {database: @database, databases: @dbs}

data/app/views/pg_bouncer_hero/database/pools.html.haml

@@ -0,0 +1,4 @@
+= render partial: "menu", locals: {database: @database}
+-# = render partial: "actions", locals: {database: @database}
+- if @pools
+  = render partial: "pools", locals: {database: @database, pools: @pools}

data/app/views/pg_bouncer_hero/database/stats.html.haml

@@ -0,0 +1,4 @@
+= render partial: "menu", locals: {database: @database}
+-# = render partial: "actions", locals: {database: @database}
+- if @stats
+  = render partial: "stats", locals: {database: @database, stats: @stats}

data/app/views/pg_bouncer_hero/database/summary.js.haml

@@ -0,0 +1,3 @@
+$("#segment_content_#{@database.group.name.parameterize}_#{@database.name.parameterize}").replaceWith("#{j(render partial: 'pg_bouncer_hero/home/card_content', locals: {database: @database, id: "segment_content_#{@database.group.name.parameterize}_#{@database.name.parameterize}"})}")
+$("#segment_refreshed_#{@database.group.name.parameterize}_#{@database.name.parameterize}").css('display', '')
+$("#segment_content_#{@database.group.name.parameterize}_#{@database.name.parameterize} .ui.progress").each(function(idx, el) { $(el).progress({ text: {active: '{value} of {total} connections'}, autoSuccess: false }); });

data/app/views/pg_bouncer_hero/home/_card.html.haml

@@ -0,0 +1,15 @@
+- connected = database.connection
+.eight.wide.column{id: "segment_href_#{database.group.name.parameterize}_#{database.name.parameterize}", data: {href: summary_path(group: database.group.name.parameterize, database: database.name.parameterize)}}
+  = link_to databases_path(group: database.group.name.parameterize, database: database.name.parameterize), style: 'text-decoration: none !important; color: inherit !important' do
+    .ui.segments{style: 'margin-bottom: 10px;'}
+      .ui.segment{style: 'height: 60px;'}
+        %div{style: 'float: left;'}
+          %div{style: 'font-size: 16px; font-weight: bold;'}= database.name
+          %div{style: 'font-size: 12px; color: grey'}= database.host
+        %button.ui.button.basic.button{style: 'float: right;', class: connected ? 'positive': 'negative'}= connected ? 'Online' : 'Offline'
+      - if connected
+        = render partial: partial, locals: {database: database, id: "segment_content_#{database.group.name.parameterize}_#{database.name.parameterize}"}
+        .ui.segment{id: "segment_refreshed_#{database.group.name.parameterize}_#{database.name.parameterize}", style: 'height: 35px; display: none;'}
+          %span.right.floated{id: "segment_refreshed_span_#{database.group.name.parameterize}_#{database.name.parameterize}", style: 'float:right; margin-top: -6px; color: grey'}
+            Last refresh:
+            = Time.now.strftime("%H:%M:%S GMT%z (%Z)") # Same as new Date().toTimeString() in JS

data/app/views/pg_bouncer_hero/home/_card_content.html.haml

@@ -0,0 +1,26 @@
+.ui.horizontal.segments{id: id, style: 'background-color: white !important;'}
+  .ui.segment
+    - summary = database.summary
+    %div
+      %i.users.icon
+      - number = summary.select { |row| row['list'] == 'users' }.first['items']
+      = pluralize(number, 'user')
+    %div
+      %i.database.icon
+      - number = summary.select { |row| row['list'] == 'databases' }.first['items'].to_i - 1 # Removed pgbouncer
+      = pluralize(number, 'database')
+    %div
+      %i.plug.icon
+      - number = summary.select { |row| row['list'] == 'pools' }.first['items'].to_i - 1 # Removed pgbouncer
+      = pluralize(number, 'pool')
+  .ui.segment
+    %ul{style: 'margin: 0; padding-left: 5px;'}
+      - summary.select{ |row| row.key?(:databases_details) }.first[:databases_details].each do |db|
+        %li{style: 'margin: 0; list-style-type: none;'}
+          %span
+            %i.database.icon
+            %strong=db['name']
+            .ui.progress.small.success{data:{value: db['current_connections'], total: db['max_connections']}, style: 'float:right; width: 150px; margin-top:4px;'}
+              .bar
+                .progress
+              .label Connections

data/app/views/pg_bouncer_hero/home/_card_loading_content.html.haml

@@ -0,0 +1,5 @@
+.ui.segment{id: id}
+  .ui.active.inverted.dimmer
+    .ui.medium.text.loader Loading...
+  = image_tag('pgbouncerhero/short-paragraph.png', class: 'ui wireframe image')
+  = image_tag('pgbouncerhero/short-paragraph.png', class: 'ui wireframe image')

data/app/views/pg_bouncer_hero/home/index.html.haml

@@ -0,0 +1,31 @@
+.ui.pointing.menu
+  .header.item
+    Overview
+
+.ui.grid
+  - @groups.each do |_, group|
+    .sixteen.wide.row
+      .sixteen.wide.column
+        %h3.ui.dividing.header{style: 'margin-bottom: 10px;'}= group.name
+      = render partial: "card", collection: group.databases, as: :database, locals: {partial: 'card_loading_content'}
+
+:javascript
+  document.addEventListener("DOMContentLoaded", function(event) {
+    // First
+    $("[id^='segment_href']").each(function(idx, el) {
+      $.get($(el).data('href'));
+    });
+    // Refresh every minutes
+    setInterval(function (){
+      $("[id^='segment_href']").each(function(idx, el) {
+        var el = $(el)
+        var id = el.attr('id');
+        var href = el.data('href');
+        var refreshed_id = id.replace('href', 'refreshed_span');
+        document.getElementById(refreshed_id).innerHTML = 'Refreshing...';
+        $.get(href, function(data) {
+          document.getElementById(refreshed_id).innerHTML = 'Last refresh: ' + new Date().toTimeString();
+        });
+      });
+    }, 60 * 1000)
+  });

data/app/views/shared/_alert.html.haml

@@ -0,0 +1,4 @@
+.alert{class: "ui #{class_name} message"}
+  - if defined?(closable) && closable && closable == true
+    %i.close.icon
+  = message.html_safe

data/app/views/shared/_flash_messages.html.haml

@@ -0,0 +1,2 @@
+- flash.each do |type, message|
+  = render partial: 'shared/alert', locals: {class_name: alert_class_for(type), message: message}

data/config/routes.rb

@@ -0,0 +1,16 @@
+PgBouncerHero::Engine.routes.draw do
+  root to: "home#index"
+  scope path: ":group", constraints: proc { |req| (PgBouncerHero.groups.keys.map(&:parameterize) + [nil]).include?(req.params[:group]) } do
+    scope path: ":database", constraints: proc { |req| (PgBouncerHero.groups[req.params[:group]].databases.map(&:name).map(&:parameterize) + [nil]).include?(req.params[:database]) } do
+      get :summary, controller: :database
+      get :databases, controller: :database
+      get :stats, controller: :database
+      get :pools, controller: :database
+      get :clients, controller: :database
+      get :conf, controller: :database
+      post :reload, controller: :database
+      post :suspend, controller: :database
+      post :shutdown, controller: :database
+    end
+  end
+end

data/lib/generators/pgbouncerhero/config_generator.rb

@@ -0,0 +1,13 @@
+require "rails/generators"
+
+module PgBouncerHero
+  module Generators
+    class ConfigGenerator < Rails::Generators::Base
+      source_root File.expand_path("../templates", __FILE__)
+
+      def create_initializer
+        template "config.yml", "config/pgbouncerhero.yml"
+      end
+    end
+  end
+end

data/lib/generators/pgbouncerhero/templates/config.yml

@@ -0,0 +1,16 @@
+pgbouncers:
+  production:
+    master:
+      # eg. postgres://user:password@host:port/pgbouncer
+      # url: <%= ENV["PGBOUNCER_PRODUCTION_MASTER_DATABASE_URL"] %>
+    # Add more databases
+    # slave:
+    #   url: <%= ENV["PGBOUNCER_PRODUCTION_SLAVE_DATABASE_URL"] %>
+  # staging:
+  #   master:
+  #     url: <%= ENV["PGBOUNCER_STAGING_MASTER_DATABASE_URL"] %>
+  #   slave:
+  #     url: <%= ENV["PGBOUNCER_STAGING_SLAVE_DATABASE_URL"] %>
+
+# Time zone (defaults to app time zone)
+# time_zone: "Pacific Time (US & Canada)"

data/lib/pgbouncerhero.rb

@@ -0,0 +1,65 @@
+require "pgbouncerhero/version"
+require "active_record"
+
+# methods
+require "pgbouncerhero/methods/basics"
+
+require "pgbouncerhero/group"
+require "pgbouncerhero/engine" if defined?(Rails)
+
+# models
+require "pgbouncerhero/connection"
+
+module PgBouncerHero
+  # settings
+  class << self
+    attr_accessor :env
+  end
+  self.env = ENV["RAILS_ENV"] || ENV["RACK_ENV"] || "development"
+
+  class << self
+    extend Forwardable
+
+    def time_zone=(time_zone)
+      @time_zone = time_zone.is_a?(ActiveSupport::TimeZone) ? time_zone : ActiveSupport::TimeZone[time_zone.to_s]
+    end
+
+    def time_zone
+      @time_zone || Time.zone
+    end
+
+    def config
+      Thread.current[:PgBouncerHero_config] ||= begin
+        path = "config/pgbouncerhero.yml"
+
+        config = YAML.load(ERB.new(File.read(path)).result) if File.exist?(path)
+        config ||= {}
+
+        if config[env]
+          config[env]
+        elsif config["pgbouncers"] # preferred format
+          config
+        else
+          {
+            "pgbouncers" => {
+              "default" => {
+                "primary" => {
+                  "url" => ENV["PGBOUNCERHERO_DATABASE_URL"]
+                }
+              }
+            }
+          }
+        end
+      end
+    end
+
+    def groups
+      @groups ||= begin
+        mapped = config['pgbouncers'].map do |group_id, hash|
+          [group_id.parameterize, PgBouncerHero::Group.new(group_id, config['pgbouncers'])]
+        end
+        Hash[mapped]
+      end
+    end
+  end
+end

data/lib/pgbouncerhero/connection.rb

@@ -0,0 +1,29 @@
+module PgBouncerHero
+  class Connection
+
+    def initialize(host, port, user, password, dbname)
+      @host = host
+      @port = port
+      @user = user
+      @password = password
+      @dbname = dbname
+    end
+
+    def connection
+      begin
+        PG.connect(
+          host: @host,
+          port: @port,
+          user: @user,
+          password: @password,
+          dbname: @dbname,
+          connect_timeout: 5
+        )
+      rescue => e
+        Rails.logger.info(e)
+        nil
+      end
+    end
+
+  end
+end

data/lib/pgbouncerhero/database.rb

@@ -0,0 +1,55 @@
+module PgBouncerHero
+  class Database
+
+    include Methods::Basics
+
+    attr_reader :id, :config, :group
+
+    def initialize(group, id, config)
+      @id = id
+      @config = config || {}
+      @url = URI.parse(config["url"])
+      @group = group
+    end
+
+    def name
+      @name ||= id.to_s
+    end
+
+    def connection
+      connection_model.new(host, port, user, password, dbname).connection
+    end
+
+    def host
+      @url.host if @url
+    end
+
+    def port
+      @url.port if @url
+    end
+
+    def user
+      @url.user if @url
+    end
+
+    def password
+      @url.password if @url
+    end
+
+    def dbname
+      @url.path[1..-1] if @url
+    end
+
+    private
+
+    def connection_model
+      @connection_model ||= begin
+        Class.new(PgBouncerHero::Connection) do
+          def self.name
+            "PgBouncerHero::Connection::Database#{object_id}"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/pgbouncerhero/engine.rb

@@ -0,0 +1,20 @@
+module PgBouncerHero
+  class Engine < ::Rails::Engine
+    isolate_namespace PgBouncerHero
+
+    initializer "pgbouncerhero", group: :all do |app|
+      if defined?(Sprockets) && Sprockets::VERSION >= "4"
+        app.config.assets.precompile << "pgbouncerhero/application.js"
+        app.config.assets.precompile << "pgbouncerhero/application.css"
+        app.config.assets.precompile << "pgbouncerhero/short-paragraph.png"
+      else
+        # use a proc instead of a string
+        app.config.assets.precompile << proc { |path| path == "pgbouncerhero/application.js" }
+        app.config.assets.precompile << proc { |path| path == "pgbouncerhero/application.css" }
+        app.config.assets.precompile << proc { |path| path == "pgbouncerhero/short-paragraph.png" }
+      end
+
+      PgBouncerHero.time_zone = PgBouncerHero.config["time_zone"] if PgBouncerHero.config["time_zone"]
+    end
+  end
+end

data/lib/pgbouncerhero/group.rb

@@ -0,0 +1,21 @@
+require "pgbouncerhero/database"
+
+module PgBouncerHero
+  class Group
+
+    attr_reader :id, :config, :databases
+
+    def initialize(id, config)
+      @id = id
+      @config = config || {}
+      @databases = config[id].map do |k, v|
+        PgBouncerHero::Database.new(self, k, config[id][k])
+      end
+    end
+
+    def name
+      @name ||= id.to_s
+    end
+
+  end
+end

data/lib/pgbouncerhero/methods/basics.rb

@@ -0,0 +1,41 @@
+module PgBouncerHero
+  module Methods
+    module Basics
+      def summary
+        l = lists
+        d = databases
+        l = l.as_json
+        d = d.as_json.reject { |a| a['name'] == 'pgbouncer' }
+        l.push({databases_details: d})
+        l
+      end
+      def databases
+        connection.exec("SHOW databases")
+      end
+      def stats
+        connection.exec("SHOW stats")
+      end
+      def lists
+        connection.exec("SHOW lists")
+      end
+      def pools
+        connection.exec("SHOW pools")
+      end
+      def clients
+        connection.exec("SHOW clients")
+      end
+      def conf
+        connection.exec("SHOW config")
+      end
+      def reload
+        connection.exec("RELOAD")
+      end
+      def suspend
+        connection.exec("SUSPEND")
+      end
+      def shutdown
+        connection.exec("SHUTDOWN")
+      end
+    end
+  end
+end

data/lib/pgbouncerhero/version.rb

@@ -0,0 +1,3 @@
+module PgBouncerHero
+  VERSION = "0.1.0"
+end

data/pgbouncerhero.gemspec

@@ -0,0 +1,32 @@
+# coding: utf-8
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "pgbouncerhero/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "pgbouncerhero"
+  spec.version       = PgBouncerHero::VERSION
+  spec.authors       = ["Quentin Rousseau"]
+  spec.email         = ["contact@quent.in"]
+  spec.summary       = "A graphical user interface for your PGBouncers"
+  spec.description   = "A graphical user interface for your PGBouncers"
+  spec.homepage      = "https://github.com/kwent/pgbouncerhero"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.6"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "minitest"
+  spec.add_runtime_dependency "sass-rails"
+  spec.add_runtime_dependency "semantic-ui-sass"
+  spec.add_runtime_dependency "haml-rails"
+
+  if RUBY_PLATFORM == "java"
+    spec.add_runtime_dependency "pg_jruby"
+  else
+    spec.add_runtime_dependency "pg"
+  end
+end

metadata

@@ -0,0 +1,188 @@
+--- !ruby/object:Gem::Specification
+name: pgbouncerhero
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Quentin Rousseau
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-02-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: sass-rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: semantic-ui-sass
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: haml-rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A graphical user interface for your PGBouncers
+email:
+- contact@quent.in
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- CHANGELOG.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- app/assets/images/pgbouncerhero/short-paragraph.png
+- app/assets/javascripts/pgbouncerhero/application.js
+- app/assets/stylesheets/pgbouncerhero/application.css.scss
+- app/controllers/pg_bouncer_hero/application_controller.rb
+- app/controllers/pg_bouncer_hero/database_controller.rb
+- app/controllers/pg_bouncer_hero/home_controller.rb
+- app/helpers/pg_bouncer_hero/application_helper.rb
+- app/views/layouts/pg_bouncer_hero/application.html.haml
+- app/views/pg_bouncer_hero/database/_actions.html.haml
+- app/views/pg_bouncer_hero/database/_clients.html.haml
+- app/views/pg_bouncer_hero/database/_conf.html.haml
+- app/views/pg_bouncer_hero/database/_databases.html.haml
+- app/views/pg_bouncer_hero/database/_menu.html.haml
+- app/views/pg_bouncer_hero/database/_pools.html.haml
+- app/views/pg_bouncer_hero/database/_stats.html.haml
+- app/views/pg_bouncer_hero/database/clients.html.haml
+- app/views/pg_bouncer_hero/database/conf.html.haml
+- app/views/pg_bouncer_hero/database/databases.html.haml
+- app/views/pg_bouncer_hero/database/pools.html.haml
+- app/views/pg_bouncer_hero/database/stats.html.haml
+- app/views/pg_bouncer_hero/database/summary.js.haml
+- app/views/pg_bouncer_hero/home/_card.html.haml
+- app/views/pg_bouncer_hero/home/_card_content.html.haml
+- app/views/pg_bouncer_hero/home/_card_loading_content.html.haml
+- app/views/pg_bouncer_hero/home/index.html.haml
+- app/views/shared/_alert.html.haml
+- app/views/shared/_flash_messages.html.haml
+- config/routes.rb
+- doc/screenshot-1.png
+- doc/screenshot-2.png
+- doc/screenshot-3.png
+- lib/generators/pgbouncerhero/config_generator.rb
+- lib/generators/pgbouncerhero/templates/config.yml
+- lib/pgbouncerhero.rb
+- lib/pgbouncerhero/connection.rb
+- lib/pgbouncerhero/database.rb
+- lib/pgbouncerhero/engine.rb
+- lib/pgbouncerhero/group.rb
+- lib/pgbouncerhero/methods/basics.rb
+- lib/pgbouncerhero/version.rb
+- pgbouncerhero.gemspec
+homepage: https://github.com/kwent/pgbouncerhero
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: A graphical user interface for your PGBouncers
+test_files: []