Create & Init Project...

app/service/main/push/api/grpc/v1/BUILD

@@ -0,0 +1,66 @@
+load(
+    "@io_bazel_rules_go//proto:def.bzl",
+    "go_proto_library",
+)
+
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+    "go_test",
+)
+
+proto_library(
+    name = "v1_proto",
+    srcs = ["api.proto"],
+    tags = ["automanaged"],
+    deps = ["@gogo_special_proto//github.com/gogo/protobuf/gogoproto"],
+)
+
+go_proto_library(
+    name = "v1_go_proto",
+    compilers = ["@io_bazel_rules_go//proto:gogofast_grpc"],
+    importpath = "go-common/app/service/main/push/api/grpc/v1",
+    proto = ":v1_proto",
+    tags = ["automanaged"],
+    deps = ["@com_github_gogo_protobuf//gogoproto:go_default_library"],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["client.go"],
+    embed = [":v1_go_proto"],
+    importpath = "go-common/app/service/main/push/api/grpc/v1",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//library/net/rpc/warden:go_default_library",
+        "@com_github_gogo_protobuf//gogoproto:go_default_library",
+        "@com_github_gogo_protobuf//proto:go_default_library",
+        "@org_golang_google_grpc//:go_default_library",
+        "@org_golang_x_net//context:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = ["client_test.go"],
+    embed = [":go_default_library"],
+    tags = ["automanaged"],
+    deps = ["//library/net/rpc/warden:go_default_library"],
+)

app/service/main/push/api/grpc/v1/api.proto

@@ -0,0 +1,125 @@
+syntax = "proto3";
+
+package push.service.v1;
+
+option go_package = "v1";
+
+import "github.com/gogo/protobuf/gogoproto/gogo.proto";
+
+service Push {
+    rpc AddReport (AddReportRequest) returns (AddReportReply) {}
+    rpc DelReport (DelReportRequest) returns (DelReportReply) {}
+    rpc DelInvalidReports (DelInvalidReportsRequest) returns (DelInvalidReportsReply) {}
+    rpc AddReportCache (AddReportCacheRequest) returns (AddReportCacheReply) {}
+    rpc AddUserReportCache (AddUserReportCacheRequest) returns (AddUserReportCacheReply) {}
+    rpc AddTokenCache (AddTokenCacheRequest) returns (AddTokenCacheReply) {}
+    rpc AddTokensCache (AddTokensCacheRequest) returns (AddTokensCacheReply) {}
+    rpc AddCallback (AddCallbackRequest) returns (AddCallbackReply) {}
+    rpc AddMidProgress (AddMidProgressRequest) returns (AddMidProgressReply) {}
+    rpc Setting (SettingRequest) returns (SettingReply) {}
+    rpc SetSetting (SetSettingRequest) returns (SetSettingReply) {}
+}
+
+message ModelReport {
+    int64 id = 1 [(gogoproto.customname) = "ID"];
+    int32 app_id = 2 [(gogoproto.customname) = "APPID"];
+    int32 platform_id = 3 [(gogoproto.customname) = "PlatformID"];
+    int64 mid = 4;
+    string buvid = 5;
+    string device_token = 6;
+    int32 build = 7;
+    int32 time_zone = 8;
+    int32 notify_switch = 9;
+    string device_brand = 10;
+    string device_model = 11;
+    string os_version = 12 [(gogoproto.customname) = "OSVersion"];
+    string extra = 13;
+}
+
+message AddReportRequest {
+    ModelReport report = 1;
+}
+
+message AddReportReply {}
+
+message DelReportRequest {
+    int32 app_id = 1 [(gogoproto.customname) = "APPID"];
+    int64 mid = 2;
+    string device_token = 3;
+}
+
+message DelReportReply {}
+
+message DelInvalidReportsRequest {
+    int32 type = 1;
+}
+
+message DelInvalidReportsReply {}
+
+message AddReportCacheRequest {
+    ModelReport report = 1;
+}
+
+message AddReportCacheReply {}
+
+message AddUserReportCacheRequest {
+    int64 mid = 1;
+    repeated ModelReport reports = 2;
+}
+
+message AddUserReportCacheReply {}
+
+message AddTokenCacheRequest {
+    ModelReport report = 1;
+}
+
+message AddTokenCacheReply {}
+
+message AddTokensCacheRequest {
+    repeated ModelReport reports = 2;
+}
+
+message AddTokensCacheReply {}
+
+message AddCallbackRequest {
+    string task = 1;
+    int64 app = 2 [(gogoproto.customname) = "APP"];
+    int32 platform = 3;
+    int64 mid = 4;
+    int32 pid = 5;
+    string token = 6;
+    string buvid = 7;
+    int32 click = 8;
+    CallbackExtra extra = 9;
+}
+
+message AddCallbackReply {}
+
+message CallbackExtra {
+    int32 status = 1 [(gogoproto.moretags) = 'json:"st'];
+    int32 channel = 2 [(gogoproto.moretags) = 'json:"chan"'];
+}
+
+message AddMidProgressRequest {
+    string task = 1;
+    int64 mid_total = 2;
+    int64 mid_valid = 3;
+}
+
+message AddMidProgressReply {}
+
+message SettingRequest {
+    int64 mid = 1;
+}
+
+message SettingReply {
+    map<int32, int32> settings = 1;
+}
+
+message SetSettingRequest {
+    int64 mid = 1;
+    int32 type = 2;
+    int32 value = 3;
+}
+
+message SetSettingReply {}

