-
Notifications
You must be signed in to change notification settings - Fork 2
/
operator.go
131 lines (110 loc) · 3.14 KB
/
operator.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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
// 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.
//
// author: wsfuyibing <websearch@163.com>
// date: 2023-03-04
package tracers
import (
"fmt"
"github.com/fuyibing/log/v5/common"
"github.com/fuyibing/log/v5/loggers"
"net"
"os"
"runtime"
"strings"
"sync"
)
var (
// Operator
// 链路操作.
Operator OperatorManager
nilTraceId TraceId
nilSpanId SpanId
)
type (
// OperatorManager
// 链路操作接口.
OperatorManager interface {
// Generator
// ID生成器.
Generator() (generator *id)
// GetExecutor
// 执行器.
GetExecutor() (executor Executor)
// GetResource
// 基础资源.
GetResource() (kv loggers.Kv)
// Push
// 推送跨度.
Push(span Span)
// SetExecutor
// 设置执行器.
SetExecutor(executor Executor)
}
operator struct {
executor Executor
generator *id
name string
resource loggers.Kv
}
)
// /////////////////////////////////////////////////////////////////////////////
// Interface methods
// /////////////////////////////////////////////////////////////////////////////
func (o *operator) Generator() (generator *id) { return o.generator }
func (o *operator) GetExecutor() (executor Executor) { return o.executor }
func (o *operator) GetResource() (kv loggers.Kv) { return o.resource }
func (o *operator) Push(span Span) { o.push(span) }
func (o *operator) SetExecutor(executor Executor) { o.executor = executor }
// /////////////////////////////////////////////////////////////////////////////
// Access and constructor
// /////////////////////////////////////////////////////////////////////////////
func (o *operator) init() *operator {
o.generator = (&id{}).init()
o.name = "tracers.operator"
o.resource = loggers.Kv{}
o.initResource()
return o
}
func (o *operator) initResource() {
// 基础项.
o.resource.Add("process.id", os.Getpid()).
Add("system.arch", fmt.Sprintf("%s/%s", runtime.GOOS, runtime.GOARCH)).
Add("system.env", runtime.Version())
// 主机名.
if s, se := os.Hostname(); se == nil {
o.resource.Add("system.host", s)
}
// IP地址.
if l, le := net.InterfaceAddrs(); le == nil {
ls := make([]string, 0)
for _, la := range l {
if ipn, ok := la.(*net.IPNet); ok && !ipn.IP.IsLoopback() {
if ipn.IP.To4() != nil {
ls = append(ls, ipn.IP.String())
}
}
}
o.resource.Add("system.addr", strings.Join(ls, ", "))
}
}
func (o *operator) push(span Span) {
// 链路禁用.
if o.executor == nil {
return
}
// 推送链路.
if err := o.executor.Publish(span); err != nil {
common.InternalFatal("<%s> send: %v", o.name, err)
}
}
func init() { new(sync.Once).Do(func() { Operator = (&operator{}).init() }) }