php-fpm-exporter/cmd/server.go

// Copyright © 2018 Enrico Stahn <enrico.stahn@gmail.com>
// 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.

package cmd

import (
	"github.com/spf13/cobra"
	"github.com/prometheus/client_golang/prometheus"
	"net/http"
	"github.com/prometheus/client_golang/prometheus/promhttp"
	"github.com/hipages/php-fpm_exporter/phpfpm"
)

// Configuration variables
var (
	listeningAddress  string
	metricsEndpoint   string
	scrapeURIs        []string
	customLabelNames  []string
	customLabelValues []string
)

// serverCmd represents the server command
var serverCmd = &cobra.Command{
	Use:   "server",
	Short: "A brief description of your command",
	Long: `A longer description that spans multiple lines and likely contains examples
and usage of using your command. For example:

Cobra is a CLI library for Go that empowers applications.
This application is a tool to generate the needed files
to quickly create a Cobra application.`,
	Run: func(cmd *cobra.Command, args []string) {
		log.Infof("Starting server on %v with path %v", listeningAddress, metricsEndpoint)

		pm := phpfpm.PoolManager{}

		for _, uri := range scrapeURIs {
			pm.Add(uri)
		}

		exporter := phpfpm.NewExporter(pm)
		prometheus.MustRegister(exporter)

		http.Handle(metricsEndpoint, promhttp.Handler())
		http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
			w.Write([]byte(`<html>
			 <head><title>Apache Exporter</title></head>
			 <body>
			 <h1>Apache Exporter</h1>
			 <p><a href='` + metricsEndpoint + `'>Metrics</a></p>
			 </body>
			 </html>`))
		})
		log.Fatal(http.ListenAndServe(listeningAddress, nil))
	},
}

func init() {
	RootCmd.AddCommand(serverCmd)

	// Here you will define your flags and configuration settings.

	// Cobra supports Persistent Flags which will work for this command
	// and all subcommands, e.g.:
	// serverCmd.PersistentFlags().String("foo", "", "A help for foo")

	// Cobra supports local flags which will only run when this command
	// is called directly, e.g.:
    serverCmd.Flags().StringVar(&listeningAddress, "web.listen-address", ":9253", "Address on which to expose metrics and web interface.")
    serverCmd.Flags().StringVar(&metricsEndpoint, "web.telemetry-path", "/metrics", "Path under which to expose metrics.")
	serverCmd.Flags().StringSliceVar(&scrapeURIs, "phpfpm.scrape-uri", []string{"tcp://127.0.0.1:9000/status"},"FastCGI address, e.g. unix:///tmp/php.sock;/status or tcp://127.0.0.1:9000/status")
	serverCmd.Flags().StringSliceVar(&customLabelNames, "phpfpm.label-name", []string{},"Name of the custom label that will be inserted.")
	serverCmd.Flags().StringSliceVar(&customLabelValues, "phpfpm.label-value", []string{},"Value of the custom label that will be inserted.")
}
:sparkles: Initial version 2018-02-17 12:42:58 +00:00			`// Copyright © 2018 Enrico Stahn <enrico.stahn@gmail.com>`
			`// 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.`

			`package cmd`

			`import (`
			`"github.com/spf13/cobra"`
			`"github.com/prometheus/client_golang/prometheus"`
			`"net/http"`
			`"github.com/prometheus/client_golang/prometheus/promhttp"`
Update package path 2018-02-18 09:51:02 +00:00			`"github.com/hipages/php-fpm_exporter/phpfpm"`
:sparkles: Initial version 2018-02-17 12:42:58 +00:00			`)`

			`// Configuration variables`
			`var (`
			`listeningAddress string`
			`metricsEndpoint string`
			`scrapeURIs []string`
			`customLabelNames []string`
			`customLabelValues []string`
			`)`

			`// serverCmd represents the server command`
			`var serverCmd = &cobra.Command{`
			`Use: "server",`
			`Short: "A brief description of your command",`
			Long: `A longer description that spans multiple lines and likely contains examples
			`and usage of using your command. For example:`

			`Cobra is a CLI library for Go that empowers applications.`
			`This application is a tool to generate the needed files`
			to quickly create a Cobra application.`,
			`Run: func(cmd *cobra.Command, args []string) {`
			`log.Infof("Starting server on %v with path %v", listeningAddress, metricsEndpoint)`

			`pm := phpfpm.PoolManager{}`

			`for _, uri := range scrapeURIs {`
			`pm.Add(uri)`
			`}`

			`exporter := phpfpm.NewExporter(pm)`
			`prometheus.MustRegister(exporter)`

			`http.Handle(metricsEndpoint, promhttp.Handler())`
			`http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {`
			w.Write([]byte(`<html>
			`<head><title>Apache Exporter</title></head>`
			`<body>`
			`<h1>Apache Exporter</h1>`
			<p><a href='` + metricsEndpoint + `'>Metrics</a></p>
			`</body>`
			</html>`))
			`})`
			`log.Fatal(http.ListenAndServe(listeningAddress, nil))`
			`},`
			`}`

			`func init() {`
			`RootCmd.AddCommand(serverCmd)`

			`// Here you will define your flags and configuration settings.`

			`// Cobra supports Persistent Flags which will work for this command`
			`// and all subcommands, e.g.:`
			`// serverCmd.PersistentFlags().String("foo", "", "A help for foo")`

			`// Cobra supports local flags which will only run when this command`
			`// is called directly, e.g.:`
			`serverCmd.Flags().StringVar(&listeningAddress, "web.listen-address", ":9253", "Address on which to expose metrics and web interface.")`
			`serverCmd.Flags().StringVar(&metricsEndpoint, "web.telemetry-path", "/metrics", "Path under which to expose metrics.")`
			`serverCmd.Flags().StringSliceVar(&scrapeURIs, "phpfpm.scrape-uri", []string{"tcp://127.0.0.1:9000/status"},"FastCGI address, e.g. unix:///tmp/php.sock;/status or tcp://127.0.0.1:9000/status")`
			`serverCmd.Flags().StringSliceVar(&customLabelNames, "phpfpm.label-name", []string{},"Name of the custom label that will be inserted.")`
			`serverCmd.Flags().StringSliceVar(&customLabelValues, "phpfpm.label-value", []string{},"Value of the custom label that will be inserted.")`
			`}`