Separate literal arg parsing cases (int, float)

This change allows the ID scalar implementation to more semantically handle the case for unmarshalling integer IDs.
99designs · Oct 14, 2017 · beff084 · beff084
1 parent ab1dd4b
commit beff084
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/internal/common/literals.go b/internal/common/literals.go
@@ -22,7 +22,14 @@ type BasicLit struct {
 
 func (lit *BasicLit) Value(vars map[string]interface{}) interface{} {
 	switch lit.Type {
-	case scanner.Int, scanner.Float:
+	case scanner.Int:
+		value, err := strconv.ParseInt(lit.Text, 10, 32)
+		if err != nil {
+			panic(err)
+		}
+		return int32(value)
+
+	case scanner.Float:
 		value, err := strconv.ParseFloat(lit.Text, 64)
 		if err != nil {
 			panic(err)