mirror of
https://github.com/paralus/paralus.git
synced 2026-08-24 15:47:19 +00:00
restructure rcloud-base as a single base controller (#37)
* restructure rcloud-base as a single base controller * updated master.rest * moved sentry from internal to pkg as it is used by relay * removing unused rpc and it's dependencies * Fix usermgmt tests * Don't redefine variables in rest file Co-authored-by: Abin Simon <abin.simon@rafay.co>
This commit is contained in:
+16
@@ -0,0 +1,16 @@
|
||||
package main
|
||||
|
||||
//sentry
|
||||
//go:generate go run cmd/generate-enum/main.go BootstrapInfraType $PWD/proto/types/sentry
|
||||
//go:generate go run cmd/generate-enum/main.go BootstrapAgentType $PWD/proto/types/sentry
|
||||
//go:generate go run cmd/generate-enum/main.go BootstrapAgentMode $PWD/proto/types/sentry
|
||||
//go:generate go run cmd/generate-enum/main.go BootstrapAgentState $PWD/proto/types/sentry
|
||||
//go:generate go run cmd/generate-enum/main.go BootstrapAgentTemplateType $PWD/proto/types/sentry
|
||||
//go:generate go run cmd/generate-enum/main.go BootstrapTemplateHostType $PWD/proto/types/sentry
|
||||
//go:generate go run cmd/generate-enum/main.go ClusterTokenState $PWD/proto/types/infrapb/v3
|
||||
//go:generate go run cmd/generate-enum/main.go ClusterTokenType $PWD/proto/types/infrapb/v3
|
||||
//go:generate go run cmd/generate-enum/main.go ClusterNodeState $PWD/proto/types/infrapb/v3
|
||||
//go:generate go run cmd/generate-enum/main.go ClusterConditionType $PWD/proto/types/infrapb/v3
|
||||
//go:generate go run cmd/generate-enum/main.go ClusterNamespaceConditionType $PWD/proto/types/infrapb/v3
|
||||
//go:generate go run cmd/generate-enum/main.go ClusterTaskConditionType $PWD/proto/types/infrapb/v3
|
||||
//go:generate go run cmd/generate-enum/main.go ClusterShareMode $PWD/proto/types/infrapb/v3
|
||||
@@ -0,0 +1,84 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"strings"
|
||||
"text/template"
|
||||
)
|
||||
|
||||
const tmpl = `
|
||||
// Code generated by go generate; DO NOT EDIT.
|
||||
package {{ .PackageName }}
|
||||
|
||||
import (
|
||||
driver "database/sql/driver"
|
||||
bytes "bytes"
|
||||
)
|
||||
|
||||
// Scan converts database string to {{ .EnumName }}
|
||||
func (e *{{ .EnumName }}) Scan(value interface{}) error {
|
||||
s := value.([]byte)
|
||||
*e = {{ .EnumName }}({{ .EnumName }}_value[string(s)])
|
||||
return nil
|
||||
}
|
||||
|
||||
// Value converts {{ .EnumName }} into database string
|
||||
func (e {{ .EnumName }}) Value() (driver.Value, error) {
|
||||
return {{ .EnumName }}_name[int32(e)], nil
|
||||
}
|
||||
|
||||
// MarshalJSON converts {{ .EnumName }} to JSON
|
||||
func (e {{ .EnumName }}) MarshalJSON() ([]byte, error) {
|
||||
buffer := bytes.NewBufferString("\"")
|
||||
buffer.WriteString(e.String())
|
||||
buffer.WriteString("\"")
|
||||
return buffer.Bytes(), nil
|
||||
}
|
||||
|
||||
// UnmarshalJSON converts {{ .EnumName }} from JSON
|
||||
func (e *{{ .EnumName }}) UnmarshalJSON(b []byte) error {
|
||||
if b != nil {
|
||||
*e = {{ .EnumName }}({{ .EnumName }}_value[string(b[1:len(b)-1])])
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// implement proto enum interface
|
||||
func (e {{ .EnumName }}) IsEnum() {
|
||||
}
|
||||
|
||||
`
|
||||
|
||||
type TemplateData struct {
|
||||
PackageName string
|
||||
EnumName string
|
||||
}
|
||||
|
||||
func main() {
|
||||
if len(os.Args) != 3 {
|
||||
fmt.Println("expected generate-enum <enum_type> <gen_file_dir>")
|
||||
fmt.Println("got", os.Args)
|
||||
os.Exit(1)
|
||||
}
|
||||
s := strings.Split(os.Args[2], "/")
|
||||
packageName := s[len(s)-1]
|
||||
|
||||
fName := fmt.Sprintf("%s/%s.enum.go", os.Args[2], strings.ToLower(os.Args[1]))
|
||||
|
||||
f, err := os.Create(fName)
|
||||
if err != nil {
|
||||
fmt.Println("unable to create file ", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
defer f.Close()
|
||||
pkgTemplate := template.Must(template.New("").Parse(tmpl))
|
||||
err = pkgTemplate.Execute(f, TemplateData{
|
||||
PackageName: packageName,
|
||||
EnumName: os.Args[1],
|
||||
})
|
||||
if err != nil {
|
||||
fmt.Println("render template ", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user