This repository has been archived by the owner on Jul 12, 2023. It is now read-only.
/
controller.go
71 lines (62 loc) · 2.28 KB
/
controller.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
// Copyright 2020 the Exposure Notifications Verification Server authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Package issueapi implements the API handler for taking a code request, assigning
// an OTP, saving it to the database and returning the result.
// This is invoked over AJAX from the Web frontend.
package issueapi
import (
"context"
"net/http"
"time"
"github.com/google/exposure-notifications-server/pkg/keys"
enobs "github.com/google/exposure-notifications-server/pkg/observability"
"github.com/google/exposure-notifications-verification-server/pkg/config"
"github.com/google/exposure-notifications-verification-server/pkg/database"
"github.com/google/exposure-notifications-verification-server/pkg/render"
"github.com/google/exposure-notifications-server/pkg/cache"
"github.com/sethvargo/go-limiter"
"go.opencensus.io/tag"
)
type Controller struct {
config config.IssueAPIConfig
db *database.Database
localCache *cache.Cache
limiter limiter.Store
smsSigner keys.KeyManager
h *render.Renderer
}
// New creates a new IssueAPI controller.
func New(cfg config.IssueAPIConfig, db *database.Database, limiter limiter.Store, smsSigner keys.KeyManager, h *render.Renderer) *Controller {
localCache, _ := cache.New(30 * time.Second)
return &Controller{
config: cfg,
db: db,
localCache: localCache,
limiter: limiter,
smsSigner: smsSigner,
h: h,
}
}
func recordObservability(ctx context.Context, startTime time.Time, result *IssueResult) {
var blame tag.Mutator
switch result.HTTPCode {
case http.StatusOK:
blame = enobs.BlameNone
case http.StatusInternalServerError:
blame = enobs.BlameServer
default:
blame = enobs.BlameClient
}
enobs.RecordLatency(ctx, startTime, mLatencyMs, &blame, &result.obsResult)
}