package main
import (
"log"
"net/http"
"os"
"strings"
// Import the module
sigsci "github.com/signalsciences/sigsci-module-golang"
)
func main() {
// Get the listener address from the first arg
listenerAddress := ""
if len(os.Args) > 1 {
listenerAddress = os.Args[1]
}
if len(listenerAddress) == 0 {
listenerAddress = "localhost:8000"
}
// Process sigsci-agent rpc-address from the optional second arg
sigsciAgentNetwork := "unix"
sigsciAgentAddress := ""
if len(os.Args) > 2 {
sigsciAgentAddress = os.Args[2]
}
if !strings.Contains(sigsciAgentAddress, "/") {
sigsciAgentNetwork = "tcp"
}
// Existing handler, in this case a simple http.ServeMux,
// but could be any http.Handler in the application
mux := http.NewServeMux()
mux.HandleFunc("/", helloworld)
handler := http.Handler(mux)
if len(sigsciAgentAddress) > 0 {
// Wrap the existing http.Handler with the SigSci module handler
wrapped, err := sigsci.NewModule(
// Existing handler to wrap
mux,
// Any additional module options:
sigsci.Socket(sigsciAgentNetwork, sigsciAgentAddress),
//sigsci.Timeout(100 * time.Millisecond),
//sigsci.AnomalySize(512 * 1024),
//sigsci.AnomalyDuration(1 * time.Second),
//sigsci.MaxContentLength(100000),
// Turn on debug logging for this example (do not use in production)
sigsci.Debug(true),
)
if err != nil {
log.Fatal(err)
}
log.Printf("Signal Sciences agent RPC address: %s:%s", sigsciAgentNetwork, sigsciAgentAddress)
// Use the wrapped sigsci handler
handler = wrapped
}
// Listen and Serve as usual using the wrapped sigsci handler if enabled
s := &http.Server{
Handler: handler,
Addr: listenerAddress,
}
log.Printf("Server URL: http://%s/", s.Addr)
log.Fatal(s.ListenAndServe())
}
// helloworld just displays a banner message for testing
func helloworld(w http.ResponseWriter, r *http.Request) {
status := http.StatusOK
w.WriteHeader(status)
w.Write([]byte(`
Hello World
Hello World!
`))
}