func ListServices(q *skynet.Query) {
results := q.FindServices()
for _, service := range results {
fmt.Println(service)
}
}
func ListRegions(q *skynet.Query) {
results := q.FindRegions()
for _, region := range results {
fmt.Println(region)
}
}
func ListHosts(q *skynet.Query) {
results := q.FindHosts()
for _, host := range results {
fmt.Println(host)
}
}
func ListInstances(q *skynet.Query) {
var regFlag *bool
if *RegisteredFlag == "true" {
b := true
regFlag = &b
} else if *RegisteredFlag == "false" {
b := false
regFlag = &b
}
q.Registered = regFlag
results := q.FindInstances()
for _, instance := range results {
registered := ""
if instance.Registered {
registered = " [Registered]"
}
fmt.Println(instance.Config.ServiceAddr.IPAddress + ":" + strconv.Itoa(instance.Config.ServiceAddr.Port) + " - " + instance.Config.Name + " " + instance.Config.Version + registered)
}
}
func (im *InstanceMonitor) getQueryListMux(q *skynet.Query) (r []skynet.ServiceInfo) {
for _, s := range im.instances {
if q.ServiceMatches(s) {
r = append(r, s)
}
}
return
}
func (ss *SubService) removeFromDoozer() {
q := skynet.Query{
UUID: ss.UUID,
DoozerConn: ss.doozer,
}
instances := q.FindInstances()
for _, instance := range instances {
ss.doozer.Del(instance.GetConfigPath(), ss.doozer.GetCurrentRevision())
}
}
func Unregister(q *skynet.Query) {
instances := filterDaemon(q.FindInstances())
for _, instance := range instances {
cladmin := client.Admin{
Instance: instance,
}
_, err := cladmin.Unregister(skynet.UnregisterRequest{})
if err != nil {
config.Log.Error(err.Error())
}
}
}
func PrintTopology(q *skynet.Query) {
topology := make(map[string]map[string]map[string]map[string][]*skynet.ServiceInfo)
results := q.FindInstances()
// Build topology hash first
for _, instance := range results {
if topology[instance.Config.Region] == nil {
topology[instance.Config.Region] = make(map[string]map[string]map[string][]*skynet.ServiceInfo)
}
if topology[instance.Config.Region][instance.Config.ServiceAddr.IPAddress] == nil {
topology[instance.Config.Region][instance.Config.ServiceAddr.IPAddress] = make(map[string]map[string][]*skynet.ServiceInfo)
}
if topology[instance.Config.Region][instance.Config.ServiceAddr.IPAddress][instance.Config.Name] == nil {
topology[instance.Config.Region][instance.Config.ServiceAddr.IPAddress][instance.Config.Name] = make(map[string][]*skynet.ServiceInfo)
}
if topology[instance.Config.Region][instance.Config.ServiceAddr.IPAddress][instance.Config.Name][instance.Config.Version] == nil {
topology[instance.Config.Region][instance.Config.ServiceAddr.IPAddress][instance.Config.Name][instance.Config.Version] = make([]*skynet.ServiceInfo, 0)
}
topology[instance.Config.Region][instance.Config.ServiceAddr.IPAddress][instance.Config.Name][instance.Config.Version] = append(topology[instance.Config.Region][instance.Config.ServiceAddr.IPAddress][instance.Config.Name][instance.Config.Version], instance)
}
// Now we can print the correct heirarchy
for regionName, region := range topology {
fmt.Println("Region: " + regionName)
for hostName, host := range region {
fmt.Println("\tHost: " + hostName)
for serviceName, service := range host {
fmt.Println("\t\tService: " + serviceName)
for versionName, version := range service {
fmt.Println("\t\t\tVersion: " + versionName)
for _, instance := range version {
registered := ""
if instance.Registered {
registered = " [Registered]"
}
fmt.Println("\t\t\t\t" + instance.Config.ServiceAddr.IPAddress + ":" + strconv.Itoa(instance.Config.ServiceAddr.Port) + registered)
}
}
}
}
}
}
func ListServiceVersions(q *skynet.Query) {
if q.Service == "" {
fmt.Println("Service name is required")
return
}
results := q.FindServiceVersions()
for _, version := range results {
fmt.Println(version)
}
}
func AdminStop(q *skynet.Query) {
instances := q.FindInstances()
for _, instance := range instances {
cladmin := client.Admin{
Instance: instance,
}
_, err := cladmin.Stop(skynet.StopRequest{
WaitForClients: true,
})
if err != nil {
config.Log.Error(err.Error())
}
}
}
func (ss *SubService) sendRPCStop() {
q := skynet.Query{
UUID: ss.UUID,
DoozerConn: ss.doozer,
}
instances := q.FindInstances()
for _, instance := range instances {
cladmin := client.Admin{
Instance: instance,
}
cladmin.Stop(skynet.StopRequest{
WaitForClients: true,
})
}
}
func (s *Service) cleanupDoozerEntriesForAddr(addr *skynet.BindAddr) {
if addr == nil {
return
}
q := skynet.Query{
Host: addr.IPAddress,
Port: strconv.Itoa(addr.Port),
DoozerConn: s.doozer(),
}
instances := q.FindInstances()
for _, i := range instances {
s.Log.Trace("Cleaning up old doozer entry with conflicting addr " +
addr.String() + "(" + i.GetConfigPath() + ")")
s.doozer().Del(i.GetConfigPath(), s.doozer().GetCurrentRevision())
s.doozer().Del(i.GetStatsPath(), s.doozer().GetCurrentRevision())
}
}
func Stop(q *skynet.Query) {
cl := client.NewClient(&config)
for _, instance := range filterDaemon(q.FindInstances()) {
cdaemon := daemon.GetDaemonForService(cl, instance)
in := daemon.StopSubServiceRequest{UUID: instance.Config.UUID}
out, err := cdaemon.StopSubService(in)
if err != nil {
if strings.HasPrefix(err.Error(), "No such service UUID") {
// no daemon on the service's machine, shut it down directly
AdminStop(q)
} else {
fmt.Println(err)
}
} else {
stopTemplate.Execute(os.Stdout, out)
}
}
}
// TODO: this should be smarter about which hosts it deploys to
func Deploy(q *skynet.Query, path string, args ...string) {
cl := client.NewClient(&config)
fmt.Println("deploying " + path + " " + strings.Join(args, ""))
for _, host := range q.FindHosts() {
cdaemon := daemon.GetDaemonForHost(cl, host)
in := daemon.DeployRequest{
ServicePath: path,
Args: shellquote.Join(args...),
}
out, err := cdaemon.Deploy(in)
if err != nil {
fmt.Println(err)
return
}
deployTemplate.Execute(os.Stdout, out)
}
}
func Restart(q *skynet.Query) {
cl := client.NewClient(&config)
for _, instance := range filterDaemon(q.FindInstances()) {
cdaemon := daemon.GetDaemonForService(cl, instance)
in := daemon.RestartSubServiceRequest{UUID: instance.Config.UUID}
out, err := cdaemon.RestartSubService(in)
if err != nil {
if strings.HasPrefix(err.Error(), "No such service UUID") {
// Commented out for now, we need to determine if we want to try to restart an unmanaged instance, and support it
// no daemon on the service's machine, shut it down directly
//AdminStop(q)
} else {
fmt.Println(err)
}
} else {
restartTemplate.Execute(os.Stdout, out)
}
}
}