-
Notifications
You must be signed in to change notification settings - Fork 0
/
calc.go
67 lines (57 loc) · 1.12 KB
/
calc.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package main
import (
"bufio"
"fmt"
"os"
"strconv"
"strings"
)
func main() {
fmt.Println("Calculator App")
fmt.Println("Enter a calculation (Example: 1 + 2 or 2 * 5), or type 'exit' to quit.")
scanner := bufio.NewScanner(os.Stdin)
for {
fmt.Print(">> ")
scanner.Scan()
input := scanner.Text()
if input == "exit" {
break
}
result, err := calculate(input)
if err != nil {
fmt.Println("Invalid input. Try again.")
continue
}
fmt.Printf("Result: %d\n", result)
}
}
func calculate(input string) (int, error) {
parts := strings.Fields(input)
if len(parts) != 3 {
return 0, fmt.Errorf("Invalid input")
}
left, err := strconv.Atoi(parts[0])
if err != nil {
return 0, err
}
right, err := strconv.Atoi(parts[2])
if err != nil {
return 0, err
}
operator := parts[1]
switch operator {
case "+":
return left + right, nil
case "-":
return left - right, nil
case "*":
return left * right, nil
case "/":
if right == 0 {
return 0, fmt.Errorf("Division by zero is not allowed")
}
return left / right, nil
default:
return 0, fmt.Errorf("Invalid operator: %s", operator)
}
}