Add custom CRD clientset (#97)

* Add custom CRD clientset

* nit

* nit

* clean up .DS_Store files

---------

Co-authored-by: varungupta <[email protected]>

.gitignore

@@ -23,3 +23,4 @@ go.work
 go.work.sum
 .idea
 
+.DS_Store

api/autoscaling/v1alpha1/groupversion_info.go

@@ -28,9 +28,16 @@ var (
 	// GroupVersion is group version used to register these objects
 	GroupVersion = schema.GroupVersion{Group: "autoscaling.aibrix.ai", Version: "v1alpha1"}
 
+	// SchemeGroupVersion is group version used to register these objects.
+	SchemeGroupVersion = GroupVersion
+
 	// SchemeBuilder is used to add go types to the GroupVersionKind scheme
 	SchemeBuilder = &scheme.Builder{GroupVersion: GroupVersion}
 
 	// AddToScheme adds the types in this group-version to the given scheme.
 	AddToScheme = SchemeBuilder.AddToScheme
 )
+
+func Resource(resource string) schema.GroupResource {
+	return SchemeGroupVersion.WithResource(resource).GroupResource()
+}

api/autoscaling/v1alpha1/podautoscaler_types.go

@@ -26,8 +26,10 @@ import (
 // INSERT ADDITIONAL SPEC FIELDS - desired state of cluster
 // Important: Run "make" to regenerate code after modifying this file
 
-//+kubebuilder:object:root=true
-//+kubebuilder:subresource:status
+// +genclient
+// +kubebuilder:object:root=true
+// +kubebuilder:subresource:status
+// +k8s:deepcopy-gen:interfaces=k8s.io/apimachinery/pkg/runtime.Object
 
 // PodAutoscaler is the Schema for the podautoscalers API, a resource to scale Kubernetes pods based on observed metrics.
 // The fields in the spec determine how the scaling behavior should be applied.
@@ -118,7 +120,7 @@ type PodAutoscalerStatus struct {
 	Conditions []metav1.Condition `json:"conditions,omitempty"`
 }
 
-//+kubebuilder:object:root=true
+// +kubebuilder:object:root=true
 
 // PodAutoscalerList contains a list of PodAutoscaler
 type PodAutoscalerList struct {

api/model/v1alpha1/groupversion_info.go

@@ -28,9 +28,16 @@ var (
 	// GroupVersion is group version used to register these objects
 	GroupVersion = schema.GroupVersion{Group: "model.aibrix.ai", Version: "v1alpha1"}
 
+	// SchemeGroupVersion is group version used to register these objects.
+	SchemeGroupVersion = GroupVersion
+
 	// SchemeBuilder is used to add go types to the GroupVersionKind scheme
 	SchemeBuilder = &scheme.Builder{GroupVersion: GroupVersion}
 
 	// AddToScheme adds the types in this group-version to the given scheme.
 	AddToScheme = SchemeBuilder.AddToScheme
 )
+
+func Resource(resource string) schema.GroupResource {
+	return SchemeGroupVersion.WithResource(resource).GroupResource()
+}

api/model/v1alpha1/modeladapter_types.go

@@ -85,8 +85,10 @@ const (
 	ModelAdapterConditionCleanup             ModelAdapterConditionType = "Cleanup"
 )
 
-//+kubebuilder:object:root=true
-//+kubebuilder:subresource:status
+// +genclient
+// +kubebuilder:object:root=true
+// +kubebuilder:subresource:status
+// +k8s:deepcopy-gen:interfaces=k8s.io/apimachinery/pkg/runtime.Object
 
 // ModelAdapter is the Schema for the modeladapters API
 type ModelAdapter struct {
@@ -97,7 +99,7 @@ type ModelAdapter struct {
 	Status ModelAdapterStatus `json:"status,omitempty"`
 }
 
-//+kubebuilder:object:root=true
+// +kubebuilder:object:root=true
 
 // ModelAdapterList contains a list of ModelAdapter
 type ModelAdapterList struct {

go.mod

@@ -16,6 +16,7 @@ require (
 	k8s.io/utils v0.0.0-20230726121419-3b25d923346b
 	sigs.k8s.io/controller-runtime v0.17.3
 	sigs.k8s.io/gateway-api v1.0.0
+	sigs.k8s.io/structured-merge-diff/v4 v4.4.1
 )
 
 require (
@@ -73,7 +74,6 @@ require (
 	k8s.io/component-base v0.29.2 // indirect
 	k8s.io/gengo v0.0.0-20230829151522-9cce18d56c01 // indirect
 	sigs.k8s.io/json v0.0.0-20221116044647-bc3834ca7abd // indirect
-	sigs.k8s.io/structured-merge-diff/v4 v4.4.1 // indirect
 	sigs.k8s.io/yaml v1.4.0 // indirect
 )
 

hack/update-codegen.sh

@@ -0,0 +1,53 @@
+#!/bin/bash
+
+# This shell is used to auto generate some useful tools for k8s, such as clientset, lister, informer and so on.
+# We don't use this tool to generate deepcopy because kubebuilder (controller-tools) has coverred that part.
+
+set -o errexit
+set -o nounset
+set -o pipefail
+
+SCRIPT_ROOT=$(dirname "${BASH_SOURCE[0]}")/..
+ROOT_PKG=github.com/aibrix/aibrix
+
+# Grab code-generator version from go.sum
+CODEGEN_VERSION=$(grep 'k8s.io/code-generator' go.sum | awk '{print $2}' | sed 's/\/go.mod//g' | head -1)
+CODEGEN_PKG=$(echo `go env GOPATH`"/pkg/mod/k8s.io/code-generator@${CODEGEN_VERSION}")
+
+if [[ ! -d ${CODEGEN_PKG} ]]; then
+    echo "${CODEGEN_PKG} is missing. Running 'go mod download'."
+    go mod download
+fi
+
+echo ">> Using ${CODEGEN_PKG}"
+
+# code-generator does work with go.mod but makes assumptions about
+# the project living in `$GOPATH/src`. To work around this and support
+# any location; create a temporary directory, use this as an output
+# base, and copy everything back once generated.
+TEMP_DIR=$(mktemp -d)
+cleanup() {
+    echo ">> Removing ${TEMP_DIR}"
+#    rm -rf ${TEMP_DIR}
+}
+trap "cleanup" EXIT SIGINT
+
+echo ">> Temporary output directory ${TEMP_DIR}"
+
+# Ensure we can execute.
+chmod +x ${CODEGEN_PKG}/generate-groups.sh
+
+# generate the code with:
+# --output-base    because this script should also be able to run inside the vendor dir of
+#                  k8s.io/kubernetes. The output-base is needed for the generators to output into the vendor dir
+#                  instead of the $GOPATH directly. For normal projects this can be dropped.
+#
+cd ${SCRIPT_ROOT}
+${CODEGEN_PKG}/generate-groups.sh all \
+ github.com/aibrix/aibrix/pkg/client github.com/aibrix/aibrix/api \
+ "model:v1alpha1 autoscaling:v1alpha1" \
+ --output-base "${TEMP_DIR}" \
+ --go-header-file hack/boilerplate.go.txt
+
+# Copy everything back.
+cp -a "${TEMP_DIR}/${ROOT_PKG}/." "${SCRIPT_ROOT}/"

hack/verify-codegen.sh

@@ -0,0 +1,53 @@
+#!/bin/bash
+
+# This shell is used to auto generate some useful tools for k8s, such as clientset, lister, informer and so on.
+# We don't use this tool to generate deepcopy because kubebuilder (controller-tools) has coverred that part.
+
+set -o errexit
+set -o nounset
+set -o pipefail
+
+SCRIPT_ROOT=$(dirname "${BASH_SOURCE[0]}")/..
+ROOT_PKG=github.com/aibrix/aibrix
+
+# Grab code-generator version from go.sum
+CODEGEN_VERSION=$(grep 'k8s.io/code-generator' go.sum | awk '{print $2}' | sed 's/\/go.mod//g' | head -1)
+CODEGEN_PKG=$(echo `go env GOPATH`"/pkg/mod/k8s.io/code-generator@${CODEGEN_VERSION}")
+
+if [[ ! -d ${CODEGEN_PKG} ]]; then
+    echo "${CODEGEN_PKG} is missing. Running 'go mod download'."
+    go mod download
+fi
+
+echo ">> Using ${CODEGEN_PKG}"
+
+# code-generator does work with go.mod but makes assumptions about
+# the project living in `$GOPATH/src`. To work around this and support
+# any location; create a temporary directory, use this as an output
+# base, and copy everything back once generated.
+TEMP_DIR=$(mktemp -d)
+cleanup() {
+    echo ">> Removing ${TEMP_DIR}"
+#    rm -rf ${TEMP_DIR}
+}
+trap "cleanup" EXIT SIGINT
+
+echo ">> Temporary output directory ${TEMP_DIR}"
+
+# Ensure we can execute.
+chmod +x ${CODEGEN_PKG}/generate-groups.sh
+
+# generate the code with:
+# --output-base    because this script should also be able to run inside the vendor dir of
+#                  k8s.io/kubernetes. The output-base is needed for the generators to output into the vendor dir
+#                  instead of the $GOPATH directly. For normal projects this can be dropped.
+#
+cd ${SCRIPT_ROOT}
+${CODEGEN_PKG}/generate-groups.sh all \
+ github.com/aibrix/aibrix/pkg/client github.com/aibrix/aibrix/api \
+ "model:v1alpha1 autoscaling:v1alpha1" \
+ --output-base "${TEMP_DIR}" \
+ --go-header-file hack/boilerplate.go.txt
+
+# Copy everything back.
+cp -a "${TEMP_DIR}/${ROOT_PKG}/." "${SCRIPT_ROOT}/"