Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
milvus
milvus
提交
346bb1ba
M
milvus
项目概览
milvus
/
milvus
12 个月 前同步成功
通知
261
Star
22476
Fork
2472
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
milvus
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
346bb1ba
编写于
1月 26, 2021
作者:
C
cai.zhang
提交者:
yefu.chen
1月 26, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Delete some unused code and some debug logs
Signed-off-by:
N
cai.zhang
<
cai.zhang@zilliz.com
>
上级
c6a6b143
变更
17
隐藏空白更改
内联
并排
Showing
17 changed file
with
74 addition
and
301 deletion
+74
-301
.jenkins/modules/Publish/Publish.groovy
.jenkins/modules/Publish/Publish.groovy
+8
-22
.jenkins/modules/Regression/PythonRegression.groovy
.jenkins/modules/Regression/PythonRegression.groovy
+2
-6
Makefile
Makefile
+6
-2
build/docker/deploy/distributed/.env
build/docker/deploy/distributed/.env
+0
-9
build/docker/deploy/distributed/docker-compose.yml
build/docker/deploy/distributed/docker-compose.yml
+0
-36
build/docker/deploy/docker-compose.yml
build/docker/deploy/docker-compose.yml
+29
-0
build/docker/deploy/indexnode/DockerFile
build/docker/deploy/indexnode/DockerFile
+2
-2
build/docker/deploy/indexservice/DockerFile
build/docker/deploy/indexservice/DockerFile
+3
-3
cmd/distributed/indexnode/main.go
cmd/distributed/indexnode/main.go
+1
-1
cmd/distributed/indexservice/main.go
cmd/distributed/indexservice/main.go
+1
-1
cmd/indexbuilder/indexbuilder.go
cmd/indexbuilder/indexbuilder.go
+0
-53
internal/distributed/indexnode/service.go
internal/distributed/indexnode/service.go
+20
-26
internal/distributed/indexservice/client/client.go
internal/distributed/indexservice/client/client.go
+1
-107
internal/distributed/indexservice/service.go
internal/distributed/indexservice/service.go
+1
-14
internal/indexnode/indexnode.go
internal/indexnode/indexnode.go
+0
-3
internal/indexnode/task_scheduler.go
internal/indexnode/task_scheduler.go
+0
-2
internal/indexservice/meta_table.go
internal/indexservice/meta_table.go
+0
-14
未找到文件。
.jenkins/modules/Publish/Publish.groovy
浏览文件 @
346bb1ba
...
...
@@ -25,6 +25,14 @@ dir ('build/docker/deploy') {
sh
'docker-compose build --force-rm proxynode'
sh
'docker-compose push proxynode'
sh
'docker pull ${SOURCE_REPO}/indexservice:${SOURCE_TAG} || true'
sh
'docker-compose build --force-rm indexservice'
sh
'docker-compose push indexservice'
sh
'docker pull ${SOURCE_REPO}/indexnode:${SOURCE_TAG} || true'
sh
'docker-compose build --force-rm indexnode'
sh
'docker-compose push indexnode'
sh
'docker pull ${SOURCE_REPO}/queryservice:${SOURCE_TAG} || true'
sh
'docker-compose build --force-rm queryservice'
sh
'docker-compose push queryservice'
...
...
@@ -47,25 +55,3 @@ dir ('build/docker/deploy') {
sh
'docker-compose down --rmi all'
}
}
dir
(
'build/docker/deploy/distributed'
)
{
try
{
withCredentials
([
usernamePassword
(
credentialsId:
"${env.DOCKER_CREDENTIALS_ID}"
,
usernameVariable:
'DOCKER_USERNAME'
,
passwordVariable:
'DOCKER_PASSWORD'
)])
{
sh
'docker login -u ${DOCKER_USERNAME} -p ${DOCKER_PASSWORD} ${DOKCER_REGISTRY_URL}'
sh
'docker pull ${SOURCE_REPO}/indexservice:${SOURCE_TAG} || true'
sh
'docker-compose build --force-rm indexservice'
sh
'docker-compose push indexservice'
sh
'docker pull ${SOURCE_REPO}/indexnode:${SOURCE_TAG} || true'
sh
'docker-compose build --force-rm indexnode'
sh
'docker-compose push indexnode'
}
}
catch
(
exc
)
{
throw
exc
}
finally
{
sh
'docker logout ${DOKCER_REGISTRY_URL}'
sh
"docker rmi -f \$(docker images | grep '<none>' | awk '{print \$3}') || true"
sh
'docker-compose down --rmi all'
}
}
.jenkins/modules/Regression/PythonRegression.groovy
浏览文件 @
346bb1ba
...
...
@@ -8,18 +8,14 @@ try {
sh
'docker-compose -p ${DOCKER_COMPOSE_PROJECT_NAME} up -d master'
sh
'docker-compose -p ${DOCKER_COMPOSE_PROJECT_NAME} up -d proxyservice'
sh
'docker-compose -p ${DOCKER_COMPOSE_PROJECT_NAME} up -d proxynode'
sh
'docker-compose -p ${DOCKER_COMPOSE_PROJECT_NAME} up -d indexservice'
sh
'docker-compose -p ${DOCKER_COMPOSE_PROJECT_NAME} up -d indexnode'
sh
'docker-compose -p ${DOCKER_COMPOSE_PROJECT_NAME} up -d queryservice'
sh
'docker-compose -p ${DOCKER_COMPOSE_PROJECT_NAME} run -e QUERY_NODE_ID=1 -d querynode'
sh
'docker-compose -p ${DOCKER_COMPOSE_PROJECT_NAME} run -e QUERY_NODE_ID=2 -d querynode'
sh
'docker-compose -p ${DOCKER_COMPOSE_PROJECT_NAME} run -e WRITE_NODE_ID=3 -d writenode'
}
dir
(
'build/docker/deploy/distributed'
)
{
sh
'docker-compose -p ${DOCKER_COMPOSE_PROJECT_NAME} pull'
sh
'docker-compose -p ${DOCKER_COMPOSE_PROJECT_NAME} up -d indexservice'
sh
'docker-compose -p ${DOCKER_COMPOSE_PROJECT_NAME} up -d indexnode'
}
dir
(
'build/docker/test'
)
{
sh
'docker pull ${SOURCE_REPO}/pytest:${SOURCE_TAG} || true'
sh
'docker-compose build --force-rm regression'
...
...
Makefile
浏览文件 @
346bb1ba
...
...
@@ -130,9 +130,9 @@ build-go: build-cpp
@
echo
"Building singlenode ..."
@
mkdir
-p
$(INSTALL_PATH)
&&
go
env
-w
CGO_ENABLED
=
"1"
&&
GO111MODULE
=
on
$(GO)
build
-o
$(INSTALL_PATH)
/singlenode
$(PWD)
/cmd/singlenode/main.go 1>/dev/null
@
echo
"Building distributed indexservice ..."
@
mkdir
-p
$(INSTALL_PATH)
/distributed
&&
go
env
-w
CGO_ENABLED
=
"1"
&&
GO111MODULE
=
on
$(GO)
build
-o
$(INSTALL_PATH)
/distributed
/indexservice
$(PWD)
/cmd/distributed/indexservice/main.go 1>/dev/null
@
mkdir
-p
$(INSTALL_PATH)
&&
go
env
-w
CGO_ENABLED
=
"1"
&&
GO111MODULE
=
on
$(GO)
build
-o
$(INSTALL_PATH)
/indexservice
$(PWD)
/cmd/distributed/indexservice/main.go 1>/dev/null
@
echo
"Building distributed indexnode ..."
@
mkdir
-p
$(INSTALL_PATH)
/distributed
&&
go
env
-w
CGO_ENABLED
=
"1"
&&
GO111MODULE
=
on
$(GO)
build
-o
$(INSTALL_PATH)
/distributed
/indexnode
$(PWD)
/cmd/distributed/indexnode/main.go 1>/dev/null
@
mkdir
-p
$(INSTALL_PATH)
&&
go
env
-w
CGO_ENABLED
=
"1"
&&
GO111MODULE
=
on
$(GO)
build
-o
$(INSTALL_PATH)
/indexnode
$(PWD)
/cmd/distributed/indexnode/main.go 1>/dev/null
build-cpp
:
...
...
@@ -175,6 +175,8 @@ install: all
@
mkdir
-p
$(GOPATH)
/bin
&&
cp
-f
$(PWD)
/bin/proxyservice
$(GOPATH)
/bin/proxyservice
@
mkdir
-p
$(GOPATH)
/bin
&&
cp
-f
$(PWD)
/bin/writenode
$(GOPATH)
/bin/writenode
@
mkdir
-p
$(GOPATH)
/bin
&&
cp
-f
$(PWD)
/bin/singlenode
$(GOPATH)
/bin/singlenode
@
mkdir
-p
$(GOPATH)
/bin
&&
cp
-f
$(PWD)
/bin/indexservice
$(GOPATH)
/bin/indexservice
@
mkdir
-p
$(GOPATH)
/bin
&&
cp
-f
$(PWD)
/bin/indexnode
$(GOPATH)
/bin/indexnode
@
mkdir
-p
$(LIBRARY_PATH)
&&
cp
-f
$(PWD)
/internal/core/output/lib/
*
$(LIBRARY_PATH)
@
echo
"Installation successful."
...
...
@@ -190,3 +192,5 @@ clean:
@
rm
-rf
$(GOPATH)
/bin/querynode
@
rm
-rf
$(GOPATH)
/bin/writenode
@
rm
-rf
$(GOPATH)
/bin/singlenode
@
rm
-rf
$(GOPATH)
/bin/indexservice
@
rm
-rf
$(GOPATH)
/bin/indexnode
build/docker/deploy/distributed/.env
已删除
100644 → 0
浏览文件 @
c6a6b143
SOURCE_REPO=milvusdb
TARGET_REPO=milvusdb
SOURCE_TAG=latest
TARGET_TAG=latest
PULSAR_ADDRESS=pulsar://pulsar:6650
ETCD_ADDRESS=etcd:2379
MASTER_ADDRESS=master:53100
MINIO_ADDRESS=minio:9000
INDEX_SERVICE_ADDRESS=indexservice:31000
build/docker/deploy/distributed/docker-compose.yml
已删除
100644 → 0
浏览文件 @
c6a6b143
version
:
'
3.5'
services
:
indexservice
:
image
:
${TARGET_REPO}/indexservice:${TARGET_TAG}
build
:
context
:
../../../../
dockerfile
:
build/docker/deploy/distributed/indexservice/DockerFile
cache_from
:
-
${SOURCE_REPO}/indexservice:${SOURCE_TAG}
environment
:
MASTER_ADDRESS
:
${MASTER_ADDRESS}
ETCD_ADDRESS
:
${ETCD_ADDRESS}
MINIO_ADDRESS
:
${MINIO_ADDRESS}
networks
:
-
milvus
indexnode
:
image
:
${TARGET_REPO}/indexnode:${TARGET_TAG}
build
:
context
:
../../../../
dockerfile
:
build/docker/deploy/distributed/indexnode/DockerFile
cache_from
:
-
${SOURCE_REPO}/indexnode:${SOURCE_TAG}
environment
:
INDEX_SERVICE_ADDRESS
:
${INDEX_SERVICE_ADDRESS}
MINIO_ADDRESS
:
${MINIO_ADDRESS}
depends_on
:
-
"
indexservice"
networks
:
-
milvus
networks
:
milvus
:
build/docker/deploy/docker-compose.yml
浏览文件 @
346bb1ba
...
...
@@ -85,5 +85,34 @@ services:
networks
:
-
milvus
indexservice
:
image
:
${TARGET_REPO}/indexservice:${TARGET_TAG}
build
:
context
:
../../../
dockerfile
:
build/docker/deploy/indexservice/DockerFile
cache_from
:
-
${SOURCE_REPO}/indexservice:${SOURCE_TAG}
environment
:
MASTER_ADDRESS
:
${MASTER_ADDRESS}
ETCD_ADDRESS
:
${ETCD_ADDRESS}
MINIO_ADDRESS
:
${MINIO_ADDRESS}
networks
:
-
milvus
indexnode
:
image
:
${TARGET_REPO}/indexnode:${TARGET_TAG}
build
:
context
:
../../../
dockerfile
:
build/docker/deploy/indexnode/DockerFile
cache_from
:
-
${SOURCE_REPO}/indexnode:${SOURCE_TAG}
environment
:
INDEX_SERVICE_ADDRESS
:
${INDEX_SERVICE_ADDRESS}
MINIO_ADDRESS
:
${MINIO_ADDRESS}
depends_on
:
-
"
indexservice"
networks
:
-
milvus
networks
:
milvus
:
build/docker/deploy/
distributed/
indexnode/DockerFile
→
build/docker/deploy/indexnode/DockerFile
浏览文件 @
346bb1ba
...
...
@@ -26,7 +26,7 @@ COPY --from=openblas /usr/lib/libopenblas-r0.3.9.so /usr/lib/
RUN
ln
-s
/usr/lib/libopenblas-r0.3.9.so /usr/lib/libopenblas.so.0
&&
\
ln
-s
/usr/lib/libopenblas.so.0 /usr/lib/libopenblas.so
COPY
./bin/
distributed/indexnode /milvus-distributed/bin/distributed
/indexnode
COPY
./bin/
indexnode /milvus-distributed/bin
/indexnode
COPY
./configs/ /milvus-distributed/configs/
...
...
@@ -36,5 +36,5 @@ ENV LD_LIBRARY_PATH=/milvus-distributed/lib:$LD_LIBRARY_PATH:/usr/lib
WORKDIR
/milvus-distributed/
CMD
["./bin/
distributed/
indexnode"]
CMD
["./bin/indexnode"]
build/docker/deploy/
distributed/
indexservice/DockerFile
→
build/docker/deploy/indexservice/DockerFile
浏览文件 @
346bb1ba
...
...
@@ -26,7 +26,7 @@ COPY --from=openblas /usr/lib/libopenblas-r0.3.9.so /usr/lib/
RUN
ln
-s
/usr/lib/libopenblas-r0.3.9.so /usr/lib/libopenblas.so.0
&&
\
ln
-s
/usr/lib/libopenblas.so.0 /usr/lib/libopenblas.so
COPY
./bin/
distributed/indexservice /milvus-distributed/bin/distributed
/indexservice
COPY
./bin/
indexservice /milvus-distributed/bin
/indexservice
COPY
./configs/ /milvus-distributed/configs/
...
...
@@ -36,6 +36,6 @@ ENV LD_LIBRARY_PATH=/milvus-distributed/lib:$LD_LIBRARY_PATH:/usr/lib
WORKDIR
/milvus-distributed/
CMD
["./bin/
distributed/
indexservice"]
CMD
["./bin/indexservice"]
EXPOSE
21118
EXPOSE
31000
cmd/distributed/indexnode/main.go
浏览文件 @
346bb1ba
...
...
@@ -25,7 +25,7 @@ import (
func
main
()
{
ctx
,
cancel
:=
context
.
WithCancel
(
context
.
Background
())
grpcindexnode
.
Init
()
svr
,
err
:=
grpcindexnode
.
CreateIndexNode
(
ctx
)
svr
,
err
:=
grpcindexnode
.
NewGrpcServer
(
ctx
)
if
err
!=
nil
{
log
.
Print
(
"create server failed"
,
zap
.
Error
(
err
))
}
...
...
cmd/distributed/indexservice/main.go
浏览文件 @
346bb1ba
...
...
@@ -25,7 +25,7 @@ import (
func
main
()
{
ctx
,
cancel
:=
context
.
WithCancel
(
context
.
Background
())
grpcindexserver
.
Init
()
svr
,
err
:=
grpcindexserver
.
CreateIndex
Server
(
ctx
)
svr
,
err
:=
grpcindexserver
.
New
Server
(
ctx
)
if
err
!=
nil
{
log
.
Print
(
"create server failed"
,
zap
.
Error
(
err
))
}
...
...
cmd/indexbuilder/indexbuilder.go
已删除
100644 → 0
浏览文件 @
c6a6b143
package
main
//import (
// "context"
// "log"
// "os"
// "os/signal"
// "syscall"
//
// "github.com/zilliztech/milvus-distributed/internal/indexnode"
// "go.uber.org/zap"
//)
//
//func main() {
// indexnode.Init()
// ctx, cancel := context.WithCancel(context.Background())
// svr, err := indexnode.CreateIndexNode(ctx)
// if err != nil {
// log.Print("create server failed", zap.Error(err))
// }
//
// sc := make(chan os.Signal, 1)
// signal.Notify(sc,
// syscall.SIGHUP,
// syscall.SIGINT,
// syscall.SIGTERM,
// syscall.SIGQUIT)
//
// var sig os.Signal
// go func() {
// sig = <-sc
// cancel()
// }()
//
// if err := svr.Start(); err != nil {
// log.Fatal("run builder server failed", zap.Error(err))
// }
//
// <-ctx.Done()
// log.Print("Got signal to exit", zap.String("signal", sig.String()))
//
// svr.Stop()
// switch sig {
// case syscall.SIGTERM:
// exit(0)
// default:
// exit(1)
// }
//}
//
//func exit(code int) {
// os.Exit(code)
//}
internal/distributed/indexnode/service.go
浏览文件 @
346bb1ba
...
...
@@ -25,26 +25,6 @@ type Server struct {
loopWg
sync
.
WaitGroup
}
func
NewGrpcServer
(
ctx
context
.
Context
)
*
Server
{
ctx1
,
cancel
:=
context
.
WithCancel
(
ctx
)
indexServiceClient
:=
serviceclient
.
NewClient
(
indexnode
.
Params
.
ServiceAddress
)
node
,
err
:=
indexnode
.
CreateIndexNode
(
ctx1
)
if
err
!=
nil
{
defer
cancel
()
return
nil
}
node
.
SetServiceClient
(
indexServiceClient
)
return
&
Server
{
loopCtx
:
ctx1
,
loopCancel
:
cancel
,
node
:
node
,
serverClient
:
indexServiceClient
,
}
}
func
(
s
*
Server
)
registerNode
()
error
{
log
.
Printf
(
"Registering node. IP = %s, Port = %d"
,
indexnode
.
Params
.
NodeIP
,
indexnode
.
Params
.
NodePort
)
...
...
@@ -132,11 +112,6 @@ func Init() error {
return
nil
}
func
CreateIndexNode
(
ctx
context
.
Context
)
(
*
Server
,
error
)
{
return
NewGrpcServer
(
ctx
),
nil
}
func
(
s
*
Server
)
Start
()
error
{
return
s
.
startIndexNode
()
}
...
...
@@ -153,6 +128,25 @@ func (s *Server) Stop() error {
}
func
(
s
*
Server
)
BuildIndex
(
ctx
context
.
Context
,
req
*
indexpb
.
BuildIndexCmd
)
(
*
commonpb
.
Status
,
error
)
{
log
.
Println
(
"distributed build index"
)
return
s
.
node
.
BuildIndex
(
req
)
}
func
NewGrpcServer
(
ctx
context
.
Context
)
(
*
Server
,
error
)
{
ctx1
,
cancel
:=
context
.
WithCancel
(
ctx
)
indexServiceClient
:=
serviceclient
.
NewClient
(
indexnode
.
Params
.
ServiceAddress
)
node
,
err
:=
indexnode
.
CreateIndexNode
(
ctx1
)
if
err
!=
nil
{
defer
cancel
()
return
nil
,
err
}
node
.
SetServiceClient
(
indexServiceClient
)
return
&
Server
{
loopCtx
:
ctx1
,
loopCancel
:
cancel
,
node
:
node
,
serverClient
:
indexServiceClient
,
},
nil
}
internal/distributed/indexservice/client/client.go
浏览文件 @
346bb1ba
...
...
@@ -31,112 +31,6 @@ func (g *Client) Stop() error {
return
nil
}
//func (g *Client) BuildIndex2(columnDataPaths []string, typeParams map[string]string, indexParams map[string]string) (UniqueID, error) {
//
// parseMap := func(mStr string) (map[string]string, error) {
// buffer := make(map[string]interface{})
// err := json.Unmarshal([]byte(mStr), &buffer)
// if err != nil {
// return nil, errors.New("Unmarshal params failed")
// }
// ret := make(map[string]string)
// for key, value := range buffer {
// valueStr := fmt.Sprintf("%v", value)
// ret[key] = valueStr
// }
// return ret, nil
// }
// var typeParamsKV []*commonpb.KeyValuePair
// for key := range typeParams {
// if key == "params" {
// mapParams, err := parseMap(typeParams[key])
// if err != nil {
// log.Println("parse params error: ", err)
// }
// for pk, pv := range mapParams {
// typeParamsKV = append(typeParamsKV, &commonpb.KeyValuePair{
// Key: pk,
// Value: pv,
// })
// }
// } else {
// typeParamsKV = append(typeParamsKV, &commonpb.KeyValuePair{
// Key: key,
// Value: typeParams[key],
// })
// }
// }
//
// var indexParamsKV []*commonpb.KeyValuePair
// for key := range indexParams {
// if key == "params" {
// mapParams, err := parseMap(indexParams[key])
// if err != nil {
// log.Println("parse params error: ", err)
// }
// for pk, pv := range mapParams {
// indexParamsKV = append(indexParamsKV, &commonpb.KeyValuePair{
// Key: pk,
// Value: pv,
// })
// }
// } else {
// indexParamsKV = append(indexParamsKV, &commonpb.KeyValuePair{
// Key: key,
// Value: indexParams[key],
// })
// }
// }
//
// requset := &indexpb.BuildIndexRequest{
// DataPaths: columnDataPaths,
// TypeParams: typeParamsKV,
// IndexParams: indexParamsKV,
// }
// response, err := g.BuildIndex(requset)
// if err != nil {
// return 0, err
// }
//
// indexID := response.IndexID
//
// return indexID, nil
//}
//
//func (g *Client) GetIndexStates2(indexIDs []UniqueID) (*indexpb.IndexStatesResponse, error) {
//
// request := &indexpb.IndexStatesRequest{
// IndexIDs: indexIDs,
// }
//
// response, err := g.GetIndexStates(request)
// return response, err
//}
//
//func (g *Client) GetIndexFilePaths2(indexIDs []UniqueID) ([][]string, error) {
//
// request := &indexpb.IndexFilePathsRequest{
// IndexIDs: indexIDs,
// }
//
// response, err := g.GetIndexFilePaths(request)
// if err != nil {
// return nil, err
// }
//
// var filePaths [][]string
// for _, indexID := range indexIDs {
// for _, filePathInfo := range response.FilePaths {
// if indexID == filePathInfo.IndexID {
// filePaths = append(filePaths, filePathInfo.IndexFilePaths)
// break
// }
// }
// }
//
// return filePaths, nil
//}
func
(
g
*
Client
)
GetComponentStates
()
(
*
internalpb2
.
ComponentStates
,
error
)
{
return
nil
,
nil
}
...
...
@@ -155,7 +49,7 @@ func (g *Client) tryConnect() error {
}
ctx1
,
cancel
:=
context
.
WithTimeout
(
context
.
Background
(),
10
*
time
.
Second
)
defer
cancel
()
log
.
Println
(
"indexservice address = "
,
g
.
address
)
conn
,
err
:=
grpc
.
DialContext
(
ctx1
,
g
.
address
,
grpc
.
WithInsecure
(),
grpc
.
WithBlock
())
if
err
!=
nil
{
log
.
Printf
(
"Connect to IndexService failed, error= %v"
,
err
)
...
...
internal/distributed/indexservice/service.go
浏览文件 @
346bb1ba
...
...
@@ -62,13 +62,11 @@ func (s *Server) GetStatisticsChannel() (string, error) {
func
(
s
*
Server
)
RegisterNode
(
ctx
context
.
Context
,
req
*
indexpb
.
RegisterNodeRequest
)
(
*
indexpb
.
RegisterNodeResponse
,
error
)
{
log
.
Println
(
"Register IndexNode starting..., node address = "
,
req
.
Address
)
return
s
.
server
.
RegisterNode
(
req
)
}
func
(
s
*
Server
)
BuildIndex
(
ctx
context
.
Context
,
req
*
indexpb
.
BuildIndexRequest
)
(
*
indexpb
.
BuildIndexResponse
,
error
)
{
log
.
Println
(
"Build Index ..."
)
return
s
.
server
.
BuildIndex
(
req
)
}
...
...
@@ -84,22 +82,12 @@ func (s *Server) GetIndexFilePaths(ctx context.Context, req *indexpb.IndexFilePa
func
(
s
*
Server
)
NotifyBuildIndex
(
ctx
context
.
Context
,
nty
*
indexpb
.
BuildIndexNotification
)
(
*
commonpb
.
Status
,
error
)
{
log
.
Println
(
"build index finished."
)
return
s
.
server
.
NotifyBuildIndex
(
nty
)
}
func
NewServer
()
*
Server
{
return
&
Server
{
//server: &indexservice.IndexService{},
//grpcServer: indexservice.IndexService{},
}
}
func
(
s
*
Server
)
grpcLoop
()
{
defer
s
.
loopWg
.
Done
()
log
.
Println
(
"Starting start IndexServer"
)
lis
,
err
:=
net
.
Listen
(
"tcp"
,
":"
+
strconv
.
Itoa
(
indexservice
.
Params
.
Port
))
if
err
!=
nil
{
log
.
Fatalf
(
"IndexServer grpc server fatal error=%v"
,
err
)
...
...
@@ -108,7 +96,6 @@ func (s *Server) grpcLoop() {
s
.
grpcServer
=
grpc
.
NewServer
()
indexpb
.
RegisterIndexServiceServer
(
s
.
grpcServer
,
s
)
log
.
Println
(
"IndexServer register finished"
)
if
err
=
s
.
grpcServer
.
Serve
(
lis
);
err
!=
nil
{
log
.
Fatalf
(
"IndexServer grpc server fatal error=%v"
,
err
)
}
...
...
@@ -122,7 +109,7 @@ func (s *Server) startIndexServer() error {
return
s
.
server
.
Start
()
}
func
CreateIndex
Server
(
ctx
context
.
Context
)
(
*
Server
,
error
)
{
func
New
Server
(
ctx
context
.
Context
)
(
*
Server
,
error
)
{
ctx1
,
cancel
:=
context
.
WithCancel
(
ctx
)
serverImp
,
err
:=
indexservice
.
CreateIndexService
(
ctx
)
...
...
internal/indexnode/indexnode.go
浏览文件 @
346bb1ba
...
...
@@ -71,9 +71,7 @@ func CreateIndexNode(ctx context.Context) (*IndexNode, error) {
return
nil
,
err
}
log
.
Println
(
"loopctx = "
,
b
.
loopCtx
)
b
.
sched
,
err
=
NewTaskScheduler
(
b
.
loopCtx
,
b
.
kv
)
log
.
Println
(
"err = "
,
err
)
if
err
!=
nil
{
return
nil
,
err
}
...
...
@@ -132,7 +130,6 @@ func (i *IndexNode) BuildIndex(request *indexpb.BuildIndexCmd) (*commonpb.Status
t
.
cmd
=
request
t
.
kv
=
i
.
kv
t
.
serviceClient
=
i
.
serviceClient
log
.
Println
(
"t.serviceClient = "
,
t
.
serviceClient
)
t
.
nodeID
=
Params
.
NodeID
ctx
,
cancel
:=
context
.
WithTimeout
(
context
.
Background
(),
reqTimeoutInterval
)
defer
cancel
()
...
...
internal/indexnode/task_scheduler.go
浏览文件 @
346bb1ba
...
...
@@ -216,10 +216,8 @@ func (sched *TaskScheduler) indexBuildLoop() {
for
{
select
{
case
<-
sched
.
ctx
.
Done
()
:
log
.
Println
(
"id Done?"
)
return
case
<-
sched
.
IndexBuildQueue
.
utChan
()
:
log
.
Println
(
"index build loop ..."
)
if
!
sched
.
IndexBuildQueue
.
utEmpty
()
{
tasks
:=
sched
.
scheduleIndexBuildTask
()
var
wg
sync
.
WaitGroup
...
...
internal/indexservice/meta_table.go
浏览文件 @
346bb1ba
...
...
@@ -106,20 +106,6 @@ func (mt *metaTable) NotifyBuildIndex(nty *indexpb.BuildIndexNotification) error
return
mt
.
saveIndexMeta
(
&
meta
)
}
func
(
mt
*
metaTable
)
NotifyBuildIndex2
(
indexID
UniqueID
,
dataPaths
[]
string
,
state
commonpb
.
IndexState
)
error
{
mt
.
lock
.
Lock
()
defer
mt
.
lock
.
Unlock
()
meta
,
ok
:=
mt
.
indexID2Meta
[
indexID
]
if
!
ok
{
return
errors
.
Errorf
(
"index not exists with ID = "
+
strconv
.
FormatInt
(
indexID
,
10
))
}
meta
.
State
=
state
meta
.
IndexFilePaths
=
dataPaths
return
mt
.
saveIndexMeta
(
&
meta
)
}
func
(
mt
*
metaTable
)
GetIndexState
(
indexID
UniqueID
)
(
*
indexpb
.
IndexInfo
,
error
)
{
mt
.
lock
.
Lock
()
defer
mt
.
lock
.
Unlock
()
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录