@@ -20,7 +20,6 @@ import (
20
20
"io"
21
21
"os"
22
22
"runtime"
23
- "strings"
24
23
25
24
"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
26
25
_ "github.com/GoogleCloudPlatform/kubernetes/pkg/api/v1beta1"
@@ -33,39 +32,51 @@ import (
33
32
)
34
33
35
34
var (
36
- outputDest = flag .StringP ("output" , "o" , "-" , "Output destination; '-' means stdout" )
37
- versions = flag .StringP ("versions" , "v" , "v1beta3" , "Comma separated list of versions for conversion." )
35
+ functionDest = flag .StringP ("funcDest" , "f" , "-" , "Output for conversion functions; '-' means stdout" )
36
+ namesDest = flag .StringP ("nameDest" , "n" , "-" , "Output for function names; '-' means stdout" )
37
+ version = flag .StringP ("version" , "v" , "v1beta3" , "Version for conversion." )
38
38
)
39
39
40
40
func main () {
41
41
runtime .GOMAXPROCS (runtime .NumCPU ())
42
42
flag .Parse ()
43
43
44
- var out io.Writer
45
- if * outputDest == "-" {
46
- out = os .Stdout
44
+ var funcOut io.Writer
45
+ if * functionDest == "-" {
46
+ funcOut = os .Stdout
47
47
} else {
48
- file , err := os .Create (* outputDest )
48
+ file , err := os .Create (* functionDest )
49
49
if err != nil {
50
- glog .Fatalf ("Couldn't open %v: %v" , * outputDest , err )
50
+ glog .Fatalf ("Couldn't open %v: %v" , * functionDest , err )
51
51
}
52
52
defer file .Close ()
53
- out = file
53
+ funcOut = file
54
54
}
55
-
56
- versionsForConversion := strings .Split (* versions , "," )
57
- for _ , version := range versionsForConversion {
58
- generator := conversion .NewGenerator (api .Scheme .Raw ())
59
- // TODO(wojtek-t): Change the overwrites to a flag.
60
- generator .OverwritePackage (version , "" )
61
- generator .OverwritePackage ("api" , "newer" )
62
- for _ , knownType := range api .Scheme .KnownTypes (version ) {
63
- if err := generator .GenerateConversionsForType (version , knownType ); err != nil {
64
- glog .Errorf ("error while generating conversion functions for %v: %v" , knownType , err )
65
- }
55
+ var nameOut io.Writer
56
+ if * namesDest == "-" {
57
+ nameOut = os .Stdout
58
+ } else {
59
+ file , err := os .Create (* namesDest )
60
+ if err != nil {
61
+ glog .Fatalf ("Couldn't open %v: %v" , * functionDest , err )
66
62
}
67
- if err := generator .WriteConversionFunctions (out ); err != nil {
68
- glog .Fatalf ("Error while writing conversion functions: %v" , err )
63
+ defer file .Close ()
64
+ nameOut = file
65
+ }
66
+
67
+ generator := conversion .NewGenerator (api .Scheme .Raw ())
68
+ // TODO(wojtek-t): Change the overwrites to a flag.
69
+ generator .OverwritePackage (* version , "" )
70
+ generator .OverwritePackage ("api" , "newer" )
71
+ for _ , knownType := range api .Scheme .KnownTypes (* version ) {
72
+ if err := generator .GenerateConversionsForType (* version , knownType ); err != nil {
73
+ glog .Errorf ("error while generating conversion functions for %v: %v" , knownType , err )
69
74
}
70
75
}
76
+ if err := generator .WriteConversionFunctions (funcOut ); err != nil {
77
+ glog .Fatalf ("Error while writing conversion functions: %v" , err )
78
+ }
79
+ if err := generator .WriteConversionFunctionNames (nameOut ); err != nil {
80
+ glog .Fatalf ("Error while writing conversion functions: %v" , err )
81
+ }
71
82
}
0 commit comments