DEV Community

Morten Vistisen
Morten Vistisen

Posted on

A practical approach to structuring Golang applications

gopher

2022 update

I wrote this article a year ago and there is much I would change about it today. I still post it because I think there are something to be learned from this, at least to get a end 2 end description of one way to structure Go applications. There are many other options, and today I would suggest that you check out this repository.


Go is an amazing language. It’s simple, easy to reason about and gives you much tooling right out of the box. However, when I started with Go I struggled to figure out how to structure my applications in a way that didn’t use “enterprisey” approaches. This is my take on structuring Golang applications that start out simple, but with the flexibility to grow, and what I would have wanted when I started with Go.

Disclaimer

This article is based on the following people (link to their Twitter):

All of the above-mentioned people have articles and tutorials that are amazing on their own, which you should definitely checkout.

If you just want to see the code for this article’s project, check out the Github repository.

Project & tools

It is always frustrating to learn a new language and having to create the same to-do app over and over again. So we are going to make an application that allows people to create an account, where they can track their weight and calculate their nutritional requirements (inspired by weight increase during lockdown). Feel free to add to this application with features you see fit.

Now, I’m assuming some familiarity with Golang and programming in general, as this is a practical example of how to structure a Go application. We will be using the following:

  • Gin Web Framework
  • Go version 1.15+
  • Postgresql
  • Golang-migrate
  • Your favorite IDE (could be Goland or VS code — I highly suggest Goland, it’s amazing)

Elements of a robust and scalable structure

Spend some time working in software development and you quickly learn how programs sometimes can be brittle and other times rigid. Therefore, I look for these three things in my application structure:

  • Testability
  • Readability
  • Adaptability

Writing programs in a way that also makes it easy to write tests dramatically affect developer productivity. Along with readability, it not only makes it easier for you to maintain and improve old code. You also make it easier for new people to add to the project as they can feel somewhat safe when changing or adding code as you already have test in place. Lastly, having adaptability means your project can adapt to changing needs and complexity.

Starting out as simple as possible have the advantage of you being able to quickly iterate on new ideas getting to the actual problem solving right from the get-go. Most developers don’t need to think about domain-driven development when starting new projects, and it would probably be a waste of time in most cases (unless, of course, you are working in Big Co. but then why would you be reading an article for beginner Go developers).

The tutorial is split into three parts:

  • The application structure
  • User Service implementation
  • TDD implementation of Weight Service

This should hopefully give you an overview of the overall structure, a step-by-step guide on how to add a service and finally how easy it is to write test and develop with TDD.

The application structure

Alright, enough talk, let’s go going. Create a new folder wherever you like to store your projects called weight-tracker and run the following command inside the folder:

go mod init weight-tracker
Enter fullscreen mode Exit fullscreen mode

The application is going to follow a structure that looks something like this:

weight-tracker
- cmd
  - server
    main.go
- pkg
  - api
    user.go
    weight.go
  - app
    server.go
    handlers.go
    routes.go
  - repository
    storage.go
Enter fullscreen mode Exit fullscreen mode

Go ahead and create the above structure in your designated folder.

Most go projects seem to be following a convention of having a cmd and a pkg directory. The cmd will be the entry point into the program and gives you the flexibility to interact with the program in several ways. The pkg directory will contain everything else:

  • routes
  • database interactions
  • services etc.

We will be working with four packages:

  • main
  • api
  • app
  • repository

The main package should be self-explanatory if you’ve ever done any golang programming before. All of our services, i.e. user and weight service, are going into the api package along with a definitions file that will contain all of our structs (we will create this later). In the app package, we will have our server, handlers and routes. Lastly, we have repository that will contain all of our code related to database operations.

Open up main.go and add the following:

package main

import (
    "database/sql"
    "fmt"
    "github.com/gin-contrib/cors"
    "github.com/gin-gonic/gin"
    "os"
    "weight-tracker/pkg/api"
    "weight-tracker/pkg/app"
    "weight-tracker/pkg/repository"
)

func main() {
    if err := run(); err != nil {
        fmt.Fprintf(os.Stderr, "this is the startup error: %s\\n", err)
        os.Exit(1)
    }
}