app/service/main/push/api/grpc/v1/client.go

@@ -0,0 +1,25 @@
+package v1
+
+import (
+	"context"
+	"fmt"
+
+	"go-common/library/net/rpc/warden"
+
+	"google.golang.org/grpc"
+)
+
+// AppID .
+const AppID = "push.service"
+
+// NewClient new grpc client
+func NewClient(cfg *warden.ClientConfig, opts ...grpc.DialOption) (PushClient, error) {
+	client := warden.NewClient(cfg, opts...)
+	cc, err := client.Dial(context.Background(), fmt.Sprintf("discovery://default/%s", AppID))
+	if err != nil {
+		return nil, err
+	}
+	return NewPushClient(cc), nil
+}
+
+//go:generate $GOPATH/src/go-common/app/tool/warden/protoc.sh

app/service/main/push/api/grpc/v1/client_test.go

@@ -0,0 +1,57 @@
+package v1
+
+import (
+	"context"
+	"fmt"
+	"testing"
+
+	"go-common/library/net/rpc/warden"
+)
+
+var (
+	err    error
+	rpccli PushClient
+)
+
+func init() {
+	rpccli, err = NewClient(&warden.ClientConfig{})
+	if err != nil {
+		fmt.Printf("new push grpc client error(%v)", err)
+	}
+}
+
+func Test_AddReport(t *testing.T) {
+	_, err := rpccli.AddReport(context.Background(), &AddReportRequest{Report: &ModelReport{
+		APPID:        1,
+		Mid:          91221505,
+		DeviceToken:  "tototototototo",
+		NotifySwitch: 1,
+	}})
+	if err != nil {
+		t.Errorf("AddReport error(%v)", err)
+	}
+}
+
+func Test_AddTokenCache(t *testing.T) {
+	_, err := rpccli.AddTokenCache(context.Background(), &AddTokenCacheRequest{Report: &ModelReport{
+		APPID:        1,
+		Mid:          91221505,
+		DeviceToken:  "tototototototo",
+		NotifySwitch: 1,
+	}})
+	if err != nil {
+		t.Errorf("AddTokenCache error(%v)", err)
+	}
+}
+
+func Test_AddTokensCache(t *testing.T) {
+	_, err := rpccli.AddTokensCache(context.Background(), &AddTokensCacheRequest{Reports: []*ModelReport{{
+		APPID:        1,
+		Mid:          91221505,
+		DeviceToken:  "tototototototo",
+		NotifySwitch: 1,
+	}}})
+	if err != nil {
+		t.Errorf("AddTokensCache error(%v)", err)
+	}
+}