Skip to content

Instantly share code, notes, and snippets.

@shaneutt
Forked from dwmkerr/k8s-patch.go
Created April 1, 2021 19:47
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save shaneutt/f7565a3aaedb42d32b2641ed92ddb670 to your computer and use it in GitHub Desktop.
Save shaneutt/f7565a3aaedb42d32b2641ed92ddb670 to your computer and use it in GitHub Desktop.
Example showing how to patch Kubernetes resources in Golang. Companion to the article 'https://dwmkerr.com/patching-kubernetes-resources-in-golang/'.
// Example showing how to patch kubernetes resources.
// This is the companion to my article 'Patching Kubernetes Resources in Golang':
// https://dwmkerr.com/patching-kubernetes-resources-in-golang/
package main
import (
"encoding/json"
"fmt"
types "k8s.io/apimachinery/pkg/types"
"k8s.io/client-go/kubernetes"
_ "k8s.io/client-go/plugin/pkg/client/auth"
"k8s.io/client-go/tools/clientcmd"
)
var (
// Leave blank for the default context in your kube config.
context = ""
// Name of the replication controller to scale, and the desired number of replicas.
replicationControllerName = "my-rc"
replicas = uint32(3)
)
// patchStringValue specifies a patch operation for a string.
type patchStringValue struct {
Op string `json:"op"`
Path string `json:"path"`
Value string `json:"value"`
}
// patchUint32Value specifies a patch operation for a uint32.
type patchUInt32Value struct {
Op string `json:"op"`
Path string `json:"path"`
Value uint32 `json:"value"`
}
func scaleReplicationController(clientSet *kubernetes.Clientset, replicasetName string, scale uint32) error {
payload := []patchUInt32Value{{
Op: "replace",
Path: "/spec/replicas",
Value: scale,
}}
payloadBytes, _ := json.Marshal(payload)
_, err := clientSet.
CoreV1().
ReplicationControllers("default").
Patch(replicasetName, types.JSONPatchType, payloadBytes)
return err
}
func main() {
// Get the local kube config.
fmt.Printf("Connecting to Kubernetes Context %v\n", context)
config, err := clientcmd.NewNonInteractiveDeferredLoadingClientConfig(
clientcmd.NewDefaultClientConfigLoadingRules(),
&clientcmd.ConfigOverrides{CurrentContext: context}).ClientConfig()
if err != nil {
panic(err.Error())
}
// Creates the clientset
clientset, err := kubernetes.NewForConfig(config)
if err != nil {
panic(err.Error())
}
// Scale our replication controller.
fmt.Printf("Scaling replication controller %v to %v\n", replicationControllerName, replicas)
err = scaleReplicationController(clientset, replicationControllerName, replicas)
if err != nil {
panic(err.Error())
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment