func getNamedServices(client *node.ControlClient, nmregex string, tenantID string) ([]service.Service, error) {
services := []service.Service{}
var emptySlice []string
serviceRequest := dao.ServiceRequest{
Tags: emptySlice,
TenantID: tenantID,
NameRegex: nmregex,
}
if err := client.GetServices(serviceRequest, &services); err != nil {
glog.Errorf("Could not get named services: %v", err)
return nil, err
}
return services, nil
}
func getServices(client *node.ControlClient, tenantID string, since time.Duration) ([]service.Service, error) {
services := []service.Service{}
var emptySlice []string
serviceRequest := dao.ServiceRequest{
Tags: emptySlice,
TenantID: tenantID,
UpdatedSince: since,
NameRegex: "",
}
if err := client.GetServices(serviceRequest, &services); err != nil {
glog.Errorf("Could not get services: %v", err)
return nil, err
}
glog.V(2).Infof("Returning %d services", len(services))
return services, nil
}
func restGetTopServices(w *rest.ResponseWriter, r *rest.Request, client *node.ControlClient) {
var allServices []service.Service
topServices := []service.Service{}
var serviceRequest dao.ServiceRequest
err := client.GetServices(serviceRequest, &allServices)
if err != nil {
glog.Errorf("Could not get services: %v", err)
restServerError(w, err)
return
}
for _, service := range allServices {
if len(service.ParentServiceID) == 0 {
topServices = append(topServices, service)
}
}
topServices = append(topServices, isvcs.InternalServicesISVC)
glog.V(2).Infof("Returning %d services as top services", len(topServices))
w.WriteJson(&topServices)
}
// restGetVirtualHosts gets all services, then extracts all vhost information and returns it.
func restGetVirtualHosts(w *rest.ResponseWriter, r *rest.Request, client *node.ControlClient) {
var services []service.Service
var serviceRequest dao.ServiceRequest
err := client.GetServices(serviceRequest, &services)
if err != nil {
glog.Errorf("Unexpected error retrieving virtual hosts: %v", err)
restServerError(w, err)
return
}
serviceTree := make(map[string]service.Service)
for _, service := range services {
serviceTree[service.ID] = service
}
vhosts := make([]virtualHost, 0)
for _, service := range services {
if service.Endpoints == nil {
continue
}
for _, endpoint := range service.Endpoints {
if len(endpoint.VHosts) > 0 {
parent, _ := serviceTree[service.ParentServiceID]
for ; len(parent.ParentServiceID) != 0; parent, _ = serviceTree[parent.ParentServiceID] {
}
for _, vhost := range endpoint.VHosts {
vh := virtualHost{
Name: vhost,
Application: parent.Name,
ServiceName: service.Name,
ServiceEndpoint: endpoint.Application,
}
vhosts = append(vhosts, vh)
}
}
}
}
w.WriteJson(&vhosts)
}
// restAddVirtualHost parses payload, adds the vhost to the service, then updates the service
func restAddVirtualHost(w *rest.ResponseWriter, r *rest.Request, client *node.ControlClient) {
var request virtualHostRequest
err := r.DecodeJsonPayload(&request)
if err != nil {
restBadRequest(w, err)
return
}
var services []service.Service
var serviceRequest dao.ServiceRequest
if err := client.GetServices(serviceRequest, &services); err != nil {
glog.Errorf("Could not get services: %v", err)
restServerError(w, err)
return
}
var service *service.Service
for _, _service := range services {
if _service.ID == request.ServiceID {
service = &_service
break
}
}
if service == nil {
glog.Errorf("Could not find service: %s", services)
restServerError(w, err)
return
}
//checkout other virtual hosts for redundancy
_vhost := strings.ToLower(request.VirtualHostName)
for _, service := range services {
if service.Endpoints == nil {
continue
}
for _, endpoint := range service.Endpoints {
for _, host := range endpoint.VHosts {
if host == _vhost {
glog.Errorf("vhost %s already defined for service: %s", request.VirtualHostName, service.ID)
restServerError(w, err)
return
}
}
}
}
err = service.AddVirtualHost(request.Application, request.VirtualHostName)
if err != nil {
glog.Errorf("Unexpected error adding vhost to service (%s): %v", service.Name, err)
restServerError(w, err)
return
}
var unused int
err = client.UpdateService(*service, &unused)
if err != nil {
glog.Errorf("Unexpected error adding vhost to service (%s): %v", service.Name, err)
restServerError(w, err)
return
}
restSuccess(w)
}
