Add restart-only flag

Author: bbharathkumarreddy

README.md

@@ -55,6 +55,7 @@ The following inputs can be used as `step.with` keys:
 | `docker-target-port`    | String  | `80`         | Docker target port of container.                                                                                                             |
 | `docker-opts`           | String  |              | More docker options to create docker service. eg. `--log-driver=awslogs --log-opt awslogs-group=my-logs --log-opt awslogs-region=us-east-1`. |
 | `prune`                 | Boolean | `true`       | On prune true, Will cleanup exitied containers and unused images.                                                                            |
+| `restart-only`          | Boolean | `false`      | On restart-only true, Service will be force restarted with set replicas.                                                                       |
 
 ## Contributing
 

action.yml

@@ -63,6 +63,10 @@ inputs:
     description: "On prune true, Will cleanup exitied containers and unused images."
     default: true
     required: false
+  restart-only:
+    description: "On restart-only true, Service will be force restarted with set replicas."
+    default: false
+    required: false
 
 runs:
   using: "composite"
@@ -95,6 +99,7 @@ runs:
       uses: aws-actions/amazon-ecr-login@v2
 
     - name: EC2 Docker Service Deploy
+      if: ${{ inputs.restart-only != 'true' }}
       uses: appleboy/ssh-action@master
       with:
         host: ${{ inputs.ssh-host }}
@@ -120,6 +125,21 @@ runs:
               ${{ inputs.ecr-image-uri }}
           fi
 
+    - name: EC2 Docker Restart Service
+      if: ${{ inputs.restart-only == 'true' }}
+      uses: appleboy/ssh-action@master
+      with:
+        host: ${{ inputs.ssh-host }}
+        username: ${{ inputs.ssh-username }}
+        key: ${{ inputs.ssh-key }}
+        script: |
+          sudo docker login --username AWS -p $(aws ecr get-login-password --region ${{ inputs.aws-region }}) ${{ steps.login-ecr.outputs.registry }}
+          sudo docker pull ${{ inputs.ecr-image-uri }}
+          sudo -S docker service update \
+              --force \
+              --replicas ${{ inputs.replicas }} \
+              ${{ inputs.docker-service }}
+
     - name: EC2 Prune conatiner and images
       if: ${{ inputs.prune == 'true' }}
       uses: appleboy/ssh-action@master