// func run will be responsible for setting up db connections, routers etc
func run() error {
    // I'm used to working with postgres, but feel free to use any db you like. You just have to change the driver
    // I'm not going to cover how to create a database here but create a database
    // and call it something along the lines of "weight tracker"
    connectionString := "postgres://postgres:postgres@localhost/**NAME-OF-YOUR-DATABASE-HERE**?sslmode=disable"

    // setup database connection
    db, err := setupDatabase(connectionString)

    if err != nil {
        return err
    }

    // create storage dependency
    storage := repository.NewStorage(db)

    // create router dependecy
    router := gin.Default()
    router.Use(cors.Default())

    // create user service
    userService := api.NewUserService(storage)

    // create weight service
    weightService := api.NewWeightService(storage)

    server := app.NewServer(router, userService, weightService)

    // start the server
    err = server.Run()

    if err != nil {
        return err
    }

    return nil
}

func setupDatabase(connString string) (*sql.DB, error) {
    // change "postgres" for whatever supported database you want to use
    db, err := sql.Open("postgres", connString)

    if err != nil {
        return nil, err
    }

    // ping the DB to ensure that it is connected
    err = db.Ping()

    if err != nil {
        return nil, err
    }

    return db, nil
}
Enter fullscreen mode Exit fullscreen mode

You should have quite a few errors showing in your IDE now, we are fixing that in a moment.
But first off, let me explain what is going on here. In our func main() we call a method named run, that returns an error (or nil, if there is no error). If run should return an error our programs exits and gives us an error message. Setting up our main function in this way allows us to test it and thereby following the element of a robust service structure, testability. This way of setting up the main function comes from Mat Ryer, who talks more about it in his blog post.

The next two topics we need to discuss are Clean Architecture and Dependency Injection. You might have heard about Clean Architecture under a different name, as a few people wrote about it around the same time. The majority of my inspiration comes from Robert C. Martin so this is who I will be referencing.

We are not going to be religiously following Clean Architecture but mainly adopt the idea of The Dependency Rule. Take a look at the below picture:

https://cdn-images-1.medium.com/max/1600/1*_5WeMzRt5aCVxXNWLlxAJw.png

Source: Robert C. Martin’s blog — Clean Coder Blog

A quote from Robert Martin’s article helps define this rule:

This rule says that source code dependencies can only point inwards. Nothing in an inner circle can know anything at all about something in an outer circle.

In rough terms, the inner layers should not know about the outer layers. This allows us to practically change the database we use. Say we are changing from PostgreSQL to MySQL or exposing our API through gRPC instead of HTTP. You would most likely never do this but it does speak for the adaptability of this concept.

To achieve this, we are going to use dependency injection (DI for short). Basically, DI is the idea that services should receive their dependencies when they are created. It allows us to decouple the creation of a service’s dependencies from the creation of the service itself. This will be helpful when we get to testing our code. If you want to read more about DI, I suggest this article.

I learn best by looking at actual code, so let’s start by adding some of the missing code that will make the code in main.go make more sense. Open up storage.go and add the following:

package repository

import (
    "database/sql"
    _ "github.com/lib/pq"
)

type Storage interface{}

type storage struct {
    db *sql.DB
}

func NewStorage(db *sql.DB) Storage {
    return &storage{
        db: db,
    }
}
Enter fullscreen mode Exit fullscreen mode

This allows us to create a new storage component where and whenever we want, as long as it receives a valid db argument of type *sql.DB.

As you may have noticed, we have a lowercase and an uppercase version of storage, one is a struct and one is an interface. We will define any methods (like CRUD operations) on the lowercase version of storage and define the methods in the uppercase version. By doing so, we can now easily mock out storage for unit testing. Also, we now get some nice suggestions from our IDE when we add methods to the storage interface when we haven’t implemented the methods.

Now, lets set up the base structure of one of our services, user.go. This should give you a feel for how the API package is going to be structuring services. You will have to repeat this for the weight.go service as well. Just copy-paste the content of user.goand change the naming. Open up user.go and add the following:

package api

// UserService contains the methods of the user service
type UserService interface{}

// UserRepository is what lets our service do db operations without knowing anything about the implementation
type UserRepository interface{}

type userService struct {
    storage UserRepository
}

func NewUserService(userRepo UserRepository) UserService {
    return &userService{
        storage: userRepo,
    }
}
Enter fullscreen mode Exit fullscreen mode

Notice that our user service have a repository dependency. We will define repository methods (i.e. database operations) here later on, but the important part is that we only define the methods we need. Since we are not interested in the implementation of these methods, only the behaviour, we can write mock functions that suit a given test case.

This might seem a bit fluffy right now but bear with me for now, it will make sense later on. Let’s get a running application that we can add actual logic to the services.

Open up server.go and add the following:

package app

import (
    "github.com/gin-gonic/gin"
    "log"
    "weight-tracker/pkg/api"
)

