初步实现安装界面

internal/db/models/api_token_dao.go

@@ -24,7 +24,13 @@ func NewApiTokenDAO() *ApiTokenDAO {
 	}).(*ApiTokenDAO)
 }
 
-var SharedApiTokenDAO = NewApiTokenDAO()
+var SharedApiTokenDAO *ApiTokenDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedApiTokenDAO = NewApiTokenDAO()
+	})
+}
 
 // 启用条目
 func (this *ApiTokenDAO) EnableApiToken(id uint32) (rowsAffected int64, err error) {
@@ -67,6 +73,18 @@ func (this *ApiTokenDAO) FindEnabledTokenWithNode(nodeId string) (*ApiToken, err
 	return nil, err
 }
 
+// 根据角色获取节点
+func (this *ApiTokenDAO) FindEnabledTokenWithRole(role string) (*ApiToken, error) {
+	one, err := this.Query().
+		Attr("role", role).
+		State(ApiTokenStateEnabled).
+		Find()
+	if one != nil {
+		return one.(*ApiToken), nil
+	}
+	return nil, err
+}
+
 // 保存API Token
 func (this *ApiTokenDAO) CreateAPIToken(nodeId string, secret string, role NodeRole) error {
 	op := NewApiTokenOperator()