mirrors/sablier
1
0
Fork 0
mirror of https://github.com/sablierapp/sablier.git synced 2025-12-24 06:28:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

Stop after timeout

Browse Source 
Only missing interupt
This commit is contained in:
alexandre.hiltcher@etu.unice.fr
2020-10-21 18:42:09 +02:00
parent 48fbc437fb
commit 861b1f269f
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
main.go
View File

@@ -27,7 +27,7 @@ func handleRequests(w http.ResponseWriter, r *http.Request) {
service := GetOrCreateService(serviceName, serviceTimeout)
err = service.HandleServiceState(cli)
if err != nil {
fmt.Printf("Error: %+v ", err)
fmt.Printf("Error: %+v\n ", err)
fmt.Fprintf(w, "%+v", err)
}
fmt.Printf("Service after query: %+v\n", service)
@@ -129,7 +129,8 @@ func (service *Service) setServiceStateFromDocker(client *client.Client) error {
}
status := UP
if dockerService.Spec.Mode.Replicated.Replicas == create(0) {
fmt.Printf("replicas %d\n", dockerService.Spec.Mode.Replicated.Replicas)
if *dockerService.Spec.Mode.Replicated.Replicas == zeroReplica {
status = DOWN
}
service.status = status
@@ -137,7 +138,7 @@ func (service *Service) setServiceStateFromDocker(client *client.Client) error {
}
func (service *Service) start(client *client.Client) {
fmt.Printf("Starting service %s", service.name)
fmt.Printf("Starting service %s\n", service.name)
service.setServiceReplicas(client, 1)
service.timeout = service.initialTimeout
go service.stopAfterTimeout(client)
@@ -148,22 +149,20 @@ func (service *Service) stopAfterTimeout(client *client.Client) {
time.Sleep(1 * time.Second)
service.timeout--
}
fmt.Printf("Stopping service %s", service.name)
fmt.Printf("Stopping service %s\n", service.name)
service.setServiceReplicas(client, 0)
}
func (service *Service) setServiceReplicas(client *client.Client, replicas uint64) error {
ctx := context.Background()
dockerService, err := service.getDockerService(ctx, client)
swarmCluster, err := client.SwarmInspect(ctx)
if err != nil {
return err
}
dockerService.Spec.Mode.Replicated = &swarm.ReplicatedService{
Replicas: create(replicas),
}
client.ServiceUpdate(ctx, dockerService.ID, swarmCluster.ClusterInfo.Version, dockerService.Spec, types.ServiceUpdateOptions{})
client.ServiceUpdate(ctx, dockerService.ID, dockerService.Meta.Version, dockerService.Spec, types.ServiceUpdateOptions{})
return nil
}