type Server struct {
    router        *gin.Engine
    userService   api.UserService
    weightService api.WeightService
}

func NewServer(router *gin.Engine, userService api.UserService, weightService api.WeightService) *Server {
    return &Server{
        router:        router,
        userService:   userService,
        weightService: weightService,
    }
}

func (s *Server) Run() error {
    // run function that initializes the routes
    r := s.Routes()

    // run the server through the router
    err := r.Run()

    if err != nil {
        log.Printf("Server - there was an error calling Run on router: %v", err)
        return err
    }

    return nil
}
Enter fullscreen mode Exit fullscreen mode

Next, open routes.go and add the following:

package app

import "github.com/gin-gonic/gin"

func (s *Server) Routes() *gin.Engine {
    router := s.router

    // group all routes under /v1/api
    v1 := router.Group("/v1/api")
    {
        v1.GET("/status", s.ApiStatus())
    }

    return router
}
Enter fullscreen mode Exit fullscreen mode

We are going to take advantage of gin’s group functionality so we can easily group our endpoints after the resources they intend to serve. Next up, let’s add a handler so we can actually make calls to the status endpoint and verify that our application is running. Open up handlers.go:

package app

import (
    "github.com/gin-gonic/gin"
    "net/http"
)

func (s *Server) ApiStatus() gin.HandlerFunc {
    return func(c *gin.Context) {
        c.Header("Content-Type", "application/json")

        response := map[string]string{
            "status": "success",
            "data":   "weight tracker API running smoothly",
        }

        c.JSON(http.StatusOK, response)
    }
}
Enter fullscreen mode Exit fullscreen mode

At this point, we only need to sync a few dependencies: Gin Web Framework and a driver for PostgreSQL. Go ahead and type the following into your terminal: go get github.com/gin-contrib/cors , go get github.com/gin-gonic/ginand go get github.com/lib/pg.

Everything should be ready now, so go into your terminal and write: go run cmd/server/main.go, visit http://localhost:8080/v1/api/status, and you should receive a message along the lines of:

{
    "data": "weight tracker API running smoothly",
    "status": "success"
}
Enter fullscreen mode Exit fullscreen mode

If you don’t get the above message, please go back to the previous steps and see if you missed something or checkout the accompanying GitHub repository.

User Service implementation

