Go勉強会第5回課題

db/database_sql/interfaces/controllers/user_controller.go

@@ -18,6 +18,27 @@ func NewUserController(db *sql.DB) *UserController {
 	return &UserController{db: db}
 }
 
+// Update is a function for updating a user.
+func (controller *UserController) Update(id string, firstName string, lastName string) (err error) {
+	identifier, err := strconv.Atoi(id)
+	if err != nil {
+		return
+	}
+
+	user := domain.User{
+		ID: identifier,
+		FirstName: firstName,
+		LastName:  lastName,
+	}
+
+	err = database.Update(controller.db, user)
+	if err != nil {
+		return
+	}
+
+	return
+}
+
 // Create is a function for creating a user.
 func (controller *UserController) Create(firstName string, lastName string) (*domain.User, error) {
 	user := domain.User{

db/database_sql/interfaces/database/user_repository.go

@@ -6,6 +6,15 @@ import (
 	"github.com/mf-sakura/golang_study/db/database_sql/domain"
 )
 
+// Update is a function for updating a user
+func Update(db *sql.DB, u domain.User) (err error) {
+	_, err = db.Exec(
+		"update users set first_name = ?, last_name = ? where id = ?", u.FirstName, u.LastName, u.ID,
+	)
+
+	return
+}
+
 // Store is a function for creating a user.
 func Store(db *sql.DB, u domain.User) (int, error) {
 	// `?`はmysqlのplaceholder

db/database_sql/main.go

@@ -64,6 +64,23 @@ func main() {
 		}
 		fmt.Printf("ID: %v, FirstName: %v, LastName: %v\n", user.ID, user.FirstName, user.LastName)
 		return
+	// ユーザー更新
+	case "update":
+		if *id == "" {
+			log.Fatal("You need a user.id.")
+		}
+		if *firstName == "" {
+			log.Fatal("You need a first name.")
+		}
+		if *lastName == "" {
+			log.Fatal("You need a last name.")
+		}
+		err := userController.Update(*id, *firstName, *lastName)
+		if err != nil {
+			log.Fatal(err)
+		}
+		fmt.Printf("ID: %v, FirstName: %v, LastName: %v\n", *id, *firstName, *lastName)
+		return
 	default:
 		log.Fatal("Unrecognized option.")
 	}