62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
package docker
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/moby/moby/client"
|
|
);
|
|
|
|
func scaleDown(serviceId string){
|
|
inspectresoult, err := ApiClient.ServiceInspect(context.Background(), serviceId, client.ServiceInspectOptions{});
|
|
|
|
if err != nil{
|
|
panic("Error inspecting service!"+ err.Error());
|
|
}
|
|
|
|
originalScale := inspectresoult.Service.Spec.Mode.Replicated.Replicas;
|
|
updatedSpec := inspectresoult.Service.Spec;
|
|
|
|
newScale := uint64(0);
|
|
updatedSpec.Mode.Replicated.Replicas = &newScale;
|
|
|
|
scale.Store(serviceId, *originalScale);
|
|
|
|
_, err = ApiClient.ServiceUpdate(context.Background(), serviceId, client.ServiceUpdateOptions{
|
|
Spec: updatedSpec,
|
|
Version: inspectresoult.Service.Version,
|
|
});
|
|
|
|
if(err != nil){
|
|
panic("Failed to update service."+ err.Error());
|
|
}
|
|
}
|
|
|
|
func scaleUp(serviceId string){
|
|
inspectresoult, err := ApiClient.ServiceInspect(context.Background(), serviceId, client.ServiceInspectOptions{});
|
|
|
|
if err != nil{
|
|
panic("Error inspecting service!"+ err.Error());
|
|
}
|
|
|
|
originalScale, ok := scale.Load(serviceId);
|
|
if(!ok){
|
|
panic("Its not okay!");
|
|
}
|
|
|
|
originalScaleChecked, ok := originalScale.(uint64);
|
|
if(!ok){
|
|
panic("Its very not okay!")
|
|
}
|
|
updatedSpec := inspectresoult.Service.Spec;
|
|
|
|
updatedSpec.Mode.Replicated.Replicas = &originalScaleChecked;
|
|
|
|
|
|
ApiClient.ServiceUpdate(context.Background(), serviceId, client.ServiceUpdateOptions{
|
|
Spec: updatedSpec,
|
|
Version: inspectresoult.Service.Version,
|
|
|
|
});
|
|
|
|
}
|