At this point, we are ready to start building the meat of the application. To do so, we probably need to set up a database with tables and relations. To do this, we are going to use the library [golang-migrate](https://github.com/golang-migrate/migrate/). I personally like this library as it has a CLI tool to add migrations which enables me to do things like adding Makefile commands to create migrations. I encourage you to give the library’s documentation a look, it is an amazing project.

I won’t cover how to set this up as the article would be too long. For now, go into your terminal, make sure you are in the repository folder and run:

git clone https://gist.github.com/ed090d782dc6ebb35e344ff82aafdddf.git
Enter fullscreen mode Exit fullscreen mode

This clones the migrations needed for the project. You should also now have a folder named ed090d782dc6ebb35e344ff82aafdddf, ****lets change that to migrations by running:

mv ed090d782dc6ebb35e344ff82aafdddf migrations
Enter fullscreen mode Exit fullscreen mode

The last thing we need now is to add the RunMigrations method to storage.go:

// update your imports to look like this:
import (
    "database/sql"
    "errors"
    "github.com/golang-migrate/migrate/v4"
    _ "github.com/golang-migrate/migrate/v4/database/postgres"
    _ "github.com/golang-migrate/migrate/v4/source/file"
    _ "github.com/lib/pq"
    "path/filepath"
    "runtime"
)

// add the RunMigrations method to our interface
type Storage interface {
    RunMigrations(connectionString string) error
}

// add this below NewStorage
func (s *storage) RunMigrations(connectionString string) error {
    if connectionString == "" {
        return errors.New("repository: the connString was empty")
    }
    // get base path
    _, b, _, _ := runtime.Caller(0)
    basePath := filepath.Join(filepath.Dir(b), "../..")

    migrationsPath := filepath.Join("file://", basePath, "/pkg/repository/migrations/")

    m, err := migrate.New(migrationsPath, connectionString)

    if err != nil {
        return err
    }

    err = m.Up()

    switch err {
    case errors.New("no change"):
        return nil
    }

    return nil
}
Enter fullscreen mode Exit fullscreen mode

To run the migrations, open up main.go and add the following:

// everything stays the same, so add this below
// storage := repository.NewStorage(db)
// run migrations

// note that we are passing the connectionString again here. This is so
// we can easily run migrations against another database, say a test version,
// for our integration- and end-to-end tests
err = storage.RunMigrations(connectionString)

if err != nil {
    return err
}
Enter fullscreen mode Exit fullscreen mode

You should now have two tables in your database: user and weight. Let’s get started on writing some actual business logic.

We want to let people create an account through our API. So let’s started by defining what a user request is, create a file called under the API folder definitions.go and add the following:

package api

type NewUserRequest struct {
    Name          string `json:"name"`
    Age           int    `json:"age"`
    Height        int    `json:"height"`
    Sex           string `json:"sex"`
    ActivityLevel int    `json:"activity_level"`
    WeightGoal    string `json:"weight_goal"`
    Email         string `json:"email"`
}
Enter fullscreen mode Exit fullscreen mode

We are defining what a new user request should look like. Note that this might be different from what a user struct would look like, we define our struct to only include the data we need. Next up, open user.go and the following to UserService and UserRepository:

// user.go
type UserService interface {
    New(user NewUserRequest) error
}

// User repository is what lets our service do db operations without knowing anything about the implementation
type UserRepository interface {
    CreateUser(NewUserRequest) error
}
Enter fullscreen mode Exit fullscreen mode

Here we define a method on our UserService, called New and a method on the UserRepository called CreateUser. Remember we talked about The Dependency Rule earlier? This is what’s going on with the CreateUser method on UserRepository, our service does not know about the actual implementation of the method, what type of database it is etc. Just that there is a method called CreateUser and takes in a NewUserRequest and returns an error. The benefit of this is twofold: we get some indication from our IDE that a method is missing (open up main.go and check api.NewUserService) and what it needs, and it allows us to easily write unit tests. You should also see an error from NewUserServicein user.go, telling us that we are missing a method. Let’s fixed that, add the following:

// add these imports after the package declaration
import (
    "errors"
    "strings"
)

// add this after NewUserService
func(u * userService) New(user NewUserRequest) error {
    // do some basic validations
    if user.Email == "" {
        return errors.New("user service - email required")
    }

    if user.Name == "" {
        return errors.New("user service - name required")
    }

    if user.WeightGoal == "" {
        return errors.New("user service - weight goal required")
    }

    // do some basic normalisation
    user.Name = strings.ToLower(user.Name)
    user.Email = strings.TrimSpace(user.Email)

    err := u.storage.CreateUser(user)

    if err != nil {
        return err
    }

    return nil
}
Enter fullscreen mode Exit fullscreen mode

We do some basic validations and normalisation, but this method could definitely be improved upon. We still need to add the CreateUser method, so open up storage.go and add the following to the CreateUser method to the Storage interface:

// storage.go
type Storage interface {
    RunMigrations(connectionString string) error
    CreateUser(request api.NewUserRequest) error
}
Enter fullscreen mode Exit fullscreen mode

Notice that this makes the error in main.go go away, but results in a new error on the NewStorage function. We need to implement the method, just like we did with the UserService. Add this below RunMigrations:

// add "log" to imports to your imports look like this:
import (
    "database/sql"
    "errors"
    "github.com/golang-migrate/migrate/v4"
    _ "github.com/golang-migrate/migrate/v4/database/postgres"
    _ "github.com/golang-migrate/migrate/v4/source/file"
    _ "github.com/lib/pq"
    "log"
    "path/filepath"
    "runtime"
    "weight-tracker/pkg/api"
)

// add this below RunMigrations
func (s *storage) CreateUser(request api.NewUserRequest) error {
    newUserStatement := `
        INSERT INTO "user" (name, age, height, sex, activity_level, email, weight_goal)
        VALUES ($1, $2, $3, $4, $5, $6, $7);
        `

    err := s.db.QueryRow(newUserStatement, request.Name, request.Age, request.Height, request.Sex, request.ActivityLevel, request.Email, request.WeightGoal).Err()

    if err != nil {
        log.Printf("this was the error: %v", err.Error())
        return err
    }

    return nil
}
Enter fullscreen mode Exit fullscreen mode

Again, this code could probably do with some improvements, but I’m going to keep it short.

Now, all that is left to do is to expose this through HTTP so that people can actually start using our API and create their accounts. Open handlers.go and add the following:

// update your imports to look like this
import (
    "github.com/gin-gonic/gin"
    "log"
    "net/http"
    "weight-tracker/pkg/api"
)

// add this below APIStatus method
func (s *Server) CreateUser() gin.HandlerFunc {
    return func(c *gin.Context) {
        c.Header("Content-Type", "application/json")

        var newUser api.NewUserRequest

        err := c.ShouldBindJSON(&newUser)

        if err != nil {
            log.Printf("handler error: %v", err)
            c.JSON(http.StatusBadRequest, nil)
            return
        }

        err = s.userService.New(newUser)

        if err != nil {
            log.Printf("service error: %v", err)
            c.JSON(http.StatusInternalServerError, nil)
            return
        }

        response := map[string]string{
            "status": "success",
            "data":   "new user created",
        }

        c.JSON(http.StatusOK, response)
    }
}
Enter fullscreen mode Exit fullscreen mode

We are going to accept a request with a JSON payload and using gin’s ShouldBindJSON method to extract the data. Go ahead and try it out!

It took some time to get here, so let me show you one of the benefits that I keep talking about: testability.

Create a file called user_test.go under the API folder and add the following:

package api_test

import (
    "errors"
    "reflect"
    "testing"
    "weight-tracker/pkg/api"
)

type mockUserRepo struct{}

func (m mockUserRepo) CreateUser(request api.NewUserRequest) error {
    if request.Name == "test user already created" {
        return errors.New("repository - user already exists in database")
    }

    return nil
}

func TestCreateNewUser(t *testing.T) {
    mockRepo := mockUserRepo{}
    mockUserService := api.NewUserService(&mockRepo)

    tests := []struct {
        name    string
        request api.NewUserRequest
        want    error
    }{
        {
            name: "should create a new user successfully",
            request: api.NewUserRequest{
                Name:          "test user",
                WeightGoal:    "maintain",
                Age:           20,
                Height:        180,
                Sex:           "female",
                ActivityLevel: 5,
                Email:         "test_user@gmail.com",
            },
            want: nil,
        }, {
            name: "should return an error because of missing email",
            request: api.NewUserRequest{
                Name:          "test user",
                Age:           20,
                WeightGoal:    "maintain",
                Height:        180,
                Sex:           "female",
                ActivityLevel: 5,
                Email:         "",
            },
            want: errors.New("user service - email required"),
        }, {
            name: "should return an error because of missing name",
            request: api.NewUserRequest{
                Name:          "",
                Age:           20,
                WeightGoal:    "maintain",
                Height:        180,
                Sex:           "female",
                ActivityLevel: 5,
                Email:         "test_user@gmail.com",
            },
            want: errors.New("user service - name required"),
        }, {
            name: "should return error from database because user already exists",
            request: api.NewUserRequest{
                Name:          "test user already created",
                Age:           20,
                Height:        180,
                WeightGoal:    "maintain",
                Sex:           "female",
                ActivityLevel: 5,
                Email:         "test_user@gmail.com",
            },
            want: errors.New("repository - user already exists in database"),
        },
    }

    for _, test := range tests {
        t.Run(test.name, func(t *testing.T) {
            err := mockUserService.New(test.request)

            if !reflect.DeepEqual(err, test.want) {
                t.Errorf("test: %v failed. got: %v, wanted: %v", test.name, err, test.want)
            }
        })
    }
}
Enter fullscreen mode Exit fullscreen mode

A lot of things are happening here so let us go through them step-by-step.

We start by creating a struct called mockUserRepo. Our UserServiceknows that it needs a UserRepository with the following method: CreateUser. However, it does care about the actual implementation of said method, which allows us to mock the behaviour any way we want. In this case, we say that when the Name of the request is equal to “test user already created” (a bad name I know, but hopefully you get the point), return an error, and if not, just return nil. This allows us to mock the behaviour of our database to make it fit different situations, and test if our logic handles it in a way we expect it too.

Next, we create a new variable called mockRepo of type mockUserRepo We then create a mockUserService and pass the mockRepo to it, and we are ready to go! The actual test is what is known as a table-driven test. I won’t go into details about it as it is beyond the scope of this article, but if you want to know more, check out Dave Chaney’s article about it here.

Now, whenever we add a method to the UserRepository we would also have to add it here, in our mockUserRepo. We would of course also like to have some integration test, but what I really wanted to show with this article was how to do unit tests as these are cheap and easy to write.

Run go test ./... from the root directory and all tests should pass.

TDD implementation of Weight Service

Our application is not worth much right now. We can only create a user but not track our weight or calculate the amount of calories we need. Let’s change that!

Our application is not worth much right now. We can only create a user but not track our weight or calculate the number of calories we need. Let’s change that!I’m a big fan of test-driven development (TDD) and the way this application is structured makes it really easy to use. We are going to need three methods on our weight service: New, CalculateBMR and DailyIntake and two on our repository: CreateWeightEntry and GetUser. Open weight.go and add the following to WeightService and WeightRepository:

// weight.go

// WeightService contains the methods of the user service
type WeightService interface {
    New(request NewWeightRequest) error
    CalculateBMR(height, age, weight int, sex string) (int, error)
    DailyIntake(BMR, activityLevel int, weightGoal string) (int, error)
}

type WeightRepository interface {
    CreateWeightEntry(w Weight) error
    GetUser(userID int) (User, error)
}
Enter fullscreen mode Exit fullscreen mode

Next, we need to add three structs to our definitions file: ´NewWeightRequest, Weight and User. Open up weight.go and add the following:

// add this after the package declaration
import "time"

// add this below NewUserRequest
type User struct {
    ID            int       `json:"id"`
    CreatedAt     time.Time `json:"created_at"`
    UpdatedAt     time.Time `json:"updated_at"`
    Name          string    `json:"name"`
    Age           int       `json:"age"`
    Height        int       `json:"height"`
    Sex           string    `json:"sex"`
    ActivityLevel int       `json:"activity_level"`
    WeightGoal    string    `json:"weight_goal"`
    Email         string    `json:"email"`
}

type Weight struct {
    Weight             int `json:"weight"`
    UserID             int `json:"user_id"`
    BMR                int `json:"bmr"`
    DailyCaloricIntake int `json:"daily_caloric_intake"`
}

type NewWeightRequest struct {
    Weight int `json:"weight"`
    UserID int `json:"user_id"`
}
Enter fullscreen mode Exit fullscreen mode

Now, you will see an error on NewWeightService about missing methods. We don’t want to write the actual implementation yet, as we are doing TDD, so for now, just add this below NewWeightService:

// add these below NewWeightService
func (w *weightService) CalculateBMR(height, age, weight int, sex string) (int, error) {
    panic("implement me")
}

func (w *weightService) DailyIntake(BMR, activityLevel int, weightGoal string) (int, error) {
    panic("implement me")
}

func (w *weightService) New(request NewWeightRequest) error {
    panic("implement me")
}
Enter fullscreen mode Exit fullscreen mode

Open up main.go and you will see that we also have some missing methods on storage passed to api.NewWeightService. Open up storage.go and add these:

// add this to the Storage interface
type Storage interface {
    CreateWeightEntry(request api.Weight) error
    GetUser(userID int) (api.User, error)
}

// add this below the CreateUser method
func (s *storage) CreateWeightEntry(request api.Weight) error {
    panic("implement me!")
}

func (s *storage) GetUser(userID int) (api.User, error) {
    panic("implement me!")
}
Enter fullscreen mode Exit fullscreen mode

Lets now add the tests that we will need, create a file calledweight_test.go in the API folder and add the following:

package api_test

import (
    "errors"
    "reflect"
    "testing"
    "weight-tracker/pkg/api"
)

type mockWeightRepo struct{}

func (m mockWeightRepo) CreateWeightEntry(w api.Weight) error {
    return nil
}

func (m mockWeightRepo) GetUser(userID int) (api.User, error) {
    if userID != 1 {
        return api.User{}, errors.New("storage - user doesn't exists")
    }

    return api.User{
        ID:            userID,
        Name:          "Test user",
        Age:           20,
        Height:        185,
        WeightGoal:    "maintain",
        Sex:           "female",
        ActivityLevel: 5,
        Email:         "test@mail.com",
    }, nil
}

func TestCreateWeightEntry(t *testing.T) {
    mockRepo := mockWeightRepo{}
    mockUserService := api.NewWeightService(&mockRepo)

    tests := []struct {
        name    string
        request api.NewWeightRequest
        want    error
    }{
        {
            name: "should create a new user successfully",
            request: api.NewWeightRequest{
                Weight: 70,
                UserID: 1,
            },
            want: nil,
        }, {
            name: "should return a error because user already exists",
            request: api.NewWeightRequest{
                Weight: 70,
                UserID: 2,
            },
            want: errors.New("storage - user doesn't exists"),
        },
    }

    for _, test := range tests {
        t.Run(test.name, func(t *testing.T) {
            err := mockUserService.New(test.request)
            if !reflect.DeepEqual(err, test.want) {
                t.Errorf("test: %v failed. got: %v, wanted: %v", test.name, err, test.want)
            }
        })
    }
}

func TestCalculateBMR(t *testing.T) {
    mockRepo := mockWeightRepo{}
    mockUserService := api.NewWeightService(&mockRepo)

    tests := []struct {
        name   string
        Height int
        Age    int
        Weight int
        Sex    string
        want   int
        err    error
    }{
        {
            name:   "should calculate BMR for a female",
            Height: 170,
            Age:    22,
            Weight: 65,
            Sex:    "female",
            want:   1441,
            err:    nil,
        }, {
            name:   "should calculate BMR for a male",
            Height: 170,
            Age:    22,
            Weight: 65,
            Sex:    "male",
            want:   1607,
            err:    nil,
        }, {
            name:   "should return error because sex wasn't properly specified",
            Height: 170,
            Age:    22,
            Weight: 65,
            Sex:    "",
            want:   0,
            err:    errors.New("invalid variable sex provided to CalculateBMR. needs to be either male or female"),
        },
    }

    for _, test := range tests {
        t.Run(test.name, func(t *testing.T) {
            BMR, err := mockUserService.CalculateBMR(test.Height, test.Age, test.Weight, test.Sex)
            if !reflect.DeepEqual(err, test.err) {
                t.Errorf("test: %v failed. got: %v, wanted: %v", test.name, err, test.want)
            }

            if !reflect.DeepEqual(BMR, test.want) {
                t.Errorf("test: %v failed. got: %v, wanted: %v", test.name, BMR, test.want)
            }
        })
    }
}

func TestDailyIntake(t *testing.T) {
    mockRepo := mockWeightRepo{}
    mockUserService := api.NewWeightService(&mockRepo)

    tests := []struct {
        name          string
        BMR           int
        ActivityLevel int
        weightGoal    string
        want          int
        err           error
    }{
        {
            name:          "should calculate daily intake for activity level 1 with weight loss as goal",
            weightGoal:    "loose",
            BMR:           1441,
            ActivityLevel: 1,
            want:          1229,
            err:           nil,
        }, {
            name:          "should calculate daily intake for activity level 2 with weight loss as goal",
            weightGoal:    "loose",
            BMR:           1441,
            ActivityLevel: 2,
            want:          1481,
            err:           nil,
        }, {
            name:          "should calculate daily intake for activity level 3 with weight loss as goal",
            weightGoal:    "loose",
            BMR:           1441,
            ActivityLevel: 3,
            want:          1733,
            err:           nil,
        }, {
            name:          "should calculate daily intake for activity level 4 with weight increase as goal",
            weightGoal:    "gain",
            BMR:           1441,
            ActivityLevel: 4,
            want:          2985,
            err:           nil,
        }, {
            name:          "should calculate daily intake for activity level 5 with weight maintenance as goal",
            weightGoal:    "maintain",
            BMR:           1441,
            ActivityLevel: 5,
            want:          2737,
            err:           nil,
        },
    }

    for _, test := range tests {
        t.Run(test.name, func(t *testing.T) {
            BMR, err := mockUserService.DailyIntake(test.BMR, test.ActivityLevel, test.weightGoal)
            if !reflect.DeepEqual(err, test.err) {
                t.Errorf("test: %v failed. got: %v, wanted: %v", test.name, err, test.want)
            }

            if !reflect.DeepEqual(BMR, test.want) {
                t.Errorf("test: %v failed. got: %v, wanted: %v", test.name, BMR, test.want)
            }
        })
    }
}
Enter fullscreen mode Exit fullscreen mode

The actual content of the tests is not so important as the fact that we can quickly write tests and mock external dependencies such as the methods to interact with the database.

Run go test ./... from the root directory and you should receive a lot of failed tests. We are going to fix this by implementing the logic of these methods, starting with the three in our WeightService. Add this below NewWeightService:

// add this after the package declaration
import "errors"

// we define some activity multipliers here - these are just part of the
// formula I choose to calculate your daily caloric intake.
const (
    // Very Low Intensity activity multiplier - 1
    veryLowActivity = 1.2
    // Light exercise activity multiplier - 3-4x per week - 2
    lightActivity = 1.375
    // Moderate exercise activity multiplier - 3-5x per week 30-60 mins/session - 3
    moderateActivity = 1.55
    // High exercise activity multiplier - (6-7x per week for 45-60 mins) - 4
    highActivity = 1.725
    // Very high exercise activity multiplier - for people who is an Athlete - 5
    veryHighActivity = 1.9
)

func (w *weightService) New(request NewWeightRequest) error {
    if request.UserID == 0 {
        return errors.New("weight service - user ID cannot be 0")
    }

    user, err := w.storage.GetUser(request.UserID)

    if err != nil {
        return err
    }

    bmr, err := w.CalculateBMR(user.Height, user.Age, request.Weight, user.Sex)

    if err != nil {
        return err
    }

    dailyIntake, err := w.DailyIntake(bmr, user.ActivityLevel, user.WeightGoal)

    if err != nil {
        return err
    }

    newWeight := Weight{
        Weight:             request.Weight,
        UserID:             user.ID,
        BMR:                bmr,
        DailyCaloricIntake: dailyIntake,
    }

    err = w.storage.CreateWeightEntry(newWeight)

    if err != nil {
        return err
    }

    return nil
}

func (w *weightService) CalculateBMR(height, age, weight int, sex string) (int, error) {
    var sexModifier int

    switch sex {
    case "male":
        sexModifier = -5
    case "female":
        sexModifier = 161
    default:
        return 0, errors.New("invalid variable sex provided to CalculateBMR. needs to be either male or female")
    }

    return (10 * weight) + int(float64(height)*6.25) - (5 * age) - sexModifier, nil
}

func (w *weightService) DailyIntake(BMR, activityLevel int, weightGoal string) (int, error) {
    var maintenanceCalories int

    switch activityLevel {
    case 1:
        maintenanceCalories = int(float64(BMR) * veryLowActivity)
    case 2:
        maintenanceCalories = int(float64(BMR) * lightActivity)
    case 3:
        maintenanceCalories = int(float64(BMR) * moderateActivity)
    case 4:
        maintenanceCalories = int(float64(BMR) * highActivity)
    case 5:
        maintenanceCalories = int(float64(BMR) * veryHighActivity)
    default:
        return 0, errors.New("invalid variable activityLevel - needs to be 1, 2, 3, 4 or 5")
    }

    var dailyCaloricIntake int

    switch weightGoal {
    case "gain":
        dailyCaloricIntake = maintenanceCalories + 500
    case "loose":
        dailyCaloricIntake = maintenanceCalories - 500
    case "maintain":
        dailyCaloricIntake = maintenanceCalories
    default:
        return 0, errors.New("invalid weight goal provided - must be gain, loose or maintain")
    }

    return dailyCaloricIntake, nil
}
Enter fullscreen mode Exit fullscreen mode

With these methods added all of the tests should pass.

The last thing we need is to add the methods needed to interact with the database. Open storage.go and add the following:

// replace the methods: CreateWeightEntry and GetUser with the code below
func (s *storage) CreateWeightEntry(request api.Weight) error {
    newWeightStatement := `
        INSERT INTO weight (weight, user_id, bmr, daily_caloric_intake)
        VALUES ($1, $2, $3, $4)
        RETURNING id;
        `

    var ID int
    err := s.db.QueryRow(newWeightStatement, request.Weight, request.UserID, request.BMR, request.DailyCaloricIntake).Scan(&ID)

    if err != nil {
        log.Printf("this was the error: %v", err.Error())
        return err
    }

    return nil
}

func (s *storage) GetUser(userID int) (api.User, error) {
    getUserStatement := `
        SELECT id, name, age, height, sex, activity_level, email, weight_goal FROM "user"
        where id=$1;
        `

    var user api.User
    err := s.db.QueryRow(getUserStatement, userID).Scan(&user.ID, &user.Name, &user.Age, &user.Height, &user.Sex, &user.ActivityLevel, &user.Email, &user.WeightGoal)

    if err != nil {
        log.Printf("this was the error: %v", err.Error())
        return api.User{}, err
    }

    return user, nil
}
Enter fullscreen mode Exit fullscreen mode

We didn’t do this in true TDD style, however, it should again paint a picture of how we can structure Go applications and develop them using TDD. Feel free to re-implement this section and do it in true TDD style: create one test, make it pass, create the next one, and so on.

We have almost everything we need now. The last thing we need is to add routes and a handler to create a weight entry for a given user. I’m going to leave this up to the reader to implement, as the building blocks should be in place. If you are feeling lazy you can just check out the accompanying Github repository.

Conclusion

Hopefully, this tutorial gave you some insights on how to structure your Golang applications. It was a long one I know, but hopefully, you didn’t waste your time. In a part two, I will show how to add Makefile commands, integration tests and easily deploy the application. If you have any questions or criticisms, please do not hesitate to reach out.

My twitter and github.

Resources

Top comments (0)