forked from hoanhan101/ultimate-go
-
Notifications
You must be signed in to change notification settings - Fork 0
/
function.go
94 lines (78 loc) · 2.4 KB
/
function.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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
package main
import (
"encoding/json"
"errors"
"fmt"
)
// user is a struct type that declares user information.
type user struct {
ID int
Name string
}
// updateStats provides update stats.
type updateStats struct {
Modified int
Duration float64
Success bool
Message string
}
func main() {
// Retrieve the user profile.
u, err := retrieveUser("Hoanh")
if err != nil {
fmt.Println(err)
return
}
// Display the user profile
// Since the returned u is an address, use * to get the value.
fmt.Printf("%+v\n", *u)
// Update user name. Don't care about the update stats.
// This _ is called blank identifier.
// Since we don't need anything outside the scope of if, we can use the compact syntax.
if _, err := updateUser(u); err != nil {
fmt.Println(err)
return
}
// Display the update was successful.
fmt.Println("Updated user record for ID", u.ID)
}
// retrieveUser retrieves the user document for the specified user.
// It takes a string type name and returns a pointer to a user type value and bool type error.
func retrieveUser(name string) (*user, error) {
// Make a call to get the user in a json response.
r, err := getUser(name)
if err != nil {
return nil, err
}
// Goal: Unmarshal the json document into a value of the user struct type.
// Create a value type user.
var u user
// Share the value down the call stack, which is completely safe so the Unmarshal function can
// read the document and initialize it.
err = json.Unmarshal([]byte(r), &u)
// Share it back up the call stack.
// Because of this line, we know that this create an allocation.
// The value is the previous step is not on the stack but on the heap.
return &u, err
}
// GetUser simulates a web call that returns a json
// document for the specified user.
func getUser(name string) (string, error) {
response := `{"ID":101, "Name":"Hoanh"}`
return response, nil
}
// updateUser updates the specified user document.
func updateUser(u *user) (*updateStats, error) {
// response simulates a JSON response.
response := `{"Modified":1, "Duration":0.005, "Success" : true, "Message": "updated"}`
// Unmarshal the json document into a value of the userStats struct type.
var us updateStats
if err := json.Unmarshal([]byte(response), &us); err != nil {
return nil, err
}
// Check the update status to verify the update is successful.
if us.Success != true {
return nil, errors.New(us.Message)
}
return &us, nil
}