Update go.yml

Author: priyagupta108

.github/workflows/go.yml

@@ -26,8 +26,18 @@ jobs:
         with:
           go-version: '1.24'
           cache: false
+
+      - name: Set Go cache environment variables
+        run: |
+            echo "GO_MOD_CACHE=$(go env GOMODCACHE)" >> $GITHUB_ENV
+            echo "GO_BUILD_CACHE=$(go env GOCACHE)" >> $GITHUB_ENV     
       - name: Set IMAGE_OS environment variable
-        run: echo "IMAGE_OS=$ImageOS" >> $GITHUB_ENV 
+        run: |
+            if [ "$RUNNER_OS" = "Linux" ]; then
+              echo "IMAGE_OS=$ImageOS-" >> $GITHUB_ENV
+            else
+              echo "IMAGE_OS=" >> $GITHUB_ENV
+            fi
       - name: Save lowercase arch
         run: echo "ARCH=$(echo ${{ runner.arch }} | tr '[:upper:]' '[:lower:]')" >> $GITHUB_ENV
 
@@ -36,9 +46,8 @@ jobs:
         id: go-cache
         with:
           path: |
-                ~/.cache/go-build
-                ~/go/pkg/mod
-              
+               ${{ env.GO_MOD_CACHE }}
+               ${{ env.GO_BUILD_CACHE }}    
           key: setup-go-${{ runner.os }}-${{ env.ARCH }}-${{ env.IMAGE_OS }}-go-${{ steps.setup-go.outputs.go-version }}-${{ hashFiles('**/go.sum') }}
           restore-keys: |
               setup-go-${{ runner.os }}-${{ env.ARCH }}-${{ env.IMAGE_OS }}-go-${{ steps.setup-go.outputs.go-version }}-