Add admin interface
This commit is contained in:
parent
1fa3a447d4
commit
f3ff0e8be8
|
@ -1,27 +1,33 @@
|
||||||
package enterprise
|
package enterprise
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
"log"
|
"log"
|
||||||
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
|
|
||||||
"github.com/influxdata/enterprise-client/v2"
|
"github.com/influxdata/enterprise-client/v2"
|
||||||
|
"github.com/influxdata/enterprise-client/v2/admin"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Config struct {
|
type Config struct {
|
||||||
Hosts []*client.Host
|
AdminPort uint16
|
||||||
|
Hosts []*client.Host
|
||||||
}
|
}
|
||||||
|
|
||||||
type Service struct {
|
type Service struct {
|
||||||
hosts []*client.Host
|
hosts []*client.Host
|
||||||
logger *log.Logger
|
logger *log.Logger
|
||||||
hostname string
|
hostname string
|
||||||
|
adminPort string
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewEnterprise(c Config, hostname string) *Service {
|
func NewEnterprise(c Config, hostname string) *Service {
|
||||||
return &Service{
|
return &Service{
|
||||||
hosts: c.Hosts,
|
hosts: c.Hosts,
|
||||||
hostname: hostname,
|
hostname: hostname,
|
||||||
logger: log.New(os.Stdout, "[enterprise]", log.Ldate|log.Ltime),
|
logger: log.New(os.Stdout, "[enterprise]", log.Ldate|log.Ltime),
|
||||||
|
adminPort: fmt.Sprintf(":%d", c.AdminPort),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -32,6 +38,7 @@ func (s *Service) Open() {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
go s.registerProduct(cl)
|
go s.registerProduct(cl)
|
||||||
|
go s.startAdminInterface()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Service) registerProduct(cl *client.Client) {
|
func (s *Service) registerProduct(cl *client.Client) {
|
||||||
|
@ -45,3 +52,7 @@ func (s *Service) registerProduct(cl *client.Client) {
|
||||||
s.logger.Println("Unable to register Telegraf with Enterprise")
|
s.logger.Println("Unable to register Telegraf with Enterprise")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (s *Service) startAdminInterface() {
|
||||||
|
go http.ListenAndServe(s.adminPort, admin.App("foo", []byte("bar")))
|
||||||
|
}
|
||||||
|
|
|
@ -2,6 +2,7 @@ package enterprise_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httptest"
|
"net/http/httptest"
|
||||||
"testing"
|
"testing"
|
||||||
|
@ -59,3 +60,35 @@ func Test_RegistersWithEnterprise(t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func Test_StartsAdminInterface(t *testing.T) {
|
||||||
|
hostname := "localhost"
|
||||||
|
adminPort := 2300
|
||||||
|
|
||||||
|
success, srv := mockEnterprise(func(c *client.Product, err error) {})
|
||||||
|
defer srv.Close()
|
||||||
|
|
||||||
|
c := enterprise.Config{
|
||||||
|
Hosts: []*client.Host{
|
||||||
|
&client.Host{URL: srv.URL},
|
||||||
|
},
|
||||||
|
AdminPort: 2300,
|
||||||
|
}
|
||||||
|
|
||||||
|
e := enterprise.NewEnterprise(c, hostname)
|
||||||
|
e.Open()
|
||||||
|
|
||||||
|
timeout := time.After(1 * time.Millisecond)
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-success:
|
||||||
|
_, err := http.Get(fmt.Sprintf("http://%s:%d", hostname, adminPort))
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Unable to connect to admin interface: err: %s", err)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
case <-timeout:
|
||||||
|
t.Fatal("Expected to receive call to Enterprise API, but received none")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue