/
service.go
86 lines (74 loc) · 2.99 KB
/
service.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
// Copyright 2015 The LUCI 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 coordinatorTest
import (
"context"
"github.com/TriggerMail/luci-go/logdog/api/config/svcconfig"
"github.com/TriggerMail/luci-go/logdog/appengine/coordinator"
"github.com/TriggerMail/luci-go/logdog/appengine/coordinator/config"
"github.com/TriggerMail/luci-go/logdog/appengine/coordinator/endpoints"
"github.com/TriggerMail/luci-go/logdog/appengine/coordinator/flex"
"github.com/TriggerMail/luci-go/logdog/common/types"
)
// Services is a testing stub for a coordinator.Services instance that allows
// the user to configure the various services that are returned.
type Services struct {
// C, if not nil, will be used to get the return values for Config, overriding
// local static members.
C func() (*config.Config, error)
// PC, if not nil, will be used to get the return values for ProjectConfig,
// overriding local static members.
PC func() (*svcconfig.ProjectConfig, error)
// Storage returns an intermediate storage instance for use by this service.
//
// The caller must close the returned instance if successful.
//
// By default, this will return a *BigTableStorage instance bound to the
// Environment's BigTable instance if the stream is not archived, and an
// *ArchivalStorage instance bound to this Environment's GSClient instance
// if the stream is archived.
ST func(*coordinator.LogStreamState) (coordinator.SigningStorage, error)
// ArchivalPublisher returns an ArchivalPublisher instance.
AP func() (coordinator.ArchivalPublisher, error)
}
var _ endpoints.Services = (*Services)(nil)
var _ flex.Services = (*Services)(nil)
// Config implements coordinator.Services.
func (s *Services) Config(c context.Context) (*config.Config, error) {
if s.C != nil {
return s.C()
}
return config.Load(c)
}
// ProjectConfig implements coordinator.Services.
func (s *Services) ProjectConfig(c context.Context, project types.ProjectName) (*svcconfig.ProjectConfig, error) {
if s.PC != nil {
return s.PC()
}
return config.ProjectConfig(c, project)
}
// StorageForStream implements coordinator.Services.
func (s *Services) StorageForStream(c context.Context, lst *coordinator.LogStreamState) (coordinator.SigningStorage, error) {
if s.ST != nil {
return s.ST(lst)
}
panic("not implemented")
}
// ArchivalPublisher implements coordinator.Services.
func (s *Services) ArchivalPublisher(context.Context) (coordinator.ArchivalPublisher, error) {
if s.AP != nil {
return s.AP()
}
panic("not implemented")
}